Your IP : 216.73.216.213
B
� f5G � @ 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 � assertEqualZ 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� dS )zm
pysqlite would crash with older SQLite versions unless
a workaround is implemented.
zcreate table foo(bar)zdrop table fooN)r r )r r r r
�-CheckWorkaroundForBuggySqliteTransferBindingsk s z=RegressionTests.CheckWorkaroundForBuggySqliteTransferBindingsc C s | j �d� dS )z
pysqlite used to segfault with SQLite versions 3.5.x. These return NULL
for "no-operation" statements
� 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 )
z�
pysqlite until 2.4.1 did not rebuild the row_cast_map when recompiling
a statement. This test exhibits the problem.
zselect * 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)N)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� dS )z!
See issue 3312.
N)r3 � 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 rB z:memory:)r- �DEFERREDZ IMMEDIATEZ EXCLUSIVE)�levelrC � � �abczIMMEDIATE EXCLUSIVE� é) r% r r r= �subTest�lower�
capitalizer: �
ValueErrorr3 r! )r rB rD Zpairs�value�excr )r r
�CheckSetIsolationLevel� s$
z&RegressionTests.CheckSetIsolationLevelc C sp G dd� dt j�}t �d�}||�}| �t j�� |�d��� W dQ R X | �t jd�� |�� W dQ R X dS )zd
Verifies that cursor methods check whether base class __init__ was
called.
c @ 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@ rA rP r r r r
�Cursor� s rQ z:memory:z
select 4+5Nz$^Base Cursor\.__init__ not called\.$) r rQ r r3 �ProgrammingErrorr �fetchall�assertRaisesRegexr )r rQ r r r r r
� CheckCursorConstructorCallCheck� s
z/RegressionTests.CheckCursorConstructorCallCheckc C s( G dd� dt �}| j�d|d�f� dS )za
The Python 3.0 port of the module didn't cope with values of subclasses of str.
c @ s e Zd ZdS )z/RegressionTests.CheckStrSubclass.<locals>.MyStrN)r? r@ rA r r r r
�MyStr� s rV zselect ?rG N)r% r r )r rV r r r
�CheckStrSubclass� s z RegressionTests.CheckStrSubclassc C s> G dd� dt j�}|d�}| �t j�� |�� }W dQ R X dS )zh
Verifies that connection methods check whether base class __init__ was
called.
c @ s e Zd Zdd� ZdS )zGRegressionTests.CheckConnectionConstructorCallCheck.<locals>.Connectionc S s d S )Nr )r �namer r r
rP � s zPRegressionTests.CheckConnectionConstructorCallCheck.<locals>.Connection.__init__N)r? r@ rA rP r r r r
�
Connection� s rY z:memory:N)r rY r3 rR r )r rY 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 )
z�
Verifies that subclassed cursor classes are correctly registered with
the connection object, too. (fetch-across-rollback problem)
c s e Zd Z� fdd�ZdS )z;RegressionTests.CheckCursorRegistration.<locals>.Connectionc s � | �S )Nr )r )rQ r r
r � s zBRegressionTests.CheckCursorRegistration.<locals>.Connection.cursorN)r? r@ rA r r )rQ r r
rY � s rY c @ s e Zd Zdd� ZdS )z7RegressionTests.CheckCursorRegistration.<locals>.Cursorc S s t j�| |� d S )N)r rQ rP )r r r r r
rP � s z@RegressionTests.CheckCursorRegistration.<locals>.Cursor.__init__N)r? r@ rA rP r r r r
rQ � s rQ z:memory:zcreate table foo(x)z insert into foo(x) values (?))r' )� )r zselect x from fooN)
r rY rQ r r r r r3 ZInterfaceErrorrS )r rY r r r )rQ r
�CheckCursorRegistration� s
z'RegressionTests.CheckCursorRegistrationc C s t jddd�}dS )z�
Verifies that creating a connection in autocommit mode works.
2.5.3 introduced a regression so that these could no longer
be created.
z:memory:N)r= )r r )r r r r r
�CheckAutoCommit� s z RegressionTests.CheckAutoCommitc C s4 | j �� }|�d� |�d� |�d� |�� }dS )z�
Verifies that running a PRAGMA statement that does an autocommit does
work. This did not work in 2.5.3/2.5.4.
zcreate table foo(bar)z insert into foo(bar) values (5)zpragma page_sizeN)r r r r6 )r r �rowr r r
�CheckPragmaAutocommit� s
z%RegressionTests.CheckPragmaAutocommitc C s | � tj| jd� dS )z}
Call a connection with a non-string SQL request: check error handling
of the statement constructor.
rE N)r3 r �Warningr )r r r r
�CheckConnectionCall s z#RegressionTests.CheckConnectionCallc C s" dd� }| � tj| jjd|� d S )Nc S s dS )NrE r )�a�br r r
� collation_cb s z4RegressionTests.CheckCollation.<locals>.collation_cbu �)r3 r rR r Zcreate_collation)r rd 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 )
z�
http://bugs.python.org/issue10811
Recursively using a cursor, such as when reusing it from a generator led to segfaults.
Now we catch recursive cursor usage and raise a ProgrammingError.
z:memory:zcreate table a (bar)zcreate table b (baz)c 3 s � � dd� dV d S )Nz insert into a (bar) values (?))rE rE )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>N)r r r r r3 rR r )r r rf 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� dS )z�
http://bugs.python.org/issue14720
The microsecond parsing of convert_timestamp() should pad with zeros,
since the microsecond string "456" actually represents "456000".
z: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@� N)r r r/ r r rS r! r0 )r r r �valuesr r r
�'CheckConvertTimestampMicrosecondPadding) s
z7RegressionTests.CheckConvertTimestampMicrosecondPaddingc C s | j ttjddd� d S )Nz:memory:�{ )r= )r3 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 r3 rL 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 )
z�
Connection.commit() did reset cursors, which made sqlite3
to return rows multiple times when fetched from cursors
after commit. See issues 10513 and 23129 for details.
z: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 (?)rE r( Nr' z'should have returned exactly three rows) r r Z
executescriptr! r= � enumerater rI r) )r r Zcounterr r^ r r r
�CheckCommitCursorResetR s"
z&RegressionTests.CheckCommitCursorResetc C sB dd� }t �d�}t �|�}t�||�}|�|� ~~t�� dS )zm
The interpreter shouldn't crash in case Cursor.__init__() is called
more than once.
c W s d S )Nr )�argsr r r
�callbackv s z/RegressionTests.CheckBpo31770.<locals>.callbackz:memory:N)r r rQ �weakref�refrP r Z
gc_collect)r rs r r ru r r r
�
CheckBpo31770q s
z RegressionTests.CheckBpo31770c C s | � t�� | j`W d Q R X d S )N)r3 �AttributeErrorr r= )r r r r
� CheckDelIsolation_levelSegfault� s z/RegressionTests.CheckDelIsolation_levelSegfaultN)$r? r@ rA r r
r r r r" r& �unittestZskipIfr Zsqlite_version_infor+ r, r. r1 r9 r; rO rU rW rZ r\ r] r_ ra re rh rl rn ro rq rv rx 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
rP � s zUnhashableFunc.__init__c O s | j d7 _ | jS )NrE )r{ r| )r rr �kwargsr r r
�__call__� s zUnhashableFunc.__call__)N)r? r@ rA �__hash__rP r~ r r r r
rz � s
rz c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�UnhashableCallbacksTestCasez�
https://bugs.python.org/issue34052
Registering unhashable callbacks raises TypeError, callbacks are not
registered in SQLite after such registration attempt.
c 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 typerE zSELECT 1)rz rT 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())
rz rT r: r Zcreate_functionr r* r r� r{ )r r� r� r8 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)
rz r Z SQLITE_DENYrT 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? r@ rA r r r r r
�UnhashableType� s r� � aggr_namezunhashable typer ZAggrr zno such function: %sz SELECT %s())�typerT r: r Zcreate_aggregater r* r )r r� r� r8 r r r
� test_aggr� s "z%UnhashableCallbacksTestCase.test_aggrN)
r? r@ rA �__doc__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 (?))rf )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 rP r )r )r r r
�<lambda>� rF zBRecursiveUseOfCursors.test_recursive_cursor_init.<locals>.<lambda>ZINITz#select x as "x [INIT]", x from test) r �dictr �
convertersrT rR r8 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� � rF zCRecursiveUseOfCursors.test_recursive_cursor_close.<locals>.<lambda>ZCLOSEz$select x as "x [CLOSE]", x from test) r r� r r� rT rR r8 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 r6 r$ )r �l)r r r
r� � rF zCRecursiveUseOfCursors.test_recursive_cursor_fetch.<locals>.<lambda>ZITERz#select x as "x [ITER]", x from test)
r r� r r� r r rT rR r8 rS )r r� r )r r
�test_recursive_cursor_fetch� s
z1RecursiveUseOfCursors.test_recursive_cursor_fetchN) r? r@ rA r8 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)ry Z makeSuiter Z TestSuiter� r� )Zregression_suiter r r
�suite� s
r� c C s t �� } | �t� � d S )N)ry ZTextTestRunner�runr� )Zrunnerr r r
�test� s r� �__main__)r0 ry Zsqlite3r rt r� r Z
unittest.mockr ZTestCaser rz r� r� r� r? r r r r
�<module> s g 0&