EVOLUTION-NINJA
Edit File: dbapi.cpython-38.pyc
U �i�_(� � @ s� d dl Z d dlZd dlZd dlmZmZ G dd� dej�ZG dd� dej�Z G dd� dej�Z G d d � d ej�Z G d d � d ej�Z G d d� dej�Z G dd� dej�ZG dd� dej�ZG dd� dej�Zdd� Zdd� Zedkr�e� dS )� N)�TESTFN�unlinkc @ st e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� ZdS ) � ModuleTestsc C s | � tjddtj � d S )Nz2.0z apilevel is %s, should be 2.0)� assertEqual�sqliteZapilevel��self� r �1/opt/python38/lib/python3.8/sqlite3/test/dbapi.py� CheckAPILevel s �zModuleTests.CheckAPILevelc C s | � tjddtj � d S )N� z threadsafety is %d, should be 1)r r Z threadsafetyr r r r �CheckThreadSafety$ s �z ModuleTests.CheckThreadSafetyc C s | � tjddtj � d S )NZqmarkz%paramstyle is '%s', should be 'qmark')r r Z paramstyler r r r �CheckParamStyle( s ��zModuleTests.CheckParamStylec C s | � ttjt�d� d S )Nz&Warning is not a subclass of Exception)� assertTrue� issubclassr �Warning� Exceptionr r r r � CheckWarning- s �zModuleTests.CheckWarningc C s | � ttjt�d� d S )Nz$Error is not a subclass of Exception)r r r �Errorr r r r r � CheckError1 s �zModuleTests.CheckErrorc C s | � ttjtj�d� d S )Nz)InterfaceError is not a subclass of Error)r r r �InterfaceErrorr r r r r �CheckInterfaceError5 s �z ModuleTests.CheckInterfaceErrorc C s | � ttjtj�d� d S )Nz(DatabaseError is not a subclass of Error)r r r � DatabaseErrorr r r r r �CheckDatabaseError9 s �z ModuleTests.CheckDatabaseErrorc C s | � ttjtj�d� d S )Nz,DataError is not a subclass of DatabaseError)r r r � DataErrorr r r r r �CheckDataError= s �zModuleTests.CheckDataErrorc C s | � ttjtj�d� d S )Nz3OperationalError is not a subclass of DatabaseError)r r r �OperationalErrorr r r r r �CheckOperationalErrorA s �z!ModuleTests.CheckOperationalErrorc C s | � ttjtj�d� d S )Nz1IntegrityError is not a subclass of DatabaseError)r r r �IntegrityErrorr r r r r �CheckIntegrityErrorE s �z ModuleTests.CheckIntegrityErrorc C s | � ttjtj�d� d S )Nz0InternalError is not a subclass of DatabaseError)r r r � InternalErrorr r r r r �CheckInternalErrorI s �z ModuleTests.CheckInternalErrorc C s | � ttjtj�d� d S )Nz3ProgrammingError is not a subclass of DatabaseError)r r r �ProgrammingErrorr r r r r �CheckProgrammingErrorM s �z!ModuleTests.CheckProgrammingErrorc C s | � ttjtj�d� d S )Nz4NotSupportedError is not a subclass of DatabaseError)r r r �NotSupportedErrorr r r r r �CheckNotSupportedErrorQ s ��z"ModuleTests.CheckNotSupportedErrorN)�__name__� __module__� __qualname__r r r r r r r r r r r! r# r% r r r r r s r c @ 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 � Ze�ejd kd �d d � �Zd!S )"�ConnectionTestsc C s0 t �d�| _| j�� }|�d� |�dd� d S )N�:memory:z4create table test(id integer primary key, name text)�!insert into test(name) values (?)��foo)r �connect�cx�cursor�execute�r �cur r r �setUpX s zConnectionTests.setUpc C s | j �� d S �N�r/ �closer r r r �tearDown^ s zConnectionTests.tearDownc C s | j �� d S r5 �r/ �commitr r r r � CheckCommita s zConnectionTests.CheckCommitc C s | j �� | j �� dS )zV A commit should also work when no changes were made to the database. Nr9 r r r r �CheckCommitAfterNoChangesd s z)ConnectionTests.CheckCommitAfterNoChangesc C s | j �� d S r5 �r/ �rollbackr r r r � CheckRollbackk s z ConnectionTests.CheckRollbackc C s | j �� | j �� dS )zX A rollback should also work when no changes were made to the database. Nr= r r r r �CheckRollbackAfterNoChangesn s z+ConnectionTests.CheckRollbackAfterNoChangesc C s | j �� }d S r5 )r/ r0 r2 r r r � CheckCursoru s zConnectionTests.CheckCursorc C s* d}| � tj�� t�|�}W 5 Q R X d S )Nz/foo/bar/bla/23534/mydb.db)� assertRaisesr r r. )r ZYOU_CANNOT_OPEN_THIS�conr r r �CheckFailedOpenx s z ConnectionTests.CheckFailedOpenc C s | j �� d S r5 r6 r r r r � CheckClose} s zConnectionTests.CheckClosec C s� | � | jjtj� | � | jjtj� | � | jjtj� | � | jjtj� | � | jjtj� | � | jjtj� | � | jj tj � | � | jj tj � | � | jj tj � | � | jj tj � d S r5 ) r r/ r r r r r r r r r r" r$ r r r r �CheckExceptions� s z ConnectionTests.CheckExceptionsc C s� t �d�}|�� }| �|jd� |�d� | �|jd� |�dd� | �|jd� |�ddg� |�� }| �|jd� |�� | �|jd� |�ddg� |�� }| �|jd� d S ) Nr* Fz?create table transactiontest(id integer primary key, name text)z,insert into transactiontest(name) values (?)r, Tz-select name from transactiontest where name=?r- )r r. r0 r �in_transactionr1 �fetchoner: )r r/ r3 �rowr r r �CheckInTransaction� s z"ConnectionTests.CheckInTransactionc C s" | � t�� d| j_W 5 Q R X d S )NT)rB �AttributeErrorr/ rG r r r r �CheckInTransactionRO� s z$ConnectionTests.CheckInTransactionROc C sD | � tt� G dd� d�}|� }t�|��}|�d� W 5 Q R X dS )z| Checks that we can successfully connect to a database using an object that is PathLike, i.e. has __fspath__(). c @ s e Zd Zdd� ZdS )z9ConnectionTests.CheckOpenWithPathLikeObject.<locals>.Pathc S s t S r5 )r r r r r � __fspath__� s zDConnectionTests.CheckOpenWithPathLikeObject.<locals>.Path.__fspath__N)r& r' r( rM r r r r �Path� s rN � create table test(id integer)N)� addCleanupr r r r. r1 )r rN �pathr/ r r r �CheckOpenWithPathLikeObject� s z+ConnectionTests.CheckOpenWithPathLikeObjectc C s� t jdk r4| �t j�� t jddd� W 5 Q R X d S | �tt� t �t��}|�d� W 5 Q R X t jdt dd��}|�d� W 5 Q R X t jdt d dd��(}| �t j �� |�d � W 5 Q R X W 5 Q R X d S ) N)� � rT r* T)ZurirO zfile:z insert into test(id) values(0)z?mode=roz insert into test(id) values(1)) r �sqlite_version_inforB r$ r. rP r r r1 r )r r/ r r r � CheckOpenUri� s z ConnectionTests.CheckOpenUri)rS rS r z&needs sqlite versions older than 3.3.1c C s< | � tj��}tjddd� W 5 Q R X | �t|j�d� d S )Nr* F)Zcheck_same_threadz shared connections not available)rB r r$ r. r �str� exception�r �cmr r r � CheckSameThreadErrorOnOldVersion� s z0ConnectionTests.CheckSameThreadErrorOnOldVersionN)r& r' r( r4 r8 r; r< r? r@ rA rD rE rF rJ rL rR rV �unittestZskipIfr rU r[ r r r r r) V s$ �r) c @ 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%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d9d:� Z d;d<� Z d=d>� Z!d?d@� Z"dAdB� Z#dCdD� Z$dEdF� Z%dGdH� Z&dIdJ� Z'dKdL� Z(dMdN� Z)dOdP� Z*dQdR� Z+dSdT� Z,dUdV� Z-dWdX� Z.dYdZ� Z/d[d\� Z0d]d^� Z1d_d`� Z2dadb� Z3dcdd� Z4deS )f� CursorTestsc C s6 t �d�| _| j�� | _| j�d� | j�dd� d S )Nr* z\create table test(id integer primary key, name text, income number, unique_test text unique)r+ r, �r r. r/ r0 r3 r1 r r r r r4 � s �zCursorTests.setUpc C s | j �� | j�� d S r5 �r3 r7 r/ r r r r r8 � s zCursorTests.tearDownc C s | j �d� d S )N�delete from test�r3 r1 r r r r �CheckExecuteNoArgs� s z CursorTests.CheckExecuteNoArgsc C s( | � tj�� | j�d� W 5 Q R X d S )Nz select asdf)rB r r r3 r1 r r r r �CheckExecuteIllegalSql� s z"CursorTests.CheckExecuteIllegalSqlc C s( | � tj�� | j�d� W 5 Q R X d S )Nzselect 5+4; select 4+5)rB r r r3 r1 r r r r �CheckExecuteTooMuchSql� s z"CursorTests.CheckExecuteTooMuchSqlc C s | j �d� d S )Nzselect 5+4; -- foo barra r r r r �CheckExecuteTooMuchSql2� s z#CursorTests.CheckExecuteTooMuchSql2c C s | j �d� d S )NzT select 5+4; /* foo */ ra r r r r �CheckExecuteTooMuchSql3� s z#CursorTests.CheckExecuteTooMuchSql3c C s& | � t�� | j�d� W 5 Q R X d S �N�* )rB � ValueErrorr3 r1 r r r r �CheckExecuteWrongSqlArg� s z#CursorTests.CheckExecuteWrongSqlArgc C s | j �dd� d S )N� insert into test(id) values (?))rh ra r r r r �CheckExecuteArgInt� s z CursorTests.CheckExecuteArgIntc C s | j �dd� d S )N�#insert into test(income) values (?))gq= ף��@ra r r r r �CheckExecuteArgFloat� s z CursorTests.CheckExecuteArgFloatc C s | j �dd� d S )Nr+ )ZHugora r r r r �CheckExecuteArgString� s z!CursorTests.CheckExecuteArgStringc C s@ | j �dd� | j �d| j jf� | j �� }| �|d d� d S )Nr+ )�Hu goz select name from test where id=?r rp )r3 r1 � lastrowidrH r �r rI r r r �!CheckExecuteArgStringWithZeroByte� s z-CursorTests.CheckExecuteArgStringWithZeroBytec C s: | � t��}| j�dd� W 5 Q R X | �t|j�d� d S )Nrk rh z"parameters are of unsupported type)rB ri r3 r1 r rW rX rY r r r �CheckExecuteNonIterable� s z#CursorTests.CheckExecuteNonIterablec C s* | � tj�� | j�dd� W 5 Q R X d S )Nrk )� ZEgon�rB r r" r3 r1 r r r r �CheckExecuteWrongNoOfArgs1 s z&CursorTests.CheckExecuteWrongNoOfArgs1c C s( | � tj�� | j�d� W 5 Q R X d S �Nrk rv r r r r �CheckExecuteWrongNoOfArgs2 s z&CursorTests.CheckExecuteWrongNoOfArgs2c C s( | � tj�� | j�d� W 5 Q R X d S rx rv r r r r �CheckExecuteWrongNoOfArgs3 s z&CursorTests.CheckExecuteWrongNoOfArgs3c C s: | j �d� | j �ddg� | j �� }| �|d d� d S )N�%insert into test(name) values ('foo')�"select name from test where name=?r- r �r3 r1 rH r rr r r r �CheckExecuteParamList s z!CursorTests.CheckExecuteParamListc C sJ G dd� dt �}| j�d� | j�d|� � | j�� }| �|d d� d S )Nc @ s e Zd Zdd� Zdd� ZdS )z0CursorTests.CheckExecuteParamSequence.<locals>.Lc S s dS )Nr r r r r r �__len__ s z8CursorTests.CheckExecuteParamSequence.<locals>.L.__len__c S s |dks t �dS )Nr r- )�AssertionError�r �xr r r � __getitem__ s z<CursorTests.CheckExecuteParamSequence.<locals>.L.__getitem__N)r& r' r( r r� r r r r �L s r� r{ r| r r- )�objectr3 r1 rH r )r r� rI r r r �CheckExecuteParamSequence s z%CursorTests.CheckExecuteParamSequencec C s< | j �d� | j �dddi� | j �� }| �|d d� d S )Nr{ �&select name from test where name=:name�namer- r r} rr r r r �CheckExecuteDictMapping# s z#CursorTests.CheckExecuteDictMappingc C sJ G dd� dt �}| j�d� | j�d|� � | j�� }| �|d d� d S )Nc @ s e Zd Zdd� ZdS )z6CursorTests.CheckExecuteDictMapping_Mapping.<locals>.Dc S s dS �Nr- r )r �keyr r r � __missing__+ s zBCursorTests.CheckExecuteDictMapping_Mapping.<locals>.D.__missing__N)r& r' r( r� r r r r �D* s r� r{ r� r r- )�dictr3 r1 rH r )r r� rI r r r � CheckExecuteDictMapping_Mapping) s z+CursorTests.CheckExecuteDictMapping_Mappingc C s: | j �d� | �tj�� | j �dddi� W 5 Q R X d S )Nr{ z1select name from test where name=:name and id=:idr� r- �r3 r1 rB r r" r r r r �$CheckExecuteDictMappingTooLittleArgs3 s z0CursorTests.CheckExecuteDictMappingTooLittleArgsc C s4 | j �d� | �tj�� | j �d� W 5 Q R X d S )Nr{ r� r� r r r r � CheckExecuteDictMappingNoArgs8 s z)CursorTests.CheckExecuteDictMappingNoArgsc C s: | j �d� | �tj�� | j �dddi� W 5 Q R X d S )Nr{ r| r� r- r� r r r r � CheckExecuteDictMappingUnnamed= s z*CursorTests.CheckExecuteDictMappingUnnamedc C s | j �� d S r5 )r3 r7 r r r r rE B s zCursorTests.CheckClosec C sD | j �d� | j �d� | j �d� | j �d� | �| j jd� d S )Nr` r{ zupdate test set name='bar'� �r3 r1 r �rowcountr r r r �CheckRowcountExecuteE s z CursorTests.CheckRowcountExecutec C s | j �d� | �| j jd� dS )z� pysqlite does not know the rowcount of SELECT statements, because we don't fetch all rows after executing the select statement. The rowcount has thus to be -1. zselect 5 union select 6���Nr� r r r r �CheckRowcountSelectL s z CursorTests.CheckRowcountSelectc C s4 | j �d� | j �ddddg� | �| j jd� d S )Nr` r+ )r )r� �rS rS )r3 r1 � executemanyr r� r r r r �CheckRowcountExecutemanyU s z$CursorTests.CheckRowcountExecutemanyc C s0 | j �d� | j �d� | jd| jjdd� d S )Nr{ r� z"total changes reported wrong value)�msg)r3 r1 Z assertLessr/ Z total_changesr r r r �CheckTotalChangesZ s z CursorTests.CheckTotalChangesc C s" | j �ddd� tdd�D �� d S )Nrm c S s g | ] }|f�qS r r )�.0r� r r r � <listcomp>d s z8CursorTests.CheckExecuteManySequence.<locals>.<listcomp>�d �n )r3 r� �ranger r r r �CheckExecuteManySequencec s z$CursorTests.CheckExecuteManySequencec C s" G dd� d�}| j �d|� � d S )Nc @ s e Zd Zdd� Zdd� ZdS )z4CursorTests.CheckExecuteManyIterator.<locals>.MyIterc S s d| _ d S �N� )�valuer r r r �__init__h s z=CursorTests.CheckExecuteManyIterator.<locals>.MyIter.__init__c S s* | j dkrt�n| j d7 _ | j fS d S )N� r )r� � StopIterationr r r r �__next__k s z=CursorTests.CheckExecuteManyIterator.<locals>.MyIter.__next__N)r& r' r( r� r� r r r r �MyIterg s r� rm �r3 r� )r r� r r r �CheckExecuteManyIteratorf s z$CursorTests.CheckExecuteManyIteratorc C s dd� }| j �d|� � d S )Nc s s t d�D ] } | fV qd S r� )r� )�ir r r �mygenu s z4CursorTests.CheckExecuteManyGenerator.<locals>.mygenrm r� )r r� r r r �CheckExecuteManyGeneratort s z%CursorTests.CheckExecuteManyGeneratorc C s* | � t�� | j�ddg� W 5 Q R X d S )Nrh r� )rB ri r3 r� r r r r �CheckExecuteManyWrongSqlArg{ s z'CursorTests.CheckExecuteManyWrongSqlArgc C s, | � tj�� | j�ddg� W 5 Q R X d S )Nzselect ?r� )rB r r" r3 r� r r r r �CheckExecuteManySelect s z"CursorTests.CheckExecuteManySelectc C s( | � t�� | j�dd� W 5 Q R X d S )Nrm rh )rB � TypeErrorr3 r� r r r r �CheckExecuteManyNotIterable� s z'CursorTests.CheckExecuteManyNotIterablec C sv | j �d� | j �dd� | j �dd� | j �d� g }| j D ]}|�|d � q>| �|d d� | �|d d � d S ) Nr` rk )r� )� z select id from test order by idr r� r r� )r3 r1 �appendr )r ZlstrI r r r �CheckFetchIter� s zCursorTests.CheckFetchIterc C s@ | j �d� | j �� }| �|d d� | j �� }| �|d � d S )N�select name from testr r- r} rr r r r � CheckFetchone� s zCursorTests.CheckFetchonec C s" | j �� }|�� }| �|d � d S r5 )r/ r0 rH r )r �currI r r r �CheckFetchoneNoStatement� s z$CursorTests.CheckFetchoneNoStatementc C sr | � | jjd� d| j_| j�d� | j�d� | j�d� | j�d� | j�d� | j�� }| � t|�d� d S )Nr r� r` z#insert into test(name) values ('A')z#insert into test(name) values ('B')z#insert into test(name) values ('C')r� )r r3 Z arraysizer1 � fetchmany�len�r �resr r r �CheckArraySize� s zCursorTests.CheckArraySizec C sD | j �d� | j �d�}| �t|�d� | j �d�}| �|g � d S )Nr� r� r �r3 r1 r� r r� r� r r r �CheckFetchmany� s zCursorTests.CheckFetchmanyc C s. | j �d� | j jdd�}| �t|�d� dS )z0Checks if fetchmany works with keyword argumentsr� r� )�sizer Nr� r� r r r �CheckFetchmanyKwArg� s z CursorTests.CheckFetchmanyKwArgc C s@ | j �d� | j �� }| �t|�d� | j �� }| �|g � d S )Nr� r )r3 r1 �fetchallr r� r� r r r � CheckFetchall� s zCursorTests.CheckFetchallc C s | j �dddg� d S )NrS � r� )r3 Z setinputsizesr r r r �CheckSetinputsizes� s z CursorTests.CheckSetinputsizesc C s | j �dd� d S )Nr� r �r3 Z setoutputsizer r r r �CheckSetoutputsize� s z CursorTests.CheckSetoutputsizec C s | j �d� d S rg r� r r r r �CheckSetoutputsizeNoColumn� s z&CursorTests.CheckSetoutputsizeNoColumnc C s | � | jj| j� d S r5 )r r3 � connectionr/ r r r r �CheckCursorConnection� s z!CursorTests.CheckCursorConnectionc C s. | � t�� dd� }| j�|�}W 5 Q R X d S )Nc S s d S r5 r r r r r �f� � z/CursorTests.CheckWrongCursorCallable.<locals>.f)rB r� r/ r0 )r r� r� r r r �CheckWrongCursorCallable� s z$CursorTests.CheckWrongCursorCallablec C s8 G dd� d�}|� }| � t�� t�|�}W 5 Q R X d S )Nc @ s e Zd ZdS )z.CursorTests.CheckCursorWrongClass.<locals>.FooN)r& r' r( r r r r �Foo� s r� )rB r� r ZCursor)r r� r- r� r r r �CheckCursorWrongClass� s z!CursorTests.CheckCursorWrongClassc C sN d}dD ]@}| j |d��* | j�|�|�d� | �| jjd� W 5 Q R X qdS )zV INSERT OR REPLACE and REPLACE INTO should produce the same behavior. z+{} INTO test(id, unique_test) VALUES (?, ?))zINSERT OR REPLACEZREPLACE�� statement)r r- r N)�subTestr3 r1 �formatr rq )r �sqlr� r r r �CheckLastRowIDOnReplace� s z#CursorTests.CheckLastRowIDOnReplacec C s@ | j �dd� | �| j jd� | j �dd� | �| j jd� d S )Nz2insert or ignore into test(unique_test) values (?))�testr� )r3 r1 r rq r r r r �CheckLastRowIDOnIgnore� s ��z"CursorTests.CheckLastRowIDOnIgnorec C s� g }dD ]�}d}| j d�|�d��n | j�|�|�|f� |�|| jjf� | �tj�� | j�|�|�|f� W 5 Q R X |�|| jjf� W 5 Q R X qddddddg}| � ||� d S )N)�FAIL�ABORT�ROLLBACKz.INSERT OR {} INTO test(unique_test) VALUES (?)z INSERT OR {}r� )r� r� )r� rS )r� r� ) r� r� r3 r1 r� rq rB r r r )r �resultsr� r� Zexpectedr r r �CheckLastRowIDInsertOR� s" �z"CursorTests.CheckLastRowIDInsertORN)5r& r' r( r4 r8 rb rc rd re rf rj rl rn ro rs rt rw ry rz r~ r� r� r� r� r� r� rE r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r] � sd r] c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dS )� ThreadTestsc C s( t �d�| _| j�� | _| j�d� d S )Nr* z\create table test(id integer primary key, name text, bin binary, ratio number, ts timestamp))r r. rC r0 r� r1 r r r r r4 s zThreadTests.setUpc C s | j �� | j�� d S r5 )r� r7 rC r r r r r8 s zThreadTests.tearDownc C sR dd� }g }t j|| j|d�d�}|�� |�� t|�dkrN| �d�|�� d S )Nc S sJ z| � � }|�d� W d S tjk r0 Y d S |�d� Y nX d S �N� did not raise ProgrammingError�raised wrong exception)r0 r� r r" )rC �errorsr� r r r �run s z'ThreadTests.CheckConCursor.<locals>.run�rC r� ��target�kwargsr � �� threading�ThreadrC �start�joinr� �fail�r r� r� �tr r r �CheckConCursor s zThreadTests.CheckConCursorc C sR dd� }g }t j|| j|d�d�}|�� |�� t|�dkrN| �d�|�� d S )Nc S sJ z| � � |�d� W d S tjk r0 Y d S |�d� Y nX d S r� )r: r� r r" r� r r r r� s z'ThreadTests.CheckConCommit.<locals>.runr� r� r r� r� r� r r r �CheckConCommit s zThreadTests.CheckConCommitc C sR dd� }g }t j|| j|d�d�}|�� |�� t|�dkrN| �d�|�� d S )Nc S sJ z| � � |�d� W d S tjk r0 Y d S |�d� Y nX d S r� )r> r� r r" r� r r r r� 1 s z)ThreadTests.CheckConRollback.<locals>.runr� r� r r� r� r� r r r �CheckConRollback0 s z ThreadTests.CheckConRollbackc C sR dd� }g }t j|| j|d�d�}|�� |�� t|�dkrN| �d�|�� d S )Nc S sJ z| � � |�d� W d S tjk r0 Y d S |�d� Y nX d S r� �r7 r� r r" r� r r r r� C s z&ThreadTests.CheckConClose.<locals>.runr� r� r r� r� r� r r r � CheckConCloseB s zThreadTests.CheckConClosec C sR dd� }g }t j|| j|d�d�}|�� |�� t|�dkrN| �d�|�� d S )Nc S sL z| � d� |�d� W d S tjk r2 Y d S |�d� Y nX d S )N�#insert into test(name) values ('a')r� r� �r1 r� r r" �r� r� r r r r� U s z.ThreadTests.CheckCurImplicitBegin.<locals>.runr� r� r r� �r� r� r� r� r� r� r� r� r r r �CheckCurImplicitBeginT s z!ThreadTests.CheckCurImplicitBeginc C sR dd� }g }t j|| j|d�d�}|�� |�� t|�dkrN| �d�|�� d S )Nc S sJ z| � � |�d� W d S tjk r0 Y d S |�d� Y nX d S r� r� r� r r r r� g s z&ThreadTests.CheckCurClose.<locals>.runr� r� r r� r� r� r r r � CheckCurClosef s zThreadTests.CheckCurClosec C s^ dd� }g }| j �d� tj|| j |d�d�}|�� |�� t|�dkrZ| �d�|�� d S )Nc S sL z| � d� |�d� W d S tjk r2 Y d S |�d� Y nX d S )Nr� r� r� r� r� r r r r� y s z(ThreadTests.CheckCurExecute.<locals>.runr� r� r� r r� �r� r1 r� r� r� r� r� r� r� r r r �CheckCurExecutex s zThreadTests.CheckCurExecutec C sj dd� }g }| j �d� | j �d� tj|| j |d�d�}|�� |�� t|�dkrf| �d�|�� d S ) Nc S sJ z| � � }|�d� W d S tjk r0 Y d S |�d� Y nX d S r� )rH r� r r" )r� r� rI r r r r� � s z)ThreadTests.CheckCurIterNext.<locals>.runr� r� r� r� r r� r� r� r r r �CheckCurIterNext� s z ThreadTests.CheckCurIterNextN) r& r' r( r4 r8 r� r� r� r� r� r� r r r r r r r� s r� c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )�ConstructorTestsc C s t �ddd�}d S )N�� r� � )r ZDate�r �dr r r � CheckDate� s zConstructorTests.CheckDatec C s t �ddd�}d S )N� �' �# )r ZTime�r r� r r r � CheckTime� s zConstructorTests.CheckTimec C s t �dddddd�}d S )Nr r� r r r r )r Z Timestamp�r �tsr r r �CheckTimestamp� s z ConstructorTests.CheckTimestampc C s t �d�}d S rg )r Z DateFromTicksr r r r �CheckDateFromTicks� s z#ConstructorTests.CheckDateFromTicksc C s t �d�}d S rg )r Z TimeFromTicksr r r r �CheckTimeFromTicks� s z#ConstructorTests.CheckTimeFromTicksc C s t �d�}d S rg )r ZTimestampFromTicksr r r r �CheckTimestampFromTicks� s z(ConstructorTests.CheckTimestampFromTicksc C s t �d�}d S )Ns ')r ZBinary)r �br r r � CheckBinary� s z ConstructorTests.CheckBinaryN) r& r' r( r r r r r r r r r r r r � s r c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )�ExtensionTestsc C sB t �d�}|�� }|�d� |�d� |�� d }| �|d� d S )Nr* z� -- bla bla /* a stupid comment */ create table a(i); insert into a(i) values (5); zselect i from ar r� )r r. r0 � executescriptr1 rH r )r rC r� r� r r r �CheckScriptStringSql� s z#ExtensionTests.CheckScriptStringSqlc C s8 t �d�}|�� }| �t j�� |�d� W 5 Q R X d S )Nr* z1create table test(x); asdf; create table test2(x)�r r. r0 rB r r �r rC r� r r r �CheckScriptSyntaxError� s z%ExtensionTests.CheckScriptSyntaxErrorc C s8 t �d�}|�� }| �t j�� |�d� W 5 Q R X d S )Nr* z5create table test(sadfsadfdsa); select foo from hurz;r r r r r �CheckScriptErrorNormal� s z%ExtensionTests.CheckScriptErrorNormalc C sH t �d�}|�� }| �t��}|�d� W 5 Q R X | �t|j�d� d S )Nr* s9 create table test(foo); insert into test(foo) values (5);z script argument must be unicode.) r r. r0 rB ri r r rW rX )r rC r� rZ r r r � CheckCursorExecutescriptAsBytes� s z.ExtensionTests.CheckCursorExecutescriptAsBytesc C s. t �d�}|�d��� d }| �|dd� d S )Nr* zselect 5r r� z Basic test of Connection.execute)r r. r1 rH r �r rC �resultr r r �CheckConnectionExecute� s z%ExtensionTests.CheckConnectionExecutec C sb t �d�}|�d� |�dddg� |�d��� }| �|d d dd � | �|d d d d � d S ) Nr* zcreate table test(foo)z insert into test(foo) values (?)r� �r� z!select foo from test order by foor rS z$Basic test of Connection.executemanyr r� )r r. r1 r� r� r r r r r �CheckConnectionExecutemany� s z)ExtensionTests.CheckConnectionExecutemanyc C s8 t �d�}|�d� |�d��� d }| �|dd� d S )Nr* z9create table test(foo); insert into test(foo) values (5);zselect foo from testr r� z&Basic test of Connection.executescript)r r. r r1 rH r r r r r � CheckConnectionExecutescript� s z+ExtensionTests.CheckConnectionExecutescriptN) r& r' r( r r r r r r! r" r r r r r � s r c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dS )�ClosedConTestsc C s6 t �d�}|�� | �t j�� |�� }W 5 Q R X d S �Nr* )r r. r7 rB r" r0 r r r r �CheckClosedConCursor� s z#ClosedConTests.CheckClosedConCursorc C s6 t �d�}|�� | �t j�� |�� W 5 Q R X d S r$ )r r. r7 rB r" r: �r rC r r r �CheckClosedConCommit� s z#ClosedConTests.CheckClosedConCommitc C s6 t �d�}|�� | �t j�� |�� W 5 Q R X d S r$ )r r. r7 rB r" r> r& r r r �CheckClosedConRollback� s z%ClosedConTests.CheckClosedConRollbackc C s@ t �d�}|�� }|�� | �t j�� |�d� W 5 Q R X d S )Nr* zselect 4)r r. r0 r7 rB r" r1 r r r r �CheckClosedCurExecute� s z$ClosedConTests.CheckClosedCurExecutec C sD t �d�}|�� dd� }| �t j�� |�dd|� W 5 Q R X d S )Nr* c S s dS �Nru r )r� r r r r� r� z3ClosedConTests.CheckClosedCreateFunction.<locals>.fr- r )r r. r7 rB r" Zcreate_function)r rC r� r r r �CheckClosedCreateFunction s z(ClosedConTests.CheckClosedCreateFunctionc C sJ t �d�}|�� G dd� d�}| �t j�� |�dd|� W 5 Q R X d S )Nr* c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )z6ClosedConTests.CheckClosedCreateAggregate.<locals>.Aggc S s d S r5 r r r r r r� s z?ClosedConTests.CheckClosedCreateAggregate.<locals>.Agg.__init__c S s d S r5 r r� r r r �step s z;ClosedConTests.CheckClosedCreateAggregate.<locals>.Agg.stepc S s dS r* r r r r r �finalize s z?ClosedConTests.CheckClosedCreateAggregate.<locals>.Agg.finalizeN)r& r' r( r� r, r- r r r r �Agg s r. r- r )r r. r7 rB r" Zcreate_aggregate)r rC r. r r r �CheckClosedCreateAggregate s z)ClosedConTests.CheckClosedCreateAggregatec C s@ t �d�}|�� dd� }| �t j�� |�|� W 5 Q R X d S )Nr* c W s t jS r5 )r ZDENY)�argsr r r � authorizer s z;ClosedConTests.CheckClosedSetAuthorizer.<locals>.authorizer)r r. r7 rB r" Zset_authorizer)r rC r1 r r r �CheckClosedSetAuthorizer s z'ClosedConTests.CheckClosedSetAuthorizerc C sB t �d�}|�� dd� }| �t j�� |�|d� W 5 Q R X d S )Nr* c S s d S r5 r r r r r �progress" r� z?ClosedConTests.CheckClosedSetProgressCallback.<locals>.progressr� )r r. r7 rB r" Zset_progress_handler)r rC r3 r r r � CheckClosedSetProgressCallback s z-ClosedConTests.CheckClosedSetProgressCallbackc C s4 t �d�}|�� | �t j�� |� W 5 Q R X d S r$ )r r. r7 rB r" r& r r r �CheckClosedCall& s z ClosedConTests.CheckClosedCallN) r& r' r( r% r' r( r) r+ r/ r2 r4 r5 r r r r r# � s r# c @ s e Zd Zdd� ZdS )�ClosedCurTestsc C sz t �d�}|�� }|�� dD ]V}|dkr0d}n|dkrFdddgf}ng }| �t j�� t||�}||� W 5 Q R X q d S ) Nr* )r1 r� r r� r� rH )r1 r )zselect 4 union select 5r� z insert into foo(bar) values (?)r� r )r r. r0 r7 rB r" �getattr)r rC r� Z method_name�params�methodr r r � CheckClosed- s zClosedCurTests.CheckClosedN)r& r' r( r: r r r r r6 , s r6 c @ sX e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dS )�SqliteOnConflictTestszz Tests for SQLite's "insert on conflict" feature. See https://www.sqlite.org/lang_conflict.html for details. c C s( t �d�| _| j�� | _| j�d� d S )Nr* zz CREATE TABLE test( id INTEGER PRIMARY KEY, name TEXT, unique_name TEXT UNIQUE ); r^ r r r r r4 F s zSqliteOnConflictTests.setUpc C s | j �� | j�� d S r5 r_ r r r r r8 O s z SqliteOnConflictTests.tearDownc C s� d | j _| j �� | _| j�d� | j�d� | j�d� | �tj�� | j�d� W 5 Q R X | j �� | j�d� | � | j� � g � d S )N�BEGIN�,INSERT INTO test(name) VALUES ('abort_test')�8INSERT OR ROLLBACK INTO test(unique_name) VALUES ('foo')z"SELECT name, unique_name from test� r/ Zisolation_levelr0 r3 r1 rB r r r: r r� r r r r �.CheckOnConflictRollbackWithExplicitTransactionS s zDSqliteOnConflictTests.CheckOnConflictRollbackWithExplicitTransactionc C s� d | j _| j �� | _| j�d� | j�d� | j�d� | �tj�� | j�d� W 5 Q R X | j �� | j�d� | � | j� � ddg� d S )Nr< r= �5INSERT OR ABORT INTO test(unique_name) VALUES ('foo')�"SELECT name, unique_name FROM test�Z abort_testNr� r? r r r r �2CheckOnConflictAbortRaisesWithExplicitTransactionsb s zHSqliteOnConflictTests.CheckOnConflictAbortRaisesWithExplicitTransactionsc C s^ | j �d� | j �d� | �tj�� | j �d� W 5 Q R X | j �d� | �| j �� g � d S )Nr= r> rB �r3 r1 rB r r r r� r r r r �)CheckOnConflictRollbackWithoutTransactionr s z?SqliteOnConflictTests.CheckOnConflictRollbackWithoutTransactionc C sb | j �d� | j �d� | �tj�� | j �d� W 5 Q R X | j �d� | �| j �� ddg� d S )Nr= rA rB rC r� rE r r r r �-CheckOnConflictAbortRaisesWithoutTransactions| s zCSqliteOnConflictTests.CheckOnConflictAbortRaisesWithoutTransactionsc C sF | j �d� | �tj�� | j �d� W 5 Q R X | �| j �� g � d S )Nz4INSERT OR FAIL INTO test(unique_name) VALUES ('foo')rE r r r r �CheckOnConflictFail� s z)SqliteOnConflictTests.CheckOnConflictFailc C s< | j �d� | j �d� | j �d� | �| j �� dg� d S )Nz6INSERT OR IGNORE INTO test(unique_name) VALUES ('foo')z SELECT unique_name FROM testr, �r3 r1 r r� r r r r �CheckOnConflictIgnore� s z+SqliteOnConflictTests.CheckOnConflictIgnorec C s< | j �d� | j �d� | j �d� | �| j �� dg� d S )NzFINSERT OR REPLACE INTO test(name, unique_name) VALUES ('Data!', 'foo')zUINSERT OR REPLACE INTO test(name, unique_name) VALUES ('Very different data!', 'foo')rB )zVery different data!r- rI r r r r �CheckOnConflictReplace� s z,SqliteOnConflictTests.CheckOnConflictReplaceN) r& r' r( �__doc__r4 r8 r@ rD rF rG rH rJ rK r r r r r; ? s r; c C s� t �td�} t �td�}t �td�}t �td�}t �td�}t �td�}t �td�}t �t d�}t �t d�}t � | ||||||||f �S )NZCheck) r\ Z makeSuiter r) r] r� r r r# r6 r; Z TestSuite) Z module_suiteZconnection_suiteZ cursor_suiteZ thread_suiteZconstructor_suiteZ ext_suiteZclosed_con_suiteZclosed_cur_suiteZon_conflict_suiter r r �suite� s( �rM c C s t �� } | �t� � d S r5 )r\ ZTextTestRunnerr� rM )Zrunnerr r r r� � s r� �__main__)r� r\ Zsqlite3r Z test.supportr r ZTestCaser r) r] r� r r r# r6 r; rM r� r& r r r r �<module> s& 7m A 4C]