Your IP : 216.73.216.213
B
YI~d5G � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlmZ G dd� dej �Z
G dd� d�Z G dd � d ej �Z G d
d � d ej �Z
d d
� Zdd� Zedkr�e� dS )� N)�support)�patchc @ s e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z e
� e j
dk d�dd� �Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Z d/d0� Z d1d2� Z d3d4� Z d5d6� Z d7d8� Z!d9d:� Z"d;d<� Z#d=S )>�RegressionTestsc C s t �d�| _d S )Nz:memory:)�sqlite�connect�con)�self� r �</opt/alt/python37/lib64/python3.7/sqlite3/test/regression.py�setUp" s zRegressionTests.setUpc C s | j �� d S )N)r �close)r r r r
�tearDown% s zRegressionTests.tearDownc C s | j �� }|�d� d S )Nzpragma user_version)r �cursor�execute)r �curr r r
�CheckPragmaUserVersion( s
z&RegressionTests.CheckPragmaUserVersionc C s@ t jdt jd�}z| j�� }|�d� W d |�� |�� X d S )Nz:memory:)� detect_typeszpragma schema_version)r r �PARSE_COLNAMESr r r r )r r r r r r
�CheckPragmaSchemaVersion- s
z(RegressionTests.CheckPragmaSchemaVersionc s� t jddd�� � fdd�td�D �}|d �d� x.td�D ]"}|d �d d
d� td�D �� q<W x&td�D ]}|| �d | d � qlW � �� d S )
Nz:memory:� )Zcached_statementsc s g | ] }� � � �qS r )r )�.0�x)r r r
�
<listcomp>< 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
�CheckStatementReset7 s "z#RegressionTests.CheckStatementResetc C sN | j �� }|�d� | �|jd d d� |�d� | �|jd d d� d S )Nz select 1 as "foo bar [datetime]"r zfoo bar [datetime]zselect 1 as "foo baz"zfoo baz)r r r � assertEqual� description)r r r r r
�CheckColumnNameWithSpacesF s
z)RegressionTests.CheckColumnNameWithSpacesc C sP t �d�}g }x4td�D ](}|�� }|�|� |�dt|� � qW |�� d S )Nz:memory:�i zselect 1 x union select )r r r r �appendr �strr )r r r r r r r r
�#CheckStatementFinalizationOnCloseDbN s
z3RegressionTests.CheckStatementFinalizationOnCloseDb)� � r) zneeds sqlite 3.2.2 or newerc C s� t �d�}|�d� |�d� y|�d� W n t jk
rB Y nX |�d� y |�� W n t jk
rz | �d� 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
�CheckOnConflictRollback\ s
z'RegressionTests.CheckOnConflictRollbackc C s( | j �d� | j �d� | j �d� d S )Nzcreate table foo(bar)zdrop table foo)r r )r r r r
�-CheckWorkaroundForBuggySqliteTransferBindingsk s z=RegressionTests.CheckWorkaroundForBuggySqliteTransferBindingsc C s | j �d� d S )N� )r r )r r r r
�CheckEmptyStatementt s z#RegressionTests.CheckEmptyStatementc C sh d}t jdt jd�}|�d� |�dtj�� f� |�|� |�d� |�d� |�d� |�|� 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
�CheckTypeMapUsage{ s
z!RegressionTests.CheckTypeMapUsagec C sP | � tj��$}| j�dttdg��f��� W d Q R X d}| �|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 | � ttji d � d S )N)r4 � TypeErrorr Zregister_adapter)r r r r
�CheckRegisterAdapter� s z$RegressionTests.CheckRegisterAdapterc s� G � fdd�dt �}t�d�� d � _xHdD ]@}| j|d��* |� _|�� � _|�� � _||�� _W d Q R X q*W d � _d� _dtfdtfd tfd
tfd tfg}xR|D ]J\}}| j|d��0 | � |�� |� _W d Q R X | �
� 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 rC z:memory:)r. �DEFERREDZ IMMEDIATEZ EXCLUSIVE)�levelrD � � �abczIMMEDIATE EXCLUSIVE� é) r& r r r> �subTest�lower�
capitalizer; �
ValueErrorr4 r! )r rC rE Zpairs�value�excr )r r
�CheckSetIsolationLevel� s$
z&RegressionTests.CheckSetIsolationLevelc C sp G dd� dt j�}t �d�}||�}| �t j�� |�d��� W d Q R X | �t jd�� |�� 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@ rA rB rQ r r r r
�Cursor� s rR z:memory:z
select 4+5z$^Base Cursor\.__init__ not called\.$) r rR r r4 �ProgrammingErrorr �fetchall�assertRaisesRegexr )r rR r r r r r
� CheckCursorConstructorCallCheck� s
z/RegressionTests.CheckCursorConstructorCallCheckc C s( G dd� dt �}| j�d|d�f� d S )Nc @ s e Zd ZdS )z/RegressionTests.CheckStrSubclass.<locals>.MyStrN)r@ rA rB 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�}| �t 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@ rA rB rQ r r r r
�
Connection� s rZ z:memory:)r rZ r4 rS r )r rZ r r r r r
�#CheckConnectionConstructorCallCheck� s z3RegressionTests.CheckConnectionConstructorCallCheckc s� G � fdd�dt j�}G dd� dt j�� |d�}|�� }|�d� |�ddd d
g� |�d � |�� | �t 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@ rA rB r r )rR r r
rZ � s rZ c @ s e Zd Zdd� ZdS )z7RegressionTests.CheckCursorRegistration.<locals>.Cursorc S s t j�| |� d S )N)r rR rQ )r r r r r
rQ � s z@RegressionTests.CheckCursorRegistration.<locals>.Cursor.__init__N)r@ rA rB 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 rZ rR r r r r r4 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 �� }|�d� |�d� |�d� |�� }d S )Nzcreate table foo(bar)z insert into foo(bar) values (5)zpragma page_size)r r r r7 )r r �rowr r r
�CheckPragmaAutocommit� s
z%RegressionTests.CheckPragmaAutocommitc C s | � tj| jd� d S )NrF )r4 r �Warningr )r r r r
�CheckConnectionCall s z#RegressionTests.CheckConnectionCallc C s" dd� }| � tj| jjd|� d S )Nc S s dS )NrF r )�a�br r r
� collation_cb s z4RegressionTests.CheckCollation.<locals>.collation_cbu �)r4 r rS r Zcreate_collation)r re r r r
�CheckCollation
s z RegressionTests.CheckCollationc sf t �d�}|�� � � �d� � �d� � fdd�}| �t j�� � �ddd� |� D �� W d Q R X d S ) Nz:memory:zcreate table a (bar)zcreate table b (baz)c 3 s � � 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 r4 rS r )r r rg r )r r
�CheckRecursiveCursorUse s
z'RegressionTests.CheckRecursiveCursorUsec
C s� t jdt jd�}|�� }|�d� |�d� |�d� |�d� dd� |�� D �}| �|t�d d
d
d d d
d�t�d 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 < s zKRegressionTests.CheckConvertTimestampMicrosecondPadding.<locals>.<listcomp>i� r\ � � r i@� i@� )r r r0 r r rT r! r1 )r r r �valuesr r r
�'CheckConvertTimestampMicrosecondPadding) s
z7RegressionTests.CheckConvertTimestampMicrosecondPaddingc C s | j ttjddd� d S )Nz:memory:�{ )r> )r4 r; r r )r r r r
� CheckInvalidIsolationLevelTypeC s z.RegressionTests.CheckInvalidIsolationLevelTypec C sR t �d�}| �t|d� | �t|d� |�� }| �t|jd� | �t|jd� d S )Nz:memory:z select 1z select 1 z
select 2z select 2 )r r r4 rM r r )r r r r r r
�CheckNullCharacterI s
z"RegressionTests.CheckNullCharacterc
C s� t �d�}|�d� | �|jd� d}x�t|�d��D ]�\}}| j||d��p |�d|f� |�� |dkr~| �|d d� n2|dkr�| �|d d� n|d kr�| �|d d � |d7 }W d Q R X q6W | �|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
�CheckCommitCursorResetR s"
z&RegressionTests.CheckCommitCursorResetc C sB dd� }t �d�}t �|�}t�||�}|�|� ~~t�� d S )Nc W s d S )Nr )�argsr r r
�callbackv s z/RegressionTests.CheckBpo31770.<locals>.callbackz:memory:)r r rR �weakref�refrQ r Z
gc_collect)r rt r r rv r r r
�
CheckBpo31770q s
z RegressionTests.CheckBpo31770c C s | � t�� | j`W d Q R X d S )N)r4 �AttributeErrorr r> )r r r r
� CheckDelIsolation_levelSegfault� s z/RegressionTests.CheckDelIsolation_levelSegfaultN)$r@ rA rB r r
r r r r# r' �unittestZskipIfr Zsqlite_version_infor, r- r/ r2 r: r< rP rV rX r[ r] r^ r` rb rf ri rm ro rp rr rw ry r r r r
r ! s:
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 )r| r} )r rs �kwargsr r r
�__call__� s zUnhashableFunc.__call__)N)r@ rA rB �__hash__rQ r r r r r
r{ � s
r{ c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�UnhashableCallbacksTestCasec C s t �d�| _d S )Nz:memory:)r r r )r r r r
r � s z!UnhashableCallbacksTestCase.setUpc C s | j �� d S )N)r r )r r r r
r
� s z$UnhashableCallbacksTestCase.tearDownc C sL t dd�}| �td�� | j�|d� W d Q R X | j�d� | �|j� d S )Nr )r} zunhashable typerF zSELECT 1)r{ rU r; r Zset_progress_handlerr � assertFalser| )r �fr r r
�test_progress_handler� s
z1UnhashableCallbacksTestCase.test_progress_handlerc C st d}t � }| �td�� | j�|d|� W d Q R X d| }| �tj|�� | j�d| � W d Q R X | �|j � d S )N� func_namezunhashable typer zno such function: %sz SELECT %s())
r{ rU r; r Zcreate_functionr r+ r r� r| )r r� r� r9 r r r
� test_func� s z%UnhashableCallbacksTestCase.test_funcc C sL t tjd�}| �td�� | j�|� W d Q R X | j�d� | �|j � d S )N)r} zunhashable typezSELECT 1)
r{ r Z SQLITE_DENYrU r; r Zset_authorizerr r� r| )r r� r r r
�test_authorizer� s
z+UnhashableCallbacksTestCase.test_authorizerc C sz G dd� dt �}d}| �td�� | j�|d|ddi �� W d Q R X d| }| �tj|�� | j�d | � W d Q R X d S )
Nc @ s e Zd ZdZdS )z=UnhashableCallbacksTestCase.test_aggr.<locals>.UnhashableTypeN)r@ rA rB r� r r r r
�UnhashableType� s r� � aggr_namezunhashable typer ZAggrr zno such function: %sz SELECT %s())�typerU r; r Zcreate_aggregater r+ r )r r� r� r9 r r r
� test_aggr� s "z%UnhashableCallbacksTestCase.test_aggrN) r@ rA rB r r
r� r� r� r� r r r r
r� � s
r� c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�RecursiveUseOfCursorsz$Recursive use of cursors not allowedc C s@ t jdt jd�| _| j�� | _| j�d� | j�dddg� d S )Nz:memory:)r zcreate table test(x foo)z insert into test(x) values (?))rg )Zbar)r r r r r r r r )r r r r
r � s zRecursiveUseOfCursors.setUpc C s | j �� | j�� | ` | `d S )N)r r r )r r r r
r
� s
z RecursiveUseOfCursors.tearDownc
sV � fdd�}t �tjd|i��. � �tj� j�� � j�d� W d Q R X W d Q R X d S )Nc s � j �� j�S )N)r rQ r )r )r r r
�<lambda>� rG zBRecursiveUseOfCursors.test_recursive_cursor_init.<locals>.<lambda>ZINITz#select x as "x [INIT]", x from test) r �dictr �
convertersrU rS r9 r r )r �convr )r r
�test_recursive_cursor_init� s z0RecursiveUseOfCursors.test_recursive_cursor_initc
sV � fdd�}t �tjd|i��. � �tj� j�� � j�d� W d Q R X W d Q R X d S )Nc s
� j �� S )N)r r )r )r r r
r� � rG zCRecursiveUseOfCursors.test_recursive_cursor_close.<locals>.<lambda>ZCLOSEz$select x as "x [CLOSE]", x from test) r r� r r� rU rS r9 r r )r r� r )r r
�test_recursive_cursor_close� s z1RecursiveUseOfCursors.test_recursive_cursor_closec
sd g f� fdd� }t �tjd|i��8 � j�d� � �tj� j�� � j� � W d Q R X W d Q R X d S )Nc s |r� j �� S |�d �S )N)r r7 r% )r �l)r r r
r� � rG zCRecursiveUseOfCursors.test_recursive_cursor_fetch.<locals>.<lambda>ZITERz#select x as "x [ITER]", x from test)
r r� r r� r r rU rS r9 rT )r r� r )r r
�test_recursive_cursor_fetch� s
z1RecursiveUseOfCursors.test_recursive_cursor_fetchN) r@ rA rB r9 r r
r� r� r� r r r r
r� � s r� c C s( t �td�} t �| t �t�t �t�f�S )NZCheck)rz Z makeSuiter Z TestSuiter� r� )Zregression_suiter r r
�suite� s
r� c C s t �� } | �t� � d S )N)rz ZTextTestRunner�runr� )Zrunnerr r r
�test� s r� �__main__)r1 rz Zsqlite3r ru r� r Z
unittest.mockr ZTestCaser r{ r� r� r� r@ r r r r
�<module> s g 0&