f
��_� � @ sx d dl Z d dlZd dlZdd� ZG dd� dej�ZG dd� dej�ZG dd � d ej�Zd
d � Z d d
� Z
e dkrte
� dS )� Nc C s dS )NZ
sqlite_testdb� r r r �2/usr/lib64/python3.10/sqlite3/test/transactions.py� get_db_path s r c @ sd e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dS )�TransactionTestsc C sd zt �t� � W n ty" Y n0 tjt� dd�| _| j�� | _tjt� dd�| _ | j �� | _
d S )Ng�������?)�timeout) �os�remover �OSError�sqlite�connect�con1�cursor�cur1�con2�cur2��selfr r r �setUp s zTransactionTests.setUpc C sP | j �� | j�� | j�� | j�� zt�t� � W n tyJ Y n0 d S �N) r �closer r r r �unlinkr r r r r r �tearDown+ s
zTransactionTests.tearDownc C sN | j �d� | j �d� | j �d� | j�d� | j�� }| �t|�d� d S )N�create table test(i)� insert into test(i) values (5)zcreate table test2(j)�select i from testr �r �executer �fetchall� assertEqual�len�r �resr r r � CheckDMLDoesNotAutoCommitBefore7 s
z0TransactionTests.CheckDMLDoesNotAutoCommitBeforec C sB | j �d� | j �d� | j�d� | j�� }| �t|�d� d S )Nr r r r r r r r r � CheckInsertStartsTransaction? s
z-TransactionTests.CheckInsertStartsTransactionc C sX | j �d� | j �d� | j�� | j �d� | j�d� | j�� d }| �|d� d S )Nr r zupdate test set i=6r r � )r r r �commitr Zfetchoner r r r r � CheckUpdateStartsTransactionF s
z-TransactionTests.CheckUpdateStartsTransactionc C sX | j �d� | j �d� | j�� | j �d� | j�d� | j�� }| �t|�d� d S )Nr r zdelete from testr � �r r r r% r r r r r r r r � CheckDeleteStartsTransactionO s
z-TransactionTests.CheckDeleteStartsTransactionc C sl | j �d� | j �d� | j�� | j �d� | j�d� | j�� }| �t|�d� | �|d d d� d S )Nr r z replace into test(i) values (6)r r' r r$ r( r r r r � CheckReplaceStartsTransactionX s
z.TransactionTests.CheckReplaceStartsTransactionc C s� | j �d� | j �d� d | j_| �| jjd � | j�d� | j�� }| �t|�d� d| j_| �| jjd� | j �d� | j�d� | j�� }| �t|�d� d S )Nr r r r' ZDEFERRED)r r r Zisolation_levelr r r r r r r r �CheckToggleAutoCommitb s
z&TransactionTests.CheckToggleAutoCommitc C sT | j �d� | j �d� | �tj�� | j�d� W d � n1 sF0 Y d S )Nr r )r r � assertRaisesr
�OperationalErrorr r r r r �CheckRaiseTimeoutr s z"TransactionTests.CheckRaiseTimeoutc C s^ | j �d� | j �d� | �tj�� | j�d� W d � n1 sF0 Y | j�� dS )z�
This tests the improved concurrency with pysqlite 2.3.4. You needed
to roll back con2 before you could commit con1.
r r N)r r r, r
r- r r r% r r r r � CheckLockingx s
*z TransactionTests.CheckLockingc C sp t �d�}|�� }|�d� |�d� |�d� |�� | �t j�� |�� W d � n1 sb0 Y dS )z�
Checks if cursors on the connection are set into a "reset" state
when a rollback is done on the connection.
�:memory:zcreate table test(x)z insert into test(x) values (5)z&select 1 union select 2 union select 3N)r
r r
r �rollbackr, ZInterfaceErrorr )r �con�curr r r � CheckRollbackCursorConsistency� s
z/TransactionTests.CheckRollbackCursorConsistencyN)�__name__�
__module__� __qualname__r r r" r# r& r) r* r+ r. r/ r4 r r r r r s
r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�SpecialCommandTestsc C s t �d�| _| j�� | _d S �Nr0 )r
r r2 r
r3 r r r r r � s zSpecialCommandTests.setUpc C s( | j �d� | j �d� | j �d� d S )Nr r zdrop table test�r3 r r r r r �CheckDropTable� s z"SpecialCommandTests.CheckDropTablec C s( | j �d� | j �d� | j �d� d S )Nr r zpragma count_changes=1r: r r r r � CheckPragma� s z SpecialCommandTests.CheckPragmac C s | j �� | j�� d S r )r3 r r2 r r r r r � s
z SpecialCommandTests.tearDownN)r5 r6 r7 r r; r< r r r r r r8 � s r8 c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �TransactionalDDLc C s t �d�| _d S r9 )r
r r2 r r r r r � s zTransactionalDDL.setUpc C s6 | j �d� | j �� | j �d��� }| �|g � d S )Nr �select * from test)r2 r r1 r r )r �resultr r r �#CheckDdlDoesNotAutostartTransaction� s
z4TransactionalDDL.CheckDdlDoesNotAutostartTransactionc C s^ | j �d� | j �d� | j �� | �tj�� | j �d� W d � n1 sP0 Y d S )Nzbegin immediater r> �r2 r r1 r, r
r- r r r r � CheckImmediateTransactionalDDL� s
z/TransactionalDDL.CheckImmediateTransactionalDDLc C s^ | j �d� | j �d� | j �� | �tj�� | j �d� W d � n1 sP0 Y d S )NZbeginr r> rA r r r r �CheckTransactionalDDL� s
z&TransactionalDDL.CheckTransactionalDDLc C s | j �� d S r )r2 r r r r r r � s zTransactionalDDL.tearDownN)r5 r6 r7 r r@ rB rC r r r r r r= � s
r= c C s4 t �td�} t �td�}t �td�}t �| ||f�S )NZCheck)�unittestZ makeSuiter r8 r= Z TestSuite)Z
default_suiteZspecial_command_suiteZ ddl_suiter r r �suite� s rE c C s t �� } | �t� � d S r )rD ZTextTestRunner�runrE )Zrunnerr r r �test� s rG �__main__) r rD Zsqlite3r
r ZTestCaser r8 r= rE rG r5 r r r r �<module> s u!