Your IP : 216.73.216.213
3
��2a@ � @ s� d dl Z d dlZd dlZd dlZd dlmZ G dd� dej�ZG dd� d�Z G dd� dej�Z
d d
� Z d d � Ze d
kr|e� dS )� N)�supportc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z e
j e j
d=k d�dd� �Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Z d0d1� Z d2d3� Z d4d5� Z d6d7� Z d8d9� Z!d:d;� Z"d<S )>�RegressionTestsc C s t jd�| _d S )Nz:memory:)�sqlite�connect�con)�self� r �//opt/alt/python36/lib64/python3.6/regression.py�setUp s zRegressionTests.setUpc C s | j j� d S )N)r �close)r r r r �tearDown" s zRegressionTests.tearDownc C s | j j� }|jd� d S )Nzpragma user_version)r �cursor�execute)r �curr r r �CheckPragmaUserVersion% s
z&RegressionTests.CheckPragmaUserVersionc
C s@ t jdt jd�}z| jj� }|jd� W d |j� |j� X d S )Nz:memory:)� detect_typeszpragma schema_version)r r ZPARSE_COLNAMESr r
r r )r r r r r r �CheckPragmaSchemaVersion* s
z(RegressionTests.CheckPragmaSchemaVersionc s� t jddd�� � fdd�td�D �}|d jd� x.td�D ]"}|d jd d
d� td�D �� q<W x&td�D ]}|| jd | d � qlW � j� d S )
Nz:memory:� )Zcached_statementsc s g | ] }� j � �qS r )r
)�.0�x)r r r �
<listcomp>9 s z7RegressionTests.CheckStatementReset.<locals>.<listcomp>r zcreate table test(x)�
z insert into test(x) values (?)c S s g | ]
}|f�qS r r )r r r r r r < s � zselect x from test)r r �ranger � executemany�rollback)r �cursors�ir )r r �CheckStatementReset4 s "z#RegressionTests.CheckStatementResetc C sN | j j� }|jd� | j|jd d d� |jd� | j|jd d d� d S )Nz select 1 as "foo bar [datetime]"r zfoo barzselect 1 as "foo baz"zfoo baz)r r
r � assertEqual� description)r r r r r �CheckColumnNameWithSpacesC s
z)RegressionTests.CheckColumnNameWithSpacesc C sP t jd�}g }x4td�D ](}|j� }|j|� |jdt|� � qW |j� d S )Nz:memory:�i zselect 1 x union select )r r r r
�appendr �strr )r r r r r r r r �#CheckStatementFinalizationOnCloseDbK s
z3RegressionTests.CheckStatementFinalizationOnCloseDb� � zneeds sqlite 3.2.2 or newerc C s� t jd�}|jd� |jd� y|jd� W n t jk
rB Y nX |jd� y |j� W n t jk
rz | jd� Y nX d S )Nz:memory:z3create table foo(x, unique(x) on conflict rollback)z insert into foo(x) values (1)z insert into foo(x) values (2)z1pysqlite knew nothing about the implicit ROLLBACK)r r r Z
DatabaseError�commit�OperationalErrorZfail)r r r r r �CheckOnConflictRollbackY s
z'RegressionTests.CheckOnConflictRollbackc C s( | j jd� | j jd� | j jd� d S )Nzcreate table foo(bar)zdrop table foo)r r )r r r r �-CheckWorkaroundForBuggySqliteTransferBindingsh s z=RegressionTests.CheckWorkaroundForBuggySqliteTransferBindingsc C s | j jd� d S )N� )r r )r r r r �CheckEmptyStatementq s z#RegressionTests.CheckEmptyStatementc C sh d}t jdt jd�}|jd� |jdtjj� f� |j|� |jd� |jd� |jd� |j|� d S ) Nzselect * from fooz:memory:)r z create table foo(bar timestamp)z insert into foo(bar) values (?)zdrop table fooz create table foo(bar integer)z insert into foo(bar) values (5))r r �PARSE_DECLTYPESr �datetimeZnow)r ZSELECTr r r r �CheckTypeMapUsagex s
z!RegressionTests.CheckTypeMapUsagec C sP | j tj��$}| jjdttdg��f�j� W d Q R X d}| j|t |j
�� d S )Nz"select 'xxx' || ? || 'yyy' colname� z9Could not decode to UTF-8 column 'colname' with text 'xxx) � assertRaisesr r) r r �bytes� bytearray�fetchoneZassertInr$ Z exception)r �cm�msgr r r �CheckErrorMsgDecodeError� s
z(RegressionTests.CheckErrorMsgDecodeErrorc C s | j ttji d � d S )N)r2 � TypeErrorr Zregister_adapter)r r r r �CheckRegisterAdapter� s z$RegressionTests.CheckRegisterAdapterc s� G � fdd�dt �}tjd�� d � _xHdD ]@}| j|d��* |� _|j� � _|j� � _||�� _W d Q R X q*W d � _d� _d tfd
tfd tfd tfd
tfg}xR|D ]J\}}| j|d��0 | j |�� |� _W d Q R X | j
� jd� W d Q R X q�W d S )Nc s e Zd Zdd� Z� fdd�ZdS )z9RegressionTests.CheckSetIsolationLevel.<locals>.CustomStrc S s d S )Nr )r r r r �upper� s z?RegressionTests.CheckSetIsolationLevel.<locals>.CustomStr.upperc s
d� _ d S )Nr, )�isolation_level)r )r r r �__del__� s zARegressionTests.CheckSetIsolationLevel.<locals>.CustomStr.__del__N)�__name__�
__module__� __qualname__r; r= r )r r r � CustomStr� s rA z:memory:r, �DEFERRED� IMMEDIATE� EXCLUSIVE)�level� � �abczIMMEDIATE EXCLUSIVE� é)r, rB rC rD ) r$ r r r< �subTest�lower�
capitalizer9 �
ValueErrorr2 r )r rA rE Zpairs�value�excr )r r �CheckSetIsolationLevel� s$
z&RegressionTests.CheckSetIsolationLevelc C sp G dd� dt j�}t jd�}||�}| jt j�� |jd�j� W d Q R X | jt jd�� |j� W d Q R X d S )Nc @ s e Zd Zdd� ZdS )z?RegressionTests.CheckCursorConstructorCallCheck.<locals>.Cursorc S s d S )Nr )r r r r r �__init__� s zHRegressionTests.CheckCursorConstructorCallCheck.<locals>.Cursor.__init__N)r> r? r@ rQ r r r r �Cursor� s rR z:memory:z
select 4+5z$^Base Cursor\.__init__ not called\.$) r rR r r2 �ProgrammingErrorr �fetchall�assertRaisesRegexr )r rR r r r r r � CheckCursorConstructorCallCheck� s
z/RegressionTests.CheckCursorConstructorCallCheckc C s( G dd� dt �}| jjd|d�f� d S )Nc @ s e Zd ZdS )z/RegressionTests.CheckStrSubclass.<locals>.MyStrN)r> r? r@ r r r r �MyStr� s rW zselect ?rH )r$ r r )r rW r r r �CheckStrSubclass� s z RegressionTests.CheckStrSubclassc C s> G dd� dt j�}|d�}| jt j�� |j� }W d Q R X d S )Nc @ s e Zd Zdd� ZdS )zGRegressionTests.CheckConnectionConstructorCallCheck.<locals>.Connectionc S s d S )Nr )r �namer r r rQ � s zPRegressionTests.CheckConnectionConstructorCallCheck.<locals>.Connection.__init__N)r> r? r@ rQ r r r r �
Connection� s rZ z:memory:)r rZ r2 rS r
)r rZ r r r r r �#CheckConnectionConstructorCallCheck� s z3RegressionTests.CheckConnectionConstructorCallCheckc s� G � fdd�dt j�}G dd� dt j�� |d�}|j� }|jd� |jdd d
dg� |jd � |j� | jt j�� |j � W d Q R X d S )Nc s e Zd Z� fdd�ZdS )z;RegressionTests.CheckCursorRegistration.<locals>.Connectionc s � | �S )Nr )r )rR r r r
� s zBRegressionTests.CheckCursorRegistration.<locals>.Connection.cursorN)r> r? r@ r
r )rR r r rZ � s rZ c @ s e Zd Zdd� ZdS )z7RegressionTests.CheckCursorRegistration.<locals>.Cursorc S s t jj| |� d S )N)r rR rQ )r r r r r rQ � s z@RegressionTests.CheckCursorRegistration.<locals>.Cursor.__init__N)r> r? r@ rQ r r r r rR � s rR z:memory:zcreate table foo(x)z insert into foo(x) values (?)r& � r zselect x from foo)r& )r\ )r )
r rZ rR r
r r r r2 ZInterfaceErrorrT )r rZ r r r )rR r �CheckCursorRegistration� s
z'RegressionTests.CheckCursorRegistrationc C s t jdd d�}d S )Nz:memory:)r< )r r )r r r r r �CheckAutoCommit� s z RegressionTests.CheckAutoCommitc C s4 | j j� }|jd� |jd� |jd� |j� }d S )Nzcreate table foo(bar)z insert into foo(bar) values (5)zpragma page_size)r r
r r5 )r r �rowr r r �CheckPragmaAutocommit� s
z%RegressionTests.CheckPragmaAutocommitc C s | j tj| jd� d S )NrF )r2 r �Warningr )r r r r �CheckConnectionCall s z#RegressionTests.CheckConnectionCallc C s" dd� }| j tj| jjd|� d S )Nc S s dS )NrF r )�a�br r r � collation_cb s z4RegressionTests.CheckCollation.<locals>.collation_cbu �)r2 r rS r Zcreate_collation)r re r r r �CheckCollation
s z RegressionTests.CheckCollationc sf t jd�}|j� � � jd� � jd� � fdd�}| jt j�� � jddd� |� D �� W d Q R X d S ) Nz:memory:zcreate table a (bar)zcreate table b (baz)c 3 s � j dd� dV d S )Nz insert into a (bar) values (?)rF )rF )r r )r r r �foo s z4RegressionTests.CheckRecursiveCursorUse.<locals>.fooz insert into b (baz) values (?)c s s | ] }|fV qd S )Nr )r r r r r � <genexpr>$ s z:RegressionTests.CheckRecursiveCursorUse.<locals>.<genexpr>)r r r
r r2 rS r )r r rg r )r r �CheckRecursiveCursorUse s
z'RegressionTests.CheckRecursiveCursorUsec C s� t jdt jd�}|j� }|jd� |jd� |jd� |jd� dd� |j� D �}| j|tjd d
d
d d d
d�tjd d
d
d d d
d�g� d S )Nz:memory:)r z CREATE TABLE t (x TIMESTAMP)z4INSERT INTO t (x) VALUES ('2012-04-04 15:06:00.456')z:INSERT INTO t (x) VALUES ('2012-04-04 15:06:00.123456789')zSELECT * FROM tc S s g | ] }|d �qS )r r )r r r r r r 9 s zKRegressionTests.CheckConvertTimestampMicrosecondPadding.<locals>.<listcomp>i� r\ � � r i@� i@� )r r r. r
r rT r r/ )r r r �valuesr r r �'CheckConvertTimestampMicrosecondPadding& s
z7RegressionTests.CheckConvertTimestampMicrosecondPaddingc C s | j ttjddd� d S )Nz:memory:�{ )r< )r2 r9 r r )r r r r � CheckInvalidIsolationLevelType@ s z.RegressionTests.CheckInvalidIsolationLevelTypec C sR t jd�}| jt|d� | jt|d� |j� }| jt|jd� | jt|jd� d S )Nz:memory:z select 1z select 1 z
select 2z select 2 )r r r2 rM r
r )r r r r r r �CheckNullCharacterF s
z"RegressionTests.CheckNullCharacterc C s� t jd�}|jd� | j|jd� d}x�t|jd��D ]�\}}| j||d��p |jd|f� |j� |dkr~| j|d d� n2|dkr�| j|d d� n|d kr�| j|d d � |d7 }W d Q R X q6W | j|d
d � d S ) Nz:memory:z�
create table t(c);
create table t2(c);
insert into t values(0);
insert into t values(1);
insert into t values(2);
r, r zselect c from t)r r_ z insert into t2(c) values (?)rF r' r& z'should have returned exactly three rows) r r Z
executescriptr r< � enumerater rJ r( )r r Zcounterr r_ r r r �CheckCommitCursorResetO s"
z&RegressionTests.CheckCommitCursorResetc C sB dd� }t jd�}t j|�}tj||�}|j|� ~~tj� d S )Nc W s d S )Nr )�argsr r r �callbacks s z/RegressionTests.CheckBpo31770.<locals>.callbackz:memory:)r r rR �weakref�refrQ r Z
gc_collect)r rt r r rv r r r �
CheckBpo31770n s
z RegressionTests.CheckBpo31770N)r& r' r' )#r> r? r@ r
r r r r r! r% �unittestZskipIfr Zsqlite_version_infor* r+ r- r0 r8 r: rP rV rX r[ r] r^ r` rb rf ri rm ro rp rr rw r r r r r s8
r c @ s" e Zd ZdZddd�Zdd� ZdS )�UnhashableFuncNc C s d| _ || _d S )Nr )�calls� return_value)r r{ r r r rQ � s zUnhashableFunc.__init__c O s | j d7 _ | jS )NrF )rz r{ )r rs �kwargsr r r �__call__� s zUnhashableFunc.__call__)N)r> r? r@ �__hash__rQ r} r r r r ry s
ry c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�UnhashableCallbacksTestCasec C s t jd�| _d S )Nz:memory:)r r r )r r r r r
� s z!UnhashableCallbacksTestCase.setUpc C s | j j� d S )N)r r )r r r r r � s z$UnhashableCallbacksTestCase.tearDownc
C sL t dd�}| jtd�� | jj|d� W d Q R X | jjd� | j|j� d S )Nr )r{ zunhashable typerF zSELECT 1)ry rU r9 r Zset_progress_handlerr � assertFalserz )r �fr r r �test_progress_handler� s
z1UnhashableCallbacksTestCase.test_progress_handlerc C st d}t � }| jtd�� | jj|d|� W d Q R X d| }| jtj|�� | jjd| � W d Q R X | j|j � d S )N� func_namezunhashable typer zno such function: %sz SELECT %s())
ry rU r9 r Zcreate_functionr r) r r� rz )r r� r� r7 r r r � test_func� s z%UnhashableCallbacksTestCase.test_funcc C sL t tjd�}| jtd�� | jj|� W d Q R X | jjd� | j|j � d S )N)r{ zunhashable typezSELECT 1)
ry r Z SQLITE_DENYrU r9 r Zset_authorizerr r� rz )r r� r r r �test_authorizer� s
z+UnhashableCallbacksTestCase.test_authorizerc C sz G dd� dt �}d}| jtd�� | jj|d|df i �� W d Q R X d| }| jtj|�� | jjd| � W d Q R X d S ) Nc @ s e Zd ZdZdS )z=UnhashableCallbacksTestCase.test_aggr.<locals>.UnhashableTypeN)r> r? r@ r~ r r r r �UnhashableType� s r� � aggr_namezunhashable typer ZAggrzno such function: %sz SELECT %s())�typerU r9 r Zcreate_aggregater r) r )r r� r� r7 r r r � test_aggr� s "z%UnhashableCallbacksTestCase.test_aggrN) r> r? r@ r
r r� r� r� r� r r r r r � s
r c C s t jtd�} t j| t jt�f�S )NZCheck)rx Z makeSuiter Z TestSuiter )Zregression_suiter r r �suite� s r� c C s t j� } | jt� � d S )N)rx ZTextTestRunnerZrunr� )Zrunnerr r r �test� s r� �__main__)
r/ rx Zsqlite3r ru r� r ZTestCaser ry r r� r> r r r r �<module> s c 0