Your IP : 216.73.217.129
o
0�i�� � @ s4 d dl Z d dlZd dlZd dlZd dlZd dlmZmZ d dl m
Z
m Z G dd� dej �Z
G dd� dej �ZG dd � d ej �ZG d
d � d ej �ZG d d
� d
ej �ZG dd� dej �ZG dd� dej �ZG dd� dej �ZG dd� dej �ZG dd� dej �ZG dd� dej �Zdd� Zd d � Zed kr�e� dS dS ) � N)� check_disallow_instantiation�
SHORT_TIMEOUT)�TESTFN�unlinkc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Ze�ejdkd �d d � �Zd d � Zd!S )"� ModuleTestsc C � | � tjddtj � d S )Nz2.0z apilevel is %s, should be 2.0)� assertEqual�sqliteZapilevel��self� r �9/opt/alt/python310/lib64/python3.10/sqlite3/test/dbapi.py�test_api_level" �
�zModuleTests.test_api_levelc C r )N� z threadsafety is %d, should be 1)r r Z threadsafetyr
r r r
�test_thread_safety& r z ModuleTests.test_thread_safetyc C r )NZqmarkz%paramstyle is '%s', should be 'qmark')r r Z
paramstyler
r r r
�test_param_style* s
��z ModuleTests.test_param_stylec C � | � ttjt�d� d S )Nz&Warning is not a subclass of Exception)�
assertTrue�
issubclassr �Warning� Exceptionr
r r r
� test_warning/ � �zModuleTests.test_warningc C r )Nz$Error is not a subclass of Exception)r r r �Errorr r
r r r
�
test_error3 r zModuleTests.test_errorc C � | � ttjtj�d� d S )Nz)InterfaceError is not a subclass of Error)r r r �InterfaceErrorr r
r r r
�test_interface_error7 � �z ModuleTests.test_interface_errorc C r )Nz(DatabaseError is not a subclass of Error)r r r �
DatabaseErrorr r
r r r
�test_database_error; r z ModuleTests.test_database_errorc C r )Nz,DataError is not a subclass of DatabaseError)r r r � DataErrorr r
r r r
�test_data_error? r zModuleTests.test_data_errorc C r )Nz3OperationalError is not a subclass of DatabaseError)r r r �OperationalErrorr r
r r r
�test_operational_errorC r z"ModuleTests.test_operational_errorc C r )Nz1IntegrityError is not a subclass of DatabaseError)r r r �IntegrityErrorr r
r r r
�test_integrity_errorG r z ModuleTests.test_integrity_errorc C r )Nz0InternalError is not a subclass of DatabaseError)r r r �
InternalErrorr r
r r r
�test_internal_errorK r z ModuleTests.test_internal_errorc C r )Nz3ProgrammingError is not a subclass of DatabaseError)r r r �ProgrammingErrorr r
r r r
�test_programming_errorO r z"ModuleTests.test_programming_errorc C r )Nz4NotSupportedError is not a subclass of DatabaseError)r r r �NotSupportedErrorr r
r r r
�test_not_supported_errorS s
��z$ModuleTests.test_not_supported_error�darwinz#shared cache is deprecated on macOSc C sP dD ]#}| � t��
}t�|� W d � n1 sw Y | �d|j� qd S )N)TFzdbapi.py)Z assertWarns�DeprecationWarningr Zenable_shared_cacheZassertIn�filename)r Zenable�cmr r r
� test_shared_cache_deprecatedZ s ��z(ModuleTests.test_shared_cache_deprecatedc C s t �d�}t| t|d��� d S )N�:memory:zselect 1)r �connectr �type�r �cxr r r
�test_disallow_instantiationa s
z'ModuleTests.test_disallow_instantiationN)�__name__�
__module__� __qualname__r r r r r r r! r# r% r' r) r+ r- �unittest�skipIf�sys�platformr2 r8 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 � Zd S ) �ConnectionTestsc C s0 t �d�| _| j�� }|�d� |�dd� d S )Nr3 z4create table test(id integer primary key, name text)�!insert into test(name) values (?)��foo)r r4 r7 �cursor�execute�r �cur r r
�setUph s
zConnectionTests.setUpc C � | j �� d S �N�r7 �closer
r r r
�tearDownn � zConnectionTests.tearDownc C rI rJ �r7 �commitr
r r r
� test_commitq rN zConnectionTests.test_commitc C � | j �� | j �� dS )zV
A commit should also work when no changes were made to the database.
NrO r
r r r
� test_commit_after_no_changest �
z,ConnectionTests.test_commit_after_no_changesc C rI rJ �r7 �rollbackr
r r r
�
test_rollback{ rN z ConnectionTests.test_rollbackc C rR )zX
A rollback should also work when no changes were made to the database.
NrU r
r r r
� test_rollback_after_no_changes~ rT z.ConnectionTests.test_rollback_after_no_changesc C s | j �� }d S rJ �r7 rD rF r r r
� test_cursor� rN zConnectionTests.test_cursorc C s@ d}| � tj�� t�|�}W d � d S 1 sw Y d S )Nz/foo/bar/bla/23534/mydb.db)� assertRaisesr r$ r4 )r ZYOU_CANNOT_OPEN_THIS�conr r r
�test_failed_open� s "�z ConnectionTests.test_failed_openc C rI rJ rK r
r r r
�
test_close� rN zConnectionTests.test_closec C s� | � | jjtj� | � | jjtj� | � | jjtj� | � | jjtj� | � | jjtj� | � | jjtj� | � | jj tj � | � | jj
tj
� | � | jj tj � | � | jj tj � d S rJ )
r r7 r r r r r r" r$ r& r( r* r, r
r r r
�test_exceptions� s z ConnectionTests.test_exceptionsc C s� t �d�}|�� }| �|jd� |�d� | �|jd� |�dd� | �|jd� |�ddg� |�� }| �|jd� |�� | �|jd� |�ddg� |�� }| �|jd� d S ) Nr3 Fz?create table transactiontest(id integer primary key, name text)z,insert into transactiontest(name) values (?)rB Tz-select name from transactiontest where name=?rC )r r4 rD r �in_transactionrE �fetchonerP )r r7 rG �rowr r r
�test_in_transaction� s
z#ConnectionTests.test_in_transactionc C s8 | � t��
d| j_W d � d S 1 sw Y d S )NT)r[ �AttributeErrorr7 r` r
r r r
�test_in_transaction_ro� s
"�z&ConnectionTests.test_in_transaction_roc C sZ | � tt� G dd� d�}|� }t�|��}|�d� W d � dS 1 s&w Y dS )z| Checks that we can successfully connect to a database using an object that
is PathLike, i.e. has __fspath__(). c @ � e Zd Zdd� ZdS )z=ConnectionTests.test_open_with_path_like_object.<locals>.Pathc S s t S rJ )r r
r r r
�
__fspath__� � zHConnectionTests.test_open_with_path_like_object.<locals>.Path.__fspath__N)r9 r: r; rg r r r r
�Path� � ri � create table test(id integer)N)�
addCleanupr r r r4 rE )r ri �pathr7 r r r
� test_open_with_path_like_object� s "�z/ConnectionTests.test_open_with_path_like_objectc C s� | � tt� t�t��
}|�d� W d � n1 sw Y tjdt dd��
}|�d� W d � n1 s9w Y tjdt d dd��,}| �tj��
|�d� W d � n1 s`w Y W d � d S W d � d S 1 sxw Y d S )Nrk zfile:T)Zuriz insert into test(id) values(0)z?mode=roz insert into test(id) values(1))rl r r r r4 rE r[ r$ r6 r r r
�
test_open_uri� s � � ��"�z ConnectionTests.test_open_uriN)r9 r: r; rH rM rQ rS rW rX rZ r] r^ r_ rc re rn ro r r r r
r@ f s
r@ c @ � e Zd Zdd� Zdd� ZdS )� UninitialisedConnectionTestsc C s t j�t j�| _d S rJ )r Z
Connection�__new__r7 r
r r r
rH � � z"UninitialisedConnectionTests.setUpc s� � fdd�� fdd�� fdd�� fdd�� fdd�� fdd�f}|D ] }� j |d�� � �tjd |� W d � n1 s=w Y q"d S )
Nc � � j jS rJ )r7 �isolation_levelr r
r r
�<lambda>� � zEUninitialisedConnectionTests.test_uninit_operations.<locals>.<lambda>c rt rJ )r7 �
total_changesr r
r r
rv � rw c rt rJ )r7 r` r r
r r
rv � rw c �
� j �� S rJ )r7 Ziterdumpr r
r r
rv � �
c ry rJ rY r r
r r
rv � rz c ry rJ rK r r
r r
rv � rz )�funcz#Base Connection.__init__ not called)�subTestZassertRaisesRegexr r* )r Zfuncsr{ r r
r
�test_uninit_operations� s
�� ���z3UninitialisedConnectionTests.test_uninit_operationsN)r9 r: r; rH r} r r r r
rq � s rq 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 e � e!j"d9k d:�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� Z0dWdX� Z1dYdZ� Z2d[d\� Z3d]d^� Z4d_d`� Z5dadb� Z6dcdd� Z7dedf� Z8dgdh� Z9didj� Z:dkS )l� CursorTestsc C s6 t �d�| _| j�� | _| j�d� | j�dd� d S )Nr3 z\create table test(id integer primary key, name text, income number, unique_test text unique)rA rB �r r4 r7 rD rG rE r
r r r
rH � s �zCursorTests.setUpc C � | j �� | j�� d S rJ �rG rL r7 r
r r r
rM � �
zCursorTests.tearDownc C � | j �d� d S )N�delete from test�rG rE r
r r r
�test_execute_no_args� � z CursorTests.test_execute_no_argsc C �> | � tj�� | j�d� W d � d S 1 sw Y d S )Nz select asdf)r[ r r$ rG rE r
r r r
�test_execute_illegal_sql� � "�z$CursorTests.test_execute_illegal_sqlc C r� )Nzselect 5+4; select 4+5)r[ r r rG rE r
r r r
�test_execute_too_much_sql� r� z%CursorTests.test_execute_too_much_sqlc C r� )Nzselect 5+4; -- foo barr� r
r r r
�test_execute_too_much_sql2� r� z&CursorTests.test_execute_too_much_sql2c C r� )NzT
select 5+4;
/*
foo
*/
r� r
r r r
�test_execute_too_much_sql3� r� z&CursorTests.test_execute_too_much_sql3c C s< | � t�� | j�d� W d � d S 1 sw Y d S �N�* )r[ � TypeErrorrG rE r
r r r
�test_execute_wrong_sql_arg s "�z&CursorTests.test_execute_wrong_sql_argc C � | j �dd� d S )N� insert into test(id) values (?))r� r� r
r r r
�test_execute_arg_int � z CursorTests.test_execute_arg_intc C r� )N�#insert into test(income) values (?))gq=
ף��@r� r
r r r
�test_execute_arg_float r� z"CursorTests.test_execute_arg_floatc C r� )NrA )ZHugor� r
r r r
�test_execute_arg_string r� z#CursorTests.test_execute_arg_stringc C s@ | j �dd� | j �d| j jf� | j �� }| �|d d� d S )NrA )�Hu goz select name from test where id=?r r� )rG rE � lastrowidra r �r rb r r r
�&test_execute_arg_string_with_zero_byte s
z2CursorTests.test_execute_arg_string_with_zero_bytec C sN | � t��}| j�dd� W d � n1 sw Y | �t|j�d� d S )Nr� r� z"parameters are of unsupported type)r[ �
ValueErrorrG rE r �str� exception)r r1 r r r
�test_execute_non_iterable s �z%CursorTests.test_execute_non_iterablec C s@ | � tj�� | j�dd� W d � d S 1 sw Y d S )Nr� )� ZEgon�r[ r r* rG rE r
r r r
� test_execute_wrong_no_of_args1 s "�z*CursorTests.test_execute_wrong_no_of_args1c C r� �Nr� r� r
r r r
� test_execute_wrong_no_of_args2 � "�z*CursorTests.test_execute_wrong_no_of_args2c C r� r� r� r
r r r
� test_execute_wrong_no_of_args3% r� z*CursorTests.test_execute_wrong_no_of_args3c C s: | j �d� | j �ddg� | j �� }| �|d d� d S )N�%insert into test(name) values ('foo')�"select name from test where name=?rC r �rG rE ra r r� r r r
�test_execute_param_list* s
z#CursorTests.test_execute_param_listc C sH G dd� d�}| j �d� | j �d|� � | j �� }| �|d d� d S )Nc @ rp )z2CursorTests.test_execute_param_sequence.<locals>.Lc S � dS )Nr r r
r r r
�__len__2 rh z:CursorTests.test_execute_param_sequence.<locals>.L.__len__c S s |dksJ �dS )Nr rC r �r �xr r r
� __getitem__4 s z>CursorTests.test_execute_param_sequence.<locals>.L.__getitem__N�r9 r: r; r� r� r r r r
�L1 � r� r� r� r rC r� )r r� rb r r r
�test_execute_param_sequence0 s
z'CursorTests.test_execute_param_sequencec C sZ G dd� d�}| j �d� | �t�� | j �d|� � W d � d S 1 s&w Y d S )Nc @ rp )z:CursorTests.test_execute_param_sequence_bad_len.<locals>.Lc S s dd d S )Nr r r r
r r r
r� @ � zBCursorTests.test_execute_param_sequence_bad_len.<locals>.L.__len__c S s t �rJ )�AssertionError)Zslfr� r r r
r� B rh zFCursorTests.test_execute_param_sequence_bad_len.<locals>.L.__getitem__Nr� r r r r
r� ? r� r� r� r� )rG rE r[ �ZeroDivisionError)r r� r r r
�#test_execute_param_sequence_bad_len= s
"�z/CursorTests.test_execute_param_sequence_bad_lenc C s< | j �d� | j �dddi� | j �� }| �|d d� d S )Nr� �&select name from test where name=:name�namerC r r� r� r r r
�test_execute_dict_mappingI s
z%CursorTests.test_execute_dict_mappingc C sJ G dd� dt �}| j�d� | j�d|� � | j�� }| �|d d� d S )Nc @ rf )z8CursorTests.test_execute_dict_mapping_mapping.<locals>.Dc S r� �NrC r )r �keyr r r
� __missing__Q rh zDCursorTests.test_execute_dict_mapping_mapping.<locals>.D.__missing__N)r9 r: r; r� r r r r
�DP rj r� r� r� r rC )�dictrG rE ra r )r r� rb r r r
�!test_execute_dict_mapping_mappingO s
z-CursorTests.test_execute_dict_mapping_mappingc C �P | j �d� | �tj�� | j �dddi� W d � d S 1 s!w Y d S )Nr� z1select name from test where name=:name and id=:idr� rC �rG rE r[ r r* r
r r r
�)test_execute_dict_mapping_too_little_argsY � "�z5CursorTests.test_execute_dict_mapping_too_little_argsc C sJ | j �d� | �tj�� | j �d� W d � d S 1 s w Y d S )Nr� r� r� r
r r r
�!test_execute_dict_mapping_no_args^ s "�z-CursorTests.test_execute_dict_mapping_no_argsc C r� )Nr� r� r� rC r� r
r r r
�!test_execute_dict_mapping_unnamedc r� z-CursorTests.test_execute_dict_mapping_unnamedc C rI rJ )rG rL r
r r r
r^ h rN zCursorTests.test_closec C sD | j �d� | j �d� | j �d� | j �d� | �| j jd� d S )Nr� r� zupdate test set name='bar'� �rG rE r �rowcountr
r r r
�test_rowcount_executek s
z!CursorTests.test_rowcount_executec 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
�test_rowcount_selectr s z CursorTests.test_rowcount_selectc C s2 | j �d� | j �dg d�� | �| j jd� d S )Nr� rA ))r )r� �� r� )rG rE � executemanyr r� r
r r r
�test_rowcount_executemany{ s z%CursorTests.test_rowcount_executemany)r� �# r z Requires SQLite 3.35.0 or newerc C s6 | j �d� | �| j �� d d� | �| j jd� d S )Nz7update test set name='bar' where name='foo' returning 1r r )rG rE r ra r� r
r r r
� test_rowcount_update_returning� s z*CursorTests.test_rowcount_update_returningc C s0 | j �d� | j �d� | jd| jjdd� d S )Nr� r� z"total changes reported wrong value)�msg)rG rE Z
assertLessr7 rx r
r r r
�test_total_changes� s z CursorTests.test_total_changesc C s" | j �ddd� tdd�D �� d S )Nr� c S s g | ]}|f�qS r r )�.0r� r r r
�
<listcomp>� s z:CursorTests.test_execute_many_sequence.<locals>.<listcomp>�d �n )rG r� �ranger
r r r
�test_execute_many_sequence� s "z&CursorTests.test_execute_many_sequencec C s" G dd� d�}| j �d|� � d S )Nc @ �$ e Zd Zdd� Zdd� Zdd� ZdS )z6CursorTests.test_execute_many_iterator.<locals>.MyIterc S s
d| _ d S �N� )�valuer
r r r
�__init__� s
z?CursorTests.test_execute_many_iterator.<locals>.MyIter.__init__c S s | S rJ r r
r r r
�__iter__� rh z?CursorTests.test_execute_many_iterator.<locals>.MyIter.__iter__c S s$ | j dkrt�| j d7 _ | j fS )N�
r )r� �
StopIterationr
r r r
�__next__� s
z?CursorTests.test_execute_many_iterator.<locals>.MyIter.__next__N)r9 r: r; r� r� r� r r r r
�MyIter� s r� r� �rG r� )r r� r r r
�test_execute_many_iterator� s z&CursorTests.test_execute_many_iteratorc C s dd� }| j �d|� � d S )Nc s s � t d�D ]} | fV qd S r� )r� )�ir r r
�mygen� s �
�z6CursorTests.test_execute_many_generator.<locals>.mygenr� r� )r r� r r r
�test_execute_many_generator� s z'CursorTests.test_execute_many_generatorc C s@ | � t�� | j�ddg� W d � d S 1 sw Y d S )Nr� r� �r[ r� rG r� r
r r r
� test_execute_many_wrong_sql_arg� s "�z+CursorTests.test_execute_many_wrong_sql_argc C sB | � tj�� | j�ddg� W d � d S 1 sw Y d S )Nzselect ?r� )r[ r r* rG r� r
r r r
�test_execute_many_select� s "�z$CursorTests.test_execute_many_selectc C s> | � t�� | j�dd� W d � d S 1 sw Y d S )Nr� r� r� r
r r r
� test_execute_many_not_iterable� s "�z*CursorTests.test_execute_many_not_iterablec C sv | j �d� | j �dd� | j �dd� | j �d� g }| j D ] }|�|d � q | �|d d� | �|d d � d S )
Nr� r� )r� )� z select id from test order by idr r� r r� )rG rE �appendr )r Zlstrb r r r
�test_fetch_iter� s
zCursorTests.test_fetch_iterc C s@ | j �d� | j �� }| �|d d� | j �� }| �|d � d S )N�select name from testr rC r� r� r r r
�
test_fetchone� �
zCursorTests.test_fetchonec C s" | j �� }|�� }| �|d � d S rJ )r7 rD ra r )r �currb r r r
�test_fetchone_no_statement� s
z&CursorTests.test_fetchone_no_statementc 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 rG Z arraysizerE � fetchmany�len�r �resr r r
�test_array_size� s
zCursorTests.test_array_sizec C sD | j �d� | j �d�}| �t|�d� | j �d�}| �|g � d S )Nr� r� r �rG rE r� r r� r� r r r
�test_fetchmany� s
zCursorTests.test_fetchmanyc C s. | j �d� | j jdd�}| �t|�d� dS )z0Checks if fetchmany works with keyword argumentsr� r� )�sizer Nr� r� r r r
�test_fetchmany_kw_arg� s z!CursorTests.test_fetchmany_kw_argc C s@ | j �d� | j �� }| �t|�d� | j �� }| �|g � d S )Nr� r )rG rE �fetchallr r� r� r r r
�
test_fetchall� r� zCursorTests.test_fetchallc C s | j �g d�� d S )N)r� � r� )rG Z
setinputsizesr
r r r
�test_setinputsizes� rs z CursorTests.test_setinputsizesc C r� )Nr� r �rG Z
setoutputsizer
r r r
�test_setoutputsize� r� z CursorTests.test_setoutputsizec C r� r� r r
r r r
� test_setoutputsize_no_column� r� z(CursorTests.test_setoutputsize_no_columnc C s | � | jj| j� d S rJ )r rG Z
connectionr7 r
r r r
�test_cursor_connection� s z"CursorTests.test_cursor_connectionc C sD | � t�� dd� }| j�|�}W d � d S 1 sw Y d S )Nc S � d S rJ r r r r r
�f � z1CursorTests.test_wrong_cursor_callable.<locals>.f)r[ r� r7 rD )r r r� r r r
�test_wrong_cursor_callable s "�z&CursorTests.test_wrong_cursor_callablec C sN G dd� d�}|� }| � t�� t�|�}W d � d S 1 s w Y d S )Nc @ s e Zd ZdS )z0CursorTests.test_cursor_wrong_class.<locals>.FooN)r9 r: r; r r r r
�Foo s r )r[ r� r ZCursor)r r rC r� r r r
�test_cursor_wrong_class s
"�z#CursorTests.test_cursor_wrong_classc C sb d}dD ]*}| j |d�� | j�|�|�d� | �| jjd� W d � n1 s)w Y 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 rC r N)r| rG rE �formatr r� )r �sqlr r r r
�test_last_row_id_on_replace
s ���z'CursorTests.test_last_row_id_on_replacec C s@ | j �dd� | �| j jd� | j �dd� | �| j jd� d S )Nz2insert or ignore into test(unique_test) values (?))�testr� )rG rE r r� r
r r r
�test_last_row_id_on_ignore s ��z&CursorTests.test_last_row_id_on_ignorec
C s� g }dD ][}d}| j d�|�d��F | j�|�|�|f� |�|| jjf� | �tj�� | j�|�|�|f� W d � n1 sBw Y |�|| jjf� W d � n1 sZw Y qg d�}| � ||� d S )N)�FAIL�ABORT�ROLLBACKz.INSERT OR {} INTO test(unique_test) VALUES (?)z INSERT OR {}r )�r r� r �r r� r �r r r )
r| r rG rE r� r� r[ r r& r )r Zresultsr r Zexpectedr r r
�test_last_row_id_insert_o_r! s � ��z'CursorTests.test_last_row_id_insert_o_rN);r9 r: r; rH rM 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 Zsqlite_version_infor� 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~ � sp
�
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 �( t �d�| _| j�� | _| j�d� d S )Nr3 z\create table test(id integer primary key, name text, bin binary, ratio number, ts timestamp))r r4 r\ rD r� rE r
r r r
rH 4 � zThreadTests.setUpc C r� rJ )r� rL r\ r
r r r
rM 9 r� zThreadTests.tearDownc C �V dd� }g }t j|| j|d�d�}|�� |�� t|�dkr)| �d�|�� d S d S )Nc S �D z | � � }|�d� W d S tjy Y d S |�d� Y d S �N� did not raise ProgrammingError�raised wrong exception)rD r� r r* )r\ �errorsr� r r r
�run> �
z(ThreadTests.test_con_cursor.<locals>.run�r\ r$ ��target�kwargsr �
�� threading�Threadr\ �start�joinr� �fail�r r% r$ �tr r r
�test_con_cursor= �
�zThreadTests.test_con_cursorc C r )Nc S �D z | � � |�d� W d S tjy Y d S |�d� Y d S r! )rP r� r r* r' r r r
r% P r& z(ThreadTests.test_con_commit.<locals>.runr' r( r r+ r, r2 r r r
�test_con_commitO r5 zThreadTests.test_con_commitc C r )Nc S r6 r! )rV r� r r* r' r r r
r% b r& z*ThreadTests.test_con_rollback.<locals>.runr' r( r r+ r, r2 r r r
�test_con_rollbacka r5 z ThreadTests.test_con_rollbackc C r )Nc S r6 r! �rL r� r r* r' r r r
r% t r& z'ThreadTests.test_con_close.<locals>.runr' r( r r+ r, r2 r r r
�test_con_closes r5 zThreadTests.test_con_closec C r )Nc S �F z
| � d� |�d� W d S tjy Y d S |�d� Y d S )N�#insert into test(name) values ('a')r" r# �rE r� r r* �r� r$ r r r
r% � �
z0ThreadTests.test_cur_implicit_begin.<locals>.runr> r( r r+ �r- r. r� r/ r0 r� r1 r2 r r r
�test_cur_implicit_begin� r5 z#ThreadTests.test_cur_implicit_beginc C r )Nc S r6 r! r9 r> r r r
r% � r& z'ThreadTests.test_cur_close.<locals>.runr> r( r r+ r@ r2 r r r
�test_cur_close� r5 zThreadTests.test_cur_closec C sb dd� }g }| j �d� tj|| j |d�d�}|�� |�� t|�dkr/| �d�|�� d S d S )Nc S r; )Nr� r" r# r= r> r r r
r% � r? z)ThreadTests.test_cur_execute.<locals>.runr< r> r( r r+ �r� rE r- r. r/ r0 r� r1 r2 r r r
�test_cur_execute� s
�z ThreadTests.test_cur_executec C sn dd� }g }| j �d� | j �d� tj|| j |d�d�}|�� |�� t|�dkr5| �d�|�� d S d S ) Nc S r r! )ra r� r r* )r� r$ rb r r r
r% � r& z+ThreadTests.test_cur_iter_next.<locals>.runr< r� r> r( r r+ rC r2 r r r
�test_cur_iter_next� s
�z ThreadTests.test_cur_iter_nextN)
r9 r: r; rH rM r4 r7 r8 r: rA rB rD rE r r r r
r 3 s r c @ �D e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�ConstructorTestsc C � t �ddd�}d S )N�� r� � )r ZDate�r �dr r r
� test_date� r� zConstructorTests.test_datec C rH )N� �' r� )r ZTime�r r3 r r r
� test_time� r� zConstructorTests.test_timec C s t �dddddd�}d S )NrI r� rJ rN rO r� )r Z Timestamp�r Ztsr r r
�test_timestamp� s z ConstructorTests.test_timestampc C � t �d�}d S r� )r Z
DateFromTicksrK r r r
�test_date_from_ticks� rN z%ConstructorTests.test_date_from_ticksc C rT r� )r Z
TimeFromTicksrP r r r
�test_time_from_ticks� rN z%ConstructorTests.test_time_from_ticksc C rT r� )r ZTimestampFromTicksrR r r r
�test_timestamp_from_ticks� rN z*ConstructorTests.test_timestamp_from_ticksc C rT )Ns ')r ZBinary)r �br r r
� test_binary� rN z ConstructorTests.test_binaryN)
r9 r: r; rM rQ rS rU rV rW rY r r r r
rG � s rG c @ rF )�ExtensionTestsc C sB t �d�}|�� }|�d� |�d� |�� d }| �|d� d S )Nr3 z�
-- bla bla
/* a stupid comment */
create table a(i);
insert into a(i) values (5);
zselect i from ar r� )r r4 rD �
executescriptrE ra r )r r\ r� r� r r r
�test_script_string_sql� s
z%ExtensionTests.test_script_string_sqlc C �N t �d�}|�� }| �t j�� |�d� W d � d S 1 s w Y d S )Nr3 z1create table test(x); asdf; create table test2(x)�r r4 rD r[ r$ r[ �r r\ r� r r r
�test_script_syntax_error� �
"�z'ExtensionTests.test_script_syntax_errorc C r] )Nr3 z5create table test(sadfsadfdsa); select foo from hurz;r^ r_ r r r
�test_script_error_normal� ra z'ExtensionTests.test_script_error_normalc C s\ t �d�}|�� }| �t��
}|�d� W d � n1 s w Y | �t|j�d� d S )Nr3 s9 create table test(foo); insert into test(foo) values (5);z script argument must be unicode.) r r4 rD r[ r� r[ r r� r� )r r\ r� r1 r r r
�"test_cursor_executescript_as_bytes s
�z1ExtensionTests.test_cursor_executescript_as_bytesc C s. t �d�}|�d��� d }| �|dd� d S )Nr3 zselect 5r r� z Basic test of Connection.execute)r r4 rE ra r �r r\ �resultr r r
�test_connection_execute s
z&ExtensionTests.test_connection_executec C sb t �d�}|�d� |�dddg� |�d��� }| �|d d dd � | �|d
d d d � d S ) Nr3 zcreate table test(foo)z insert into test(foo) values (?)r� �r z!select foo from test order by foor r� z$Basic test of Connection.executemanyr r )r r4 rE r� r r rd r r r
�test_connection_executemany s
z*ExtensionTests.test_connection_executemanyc C s8 t �d�}|�d� |�d��� d }| �|dd� d S )Nr3 z9create table test(foo); insert into test(foo) values (5);zselect foo from testr r� z&Basic test of Connection.executescript)r r4 r[ rE ra r rd r r r
� test_connection_executescript s
z,ExtensionTests.test_connection_executescriptN)
r9 r: r; r\ r` rb rc rf rh ri r r r r
rZ � s
rZ c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�ClosedConTestsc C sL t �d�}|�� | �t j��
|�� }W d � d S 1 s w Y d S �Nr3 )r r4 rL r[ r* rD r_ r r r
�test_closed_con_cursor �
"�z%ClosedConTests.test_closed_con_cursorc C �L t �d�}|�� | �t j��
|�� W d � d S 1 s w Y d S rk )r r4 rL r[ r* rP �r r\ r r r
�test_closed_con_commit! rm z%ClosedConTests.test_closed_con_commitc C rn rk )r r4 rL r[ r* rV ro r r r
�test_closed_con_rollback' rm z'ClosedConTests.test_closed_con_rollbackc C sV t �d�}|�� }|�� | �t j�� |�d� W d � d S 1 s$w Y d S )Nr3 zselect 4)r r4 rD rL r[ r* rE r_ r r r
�test_closed_cur_execute- s
"�z&ClosedConTests.test_closed_cur_executec C sZ t �d�}|�� dd� }| �t j�� |�dd|� W d � d S 1 s&w Y d S )Nr3 c S r� �Nr� r )r� r r r
r 7 r
z5ClosedConTests.test_closed_create_function.<locals>.frC r )r r4 rL r[ r* Zcreate_function)r r\ r r r r
�test_closed_create_function4 s
"�z*ClosedConTests.test_closed_create_functionc C s` t �d�}|�� G dd� d�}| �t j�� |�dd|� W d � d S 1 s)w Y d S )Nr3 c @ r� )z8ClosedConTests.test_closed_create_aggregate.<locals>.Aggc S r rJ r r
r r r
r� ? rh zAClosedConTests.test_closed_create_aggregate.<locals>.Agg.__init__c S r rJ r r� r r r
�stepA rh z=ClosedConTests.test_closed_create_aggregate.<locals>.Agg.stepc S r� rs r r
r r r
�finalizeC rh zAClosedConTests.test_closed_create_aggregate.<locals>.Agg.finalizeN)r9 r: r; r� ru rv r r r r
�Agg> s rw rC r )r r4 rL r[ r* Zcreate_aggregate)r r\ rw r r r
� test_closed_create_aggregate; s
"�z+ClosedConTests.test_closed_create_aggregatec C sV t �d�}|�� dd� }| �t j�� |�|� W d � d S 1 s$w Y d S )Nr3 c W s t jS rJ )r ZDENY)�argsr r r
�
authorizerK s z=ClosedConTests.test_closed_set_authorizer.<locals>.authorizer)r r4 rL r[ r* Zset_authorizer)r r\ rz r r r
�test_closed_set_authorizerH s
"�z)ClosedConTests.test_closed_set_authorizerc C sX t �d�}|�� dd� }| �t j�� |�|d� W d � d S 1 s%w Y d S )Nr3 c S r rJ r r r r r
�progressS r
zBClosedConTests.test_closed_set_progress_callback.<locals>.progressr� )r r4 rL r[ r* Zset_progress_handler)r r\ r| r r r
�!test_closed_set_progress_callbackP s
"�z0ClosedConTests.test_closed_set_progress_callbackc C sJ t �d�}|�� | �t j�� |� W d � d S 1 s w Y d S rk )r r4 rL r[ r* ro r r r
�test_closed_callW s
"�z ClosedConTests.test_closed_callN) r9 r: r; rl rp rq rr rt rx r{ r} r~ r r r r
rj s
rj c @ rf )�ClosedCurTestsc C s� t �d�}|�� }|�� dD ]5}|dv rd}n
|dkr#dddgf}ng }| �t j�� t||�}||� W d � n1 s?w Y qd S ) Nr3 )rE r� r[ r r� ra )rE r[ )zselect 4 union select 5r� z insert into foo(bar) values (?)r� rg )r r4 rD rL r[ r* �getattr)r r\ r� Z method_name�params�methodr r r
� test_closed^ s
���zClosedCurTests.test_closedN)r9 r: r; r� r r r r
r ] rj r 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 r )Nr3 zz
CREATE TABLE test(
id INTEGER PRIMARY KEY, name TEXT, unique_name TEXT UNIQUE
);
r r
r r r
rH w r zSqliteOnConflictTests.setUpc C r� rJ r� r
r r r
rM � r� z SqliteOnConflictTests.tearDownc C s� d | j _| j �� | _| j�d� | j�d� | j�d� | �tj�� | j�d� W d � n1 s3w Y | 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� r7 ru rD rG rE r[ r r& rP r r r
r r r
�3test_on_conflict_rollback_with_explicit_transaction� s �
zISqliteOnConflictTests.test_on_conflict_rollback_with_explicit_transactionc C s� d | j _| j �� | _| j�d� | j�d� | j�d� | �tj�� | j�d� W d � n1 s3w Y | 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
�8test_on_conflict_abort_raises_with_explicit_transactions� s �
zNSqliteOnConflictTests.test_on_conflict_abort_raises_with_explicit_transactionsc C sr | j �d� | j �d� | �tj�� | j �d� W d � n1 s#w Y | j �d� | �| j �� g � d S )Nr� r� r� �rG rE r[ r r& r r r
r r r
�-test_on_conflict_rollback_without_transaction� s � zCSqliteOnConflictTests.test_on_conflict_rollback_without_transactionc C sv | j �d� | j �d� | �tj�� | j �d� W d � n1 s#w Y | j �d� | �| j �� ddg� d S )Nr� r� r� r� r� r� r
r r r
�2test_on_conflict_abort_raises_without_transactions� s � zHSqliteOnConflictTests.test_on_conflict_abort_raises_without_transactionsc C sZ | j �d� | �tj�� | j �d� W d � n1 s w Y | �| j �� g � d S )Nz4INSERT OR FAIL INTO test(unique_name) VALUES ('foo')r� r
r r r
�test_on_conflict_fail� s
�z+SqliteOnConflictTests.test_on_conflict_failc 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 testrB �rG rE r r r
r r r
�test_on_conflict_ignore� � z-SqliteOnConflictTests.test_on_conflict_ignorec 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')r� )zVery different data!rC r� r
r r r
�test_on_conflict_replace� r� z.SqliteOnConflictTests.test_on_conflict_replaceN)
r9 r: r; �__doc__rH rM r� r� r� r� r� r� r� r r r r
r� p s
r� c @ s$ e Zd Zed Zdd� Zdd� ZdS )�MultiprocessTestsg @�@c C s t t� d S rJ )r r r
r r r
rM � r� zMultiprocessTests.tearDownc
C sH dt � d| j� d�}tjtjd|gddtjtjd�}| �|j� | � d|j
� � � � � t
jt | jd �}zAz|�
|�d
� W d � n1 sGw Y W n t
jyh } z|j�t|�� W Y d }~n d }~ww |j�d � W |�� n|�� w | �|j� z |jd td
� W n tjy� |�� |�� � w | � |jd� d S )Nz�if 1:
import sqlite3
def wait():
print("started")
assert "database is locked" in input()
cx = sqlite3.connect("z ", timeout=a� )
cx.create_function("wait", 0, wait)
with cx:
cx.execute("create table t(t)")
try:
# execute two transactions; both will try to lock the db
cx.executescript('''
-- start a transaction and wait for parent
begin transaction;
select * from t;
select wait();
rollback;
-- start a new transaction; would fail if parent holds lock
begin transaction;
select * from t;
rollback;
''')
finally:
cx.close()
z-czutf-8r )�encoding�bufsize�stdin�stdoutZstarted)�timeoutz insert into t values('test')zno error�end)�inputr� )r �CONNECTION_TIMEOUT�
subprocess�Popenr> �
executable�PIPErl Z communicater r� �readline�stripr r4 rE r$ r� �writer� rL Z assertIsNone�
returncoder ZTimeoutExpired�kill)r ZSCRIPT�procr7 �excr r r
�&test_ctx_mgr_rollback_if_commit_failed� sH ��
� �� �� � �z8MultiprocessTests.test_ctx_mgr_rollback_if_commit_failedN)r9 r: r; r r� rM r� r r r r
r� � s r� c C s. t ttttttttt t
g } t � dd� | D ��S )Nc S s g | ] }t �� �|��qS r )r< Z
TestLoaderZloadTestsFromTestCase)r� r3 r r r
r� # s zsuite.<locals>.<listcomp>)
rj r r@ rG r~ rZ r r� r� r rq r< Z TestSuite)Ztestsr r r
�suite s �
�r� c C s t �� } | �t� � d S rJ )r< ZTextTestRunnerr% r� )Zrunnerr r r
r & s r �__main__)r� r- r< Zsqlite3r r> Z test.supportr r Ztest.support.os_helperr r ZTestCaser r@ rq r~ r rG rZ rj r r� r� r� r r9 r r r r
�<module> s4 Ec X 4C]G
�