File: //usr/local/lib/python3.7/sqlite3/test/__pycache__/backup.cpython-37.pyc
B
��g' � @ sP d dl Zd dlZe�ejdk d�G dd� dej��Zdd� ZedkrLe� � dS ) � N)� � � zBackup API not supportedc @ 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#S )$� BackupTestsc C s6 t �d� }| _|�d� |�dddg� |�� d S )Nz:memory:z CREATE TABLE foo (key INTEGER)z INSERT INTO foo (key) VALUES (?))r )� )�sqlite�connect�cx�execute� executemany�commit)�selfr � r �//usr/local/lib/python3.7/sqlite3/test/backup.py�setUp s
zBackupTests.setUpc C s | j �� d S )N)r �close)r
r r r �tearDown
s 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 s& | � t�� | j�d � W d Q R X d S )N)� assertRaises� TypeErrorr �backup)r
r r r �test_bad_target_none s z BackupTests.test_bad_target_nonec C s& | � t�� | j�d� W d Q R X d S )Nzsome_file_name.db)r r r r )r
r r r �test_bad_target_filename s z$BackupTests.test_bad_target_filenamec C s( | � t�� | j�| j� W d Q R X d S )N)r �
ValueErrorr r )r
r r r � test_bad_target_same_connection s z+BackupTests.test_bad_target_same_connectionc C s: t �d�}|�� | �t j�� | j�|� W d Q R X d S )Nz:memory:)r r r r ZProgrammingErrorr r )r
�bckr r r �!test_bad_target_closed_connection! s
z-BackupTests.test_bad_target_closed_connectionc C sh t �d�}|�d� |�dddg� | �t j��}| j�|� W d Q R X t jdk rd| � t
|j �d� d S )Nz:memory: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 �sqlite_version_infor �str� exception)r
r �cmr r r � test_bad_target_in_transaction' s
z*BackupTests.test_bad_target_in_transactionc
C s> | � t��* t�d��}| j�|d� W d Q R X W d Q R X d S )Nz:memory:r )r r r r r r )r
r r r r �test_keyword_only_args0 s z"BackupTests.test_keyword_only_argsc C s0 t �d�� }| j�|� | �|� W d Q R X d S )Nz:memory:)r r r r r )r
r r r r � test_simple5 s zBackupTests.test_simplec sz g � � fdd�}t �d��"}| jj|d|d� | �|� W d Q R X | �t� �d� | �� d t j� | �� d t j� d S )Nc s � � | � d S )N)�append)�status� remaining�total)�journalr r �progress= s z+BackupTests.test_progress.<locals>.progressz:memory:r )�pagesr/ � r ) r r r r r r �lenZ SQLITE_OKZ SQLITE_DONE)r
r/ r r )r. r �
test_progress: s zBackupTests.test_progressc sd g � � fdd�}t �d�� }| jj||d� | �|� W d Q R X | �t� �d� | �� d d� d S )Nc s � � |� d S )N)r* )r+ r, r- )r. r r r/ K s z?BackupTests.test_progress_all_pages_at_once_1.<locals>.progressz:memory:)r/ r r )r r r r r r r2 )r
r/ r r )r. r �!test_progress_all_pages_at_once_1H s z-BackupTests.test_progress_all_pages_at_once_1c sf g � � fdd�}t �d��"}| jj|d|d� | �|� W d Q R X | �t� �d� | �� d d� d S )Nc s � � |� d S )N)r* )r+ r, r- )r. r r r/ X s z?BackupTests.test_progress_all_pages_at_once_2.<locals>.progressz:memory:���)r0 r/ r r )r r r r r r r2 )r
r/ r r )r. r �!test_progress_all_pages_at_once_2U s z-BackupTests.test_progress_all_pages_at_once_2c
C sT | � t��.}t�d��}| jj|ddd� W d Q R X W d Q R X | �t|j�d� d S )Nz:memory:r Zbar)r0 r/ z$progress argument must be a callable) r r r r r r r r$ r% )r
r& r r r r �test_non_callable_progressb s &z&BackupTests.test_non_callable_progressc s� g � � �fdd�}t �d��D}�jj|d|d� ��|� |�d��� }��|d d d� W d Q R X ��t� �d � ��� d d� ��� d d� ��� d
d� d S ) Nc s0 � s"�j �d|d f� �j �� � �|� d S )Nz INSERT INTO foo (key) VALUES (?)i� )r r
r r* )r+ r, r- )r. r
r r r/ k s
z5BackupTests.test_modifying_progress.<locals>.progressz:memory:r )r0 r/ z2SELECT key FROM foo WHERE key >= 1000 ORDER BY keyr i� r r1 ) r r r r r r
r r r2 )r
r/ r r r )r. r
r �test_modifying_progressh s
z#BackupTests.test_modifying_progressc
C sZ dd� }| � t��,}t�d��}| jj||d� W d Q R X W d Q R X | �t|j�d� d S )Nc S s t d��d S )Nznearly out of space)� SystemError)r+ r, r- r r r r/ � s z3BackupTests.test_failing_progress.<locals>.progressz:memory:)r/ znearly out of space) r r9 r r r r r r$ r% )r
r/ �errr r r r �test_failing_progress s
$z!BackupTests.test_failing_progressc
C s t �d��}| jj|dd� W d Q R X t �d��}| jj|dd� W d Q R X | �t j��,}t �d��}| jj|dd� W d Q R X W d Q R X | �t|j�ddg� | j� d� | j� d � | j�
d
d d g� | j� � t �d�� }| jj|d
d� | � |� W d Q R X d S )Nz:memory:�main)�nameZtempz non-existingzSQL logic errorz#SQL logic error or missing 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 )r
r r& 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' r( r) r3 r4 r6 r7 r8 r; r> r r r r r s"
r c C s
t �t�S )N)�unittestZ makeSuiter r r r r �suite� s rC �__main__)
Zsqlite3r rB ZskipIfr# ZTestCaser rC r? r< r r r r �<module> s