File: //usr/local/lib/python3.10/sqlite3/test/__pycache__/backup.cpython-310.opt-2.pyc
o
�i� � @ sB d dl Zd dlZG dd� dej�Zdd� Zedkr e�� dS dS )� Nc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%S )&� BackupTestsc C s6 t �d� }| _|�d� |�dddg� |�� d S )N�:memory:z CREATE TABLE foo (key INTEGER)� INSERT INTO foo (key) VALUES (?)�� �� )�sqlite�connect�cx�execute� executemany�commit)�selfr � r �0/usr/local/lib/python3.10/sqlite3/test/backup.py�setUp s
zBackupTests.setUpc C s | j �� d S �N)r �close�r r r r �tearDown � zBackupTests.tearDownc C s: |� d��� }| �|d d d� | �|d d d� d S )Nz SELECT key FROM foo ORDER BY keyr r � r )r �fetchall� assertEqual)r Zbckcx�resultr r r �
verify_backup s zBackupTests.verify_backupc C sp | � t�� | j�d � W d � n1 sw Y | � t�� | j�� W d � d S 1 s1w Y d S r �� assertRaises� TypeErrorr �backupr r r r �test_bad_target s � "�zBackupTests.test_bad_targetc C s< | � t�� | j�d� W d � d S 1 sw Y d S )Nzsome_file_name.dbr r r r r �test_bad_target_filename s "�z$BackupTests.test_bad_target_filenamec C s> | � t�� | j�| j� W d � d S 1 sw Y d S r )r �
ValueErrorr r r r r r � test_bad_target_same_connection s "�z+BackupTests.test_bad_target_same_connectionc C sP t �d�}|�� | �t j�� | j�|� W d � d S 1 s!w Y d S �Nr )r r
r r �ProgrammingErrorr r �r �bckr r r �!test_bad_target_closed_connection" s
"�z-BackupTests.test_bad_target_closed_connectionc C sX t �d�}t �d�}|�� | �t j�� |�|� W d � d S 1 s%w Y d S r% )r r
r r r& r )r r( �sourcer r r �!test_bad_source_closed_connection( s
"�z-BackupTests.test_bad_source_closed_connectionc C s� t �d�}|�d� |�dddg� | �t j��}| j�|� W d � n1 s)w Y t jdk r>| � t
|j �d� d S d S )Nr z CREATE TABLE bar (key INTEGER)z INSERT INTO bar (key) VALUES (?)r r )r � r, ztarget is in transaction) r r
r r
r �OperationalErrorr r Zsqlite_version_infor �str� exception�r r( �cmr r r � test_bad_target_in_transaction/ s
�
�z*BackupTests.test_bad_target_in_transactionc C sx | � t��- t�d��}| j�|d� W d � n1 s w Y W d � d S W d � d S 1 s5w Y d S )Nr r )r r r r
r r r' r r r �test_keyword_only_args8 s ��"�z"BackupTests.test_keyword_only_argsc C sF t �d��}| j�|� | �|� W d � d S 1 s w Y d S r% )r r
r r r r' r r r � test_simple= s "�zBackupTests.test_simplec s� g � � fdd�}t �d��}| jj|d|d� | �|� W d � n1 s&w Y | �t� �d� | �� d t j� | �� d t j� d S )Nc s � � | � d S r ��append��status� remaining�total��journalr r �progressE r z+BackupTests.test_progress.<locals>.progressr r �Zpagesr= � r ) r r
r r r r �lenZ SQLITE_OKZ SQLITE_DONE�r r= r( r r; r �
test_progressB s �zBackupTests.test_progressc sx g � � fdd�}t �d��}| jj||d� | �|� W d � n1 s%w Y | �t� �d� | �� d d� d S )Nc � � � |� d S r r5 r7 r; r r r= S r z?BackupTests.test_progress_all_pages_at_once_1.<locals>.progressr �r= r r �r r
r r r r r@ rA r r; r �!test_progress_all_pages_at_once_1P s �z-BackupTests.test_progress_all_pages_at_once_1c sz g � � fdd�}t �d��}| jj|d|d� | �|� W d � n1 s&w Y | �t� �d� | �� d d� d S )Nc rC r r5 r7 r; r r r= ` r z?BackupTests.test_progress_all_pages_at_once_2.<locals>.progressr ���r> r r rE rA r r; r �!test_progress_all_pages_at_once_2] s �z-BackupTests.test_progress_all_pages_at_once_2c C s| | � t��&}t�d��}| jj|ddd� W d � n1 s w Y W d � n1 s.w Y | �t|j�d� d S )Nr r Zbarr> z$progress argument must be a callable) r r r r
r r r r. r/ )r r1 r( r r r �test_non_callable_progressj s �� �z&BackupTests.test_non_callable_progressc s� g � � �fdd�}t �d��'}�jj|d|d� ��|� |�d��� }��|d d d� W d � n1 s8w Y ��t� �d � ��� d d� ��� d d� ��� d
d� d S ) Nc s0 � s�j �d|d f� �j �� � �|� d S )Nr i� )r r r r6 r7 �r< r r r r= s s
z5BackupTests.test_modifying_progress.<locals>.progressr r r> z2SELECT key FROM foo WHERE key >= 1000 ORDER BY keyr i� r r? ) r r
r r r r r r r@ )r r= r( r r rJ r �test_modifying_progressp s
� � z#BackupTests.test_modifying_progressc C s� dd� }| � t��%}t�d��}| jj||d� W d � n1 s"w Y W d � n1 s1w Y | �t|j�d� d S )Nc S s t d��)N�nearly out of space)� SystemErrorr7 r r r r= � s z3BackupTests.test_failing_progress.<locals>.progressr rD rL ) r rM r r
r r r r. r/ )r r= �errr( r r r �test_failing_progress� s �� �z!BackupTests.test_failing_progressc C sj t �d��}| jj|dd� W d � n1 sw Y t �d��}| jj|dd� W d � n1 s5w Y | �t j��%}t �d��}| jj|dd� W d � n1 sYw Y W d � n1 shw Y | �dt|j�� | j� d� | j� d� | j�
d d
d g� | j� � t �d��}| jj|d d� | � |� W d � d S 1 s�w Y d S )
Nr �main)�nameZtempz non-existingzunknown databasez)ATTACH DATABASE ':memory:' AS attached_dbz*CREATE TABLE attached_db.foo (key INTEGER)z,INSERT INTO attached_db.foo (key) VALUES (?)r r Z attached_db)
r r
r r r r- ZassertInr. r/ r r
r r r0 r r r �test_database_source_name� s* � � �� �
"�z%BackupTests.test_database_source_nameN)�__name__�
__module__� __qualname__r r r r! r" r$ r) r+ r2 r3 r4 rB rF rH rI rK rO rR r r r r r s&
r c C s t �� �t�S r )�unittestZ
TestLoaderZloadTestsFromTestCaser r r r r �suite� r rW �__main__)Zsqlite3r rV ZTestCaser rW rS rP r r r r �<module> s �