File: //usr/local/lib/python3.7/sqlite3/test/__pycache__/regression.cpython-37.opt-2.pyc
B
��g�@ � @ 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
� 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 �3/usr/local/lib/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 ZPARSE_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>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 �� }|�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 �CheckColumnNameWithSpacesC 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 �#CheckStatementFinalizationOnCloseDbK 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 �CheckOnConflictRollbackY 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 �-CheckWorkaroundForBuggySqliteTransferBindingsh s z=RegressionTests.CheckWorkaroundForBuggySqliteTransferBindingsc C s | j �d� d S )N� )r r )r r r r �CheckEmptyStatementq 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 �CheckTypeMapUsagex 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)r2 � 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 rA z:memory:)r, �DEFERREDZ IMMEDIATEZ EXCLUSIVE)�levelrB � � �abczIMMEDIATE EXCLUSIVE� é) r$ r r r< �subTest�lower�
capitalizer9 �
ValueErrorr2 r )r rA rC 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> r? r@ rO r r r r �Cursor� s rP z:memory:z
select 4+5z$^Base Cursor\.__init__ not called\.$) r rP r r2 �ProgrammingErrorr �fetchall�assertRaisesRegexr )r rP 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> r? r@ r r r r �MyStr� s rU zselect ?rF )r$ r r )r rU 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 rO � s zPRegressionTests.CheckConnectionConstructorCallCheck.<locals>.Connection.__init__N)r> r? r@ rO r r r r �
Connection� s rX z:memory:)r rX r2 rQ r
)r rX 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 )rP r r r
� s zBRegressionTests.CheckCursorRegistration.<locals>.Connection.cursorN)r> r? r@ r
r )rP r r rX � s rX c @ s e Zd Zdd� ZdS )z7RegressionTests.CheckCursorRegistration.<locals>.Cursorc S s t j�| |� d S )N)r rP rO )r r r r r rO � s z@RegressionTests.CheckCursorRegistration.<locals>.Cursor.__init__N)r> r? r@ rO r r r r rP � s rP z:memory:zcreate table foo(x)z insert into foo(x) values (?))r& )� )r zselect x from foo)
r rX rP r
r r r r2 ZInterfaceErrorrR )r rX r r r )rP 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 r5 )r r �rowr r r �CheckPragmaAutocommit� s
z%RegressionTests.CheckPragmaAutocommitc C s | � tj| jd� d S )NrD )r2 r �Warningr )r r r r �CheckConnectionCall s z#RegressionTests.CheckConnectionCallc C s" dd� }| � tj| jjd|� d S )Nc S s dS )NrD r )�a�br r r � collation_cb s z4RegressionTests.CheckCollation.<locals>.collation_cbu �)r2 r rQ r Zcreate_collation)r rc 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 (?))rD rD )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 rQ r )r r re 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 9 s zKRegressionTests.CheckConvertTimestampMicrosecondPadding.<locals>.<listcomp>i� rZ � � r i@� i@� )r r r. r
r rR 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 �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 r2 rK r
r )r r r r r r �CheckNullCharacterF 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 (?)rD r' r& z'should have returned exactly three rows) r r Z
executescriptr r< � enumerater rH r( )r r Zcounterr r] r r r �CheckCommitCursorResetO s"
z&RegressionTests.CheckCommitCursorResetc C sB dd� }t �d�}t �|�}t�||�}|�|� ~~t�� d S )Nc W s d S )Nr )�argsr r r �callbacks s z/RegressionTests.CheckBpo31770.<locals>.callbackz:memory:)r r rP �weakref�refrO r Z
gc_collect)r rr r r rt r r r �
CheckBpo31770n s
z RegressionTests.CheckBpo31770c C s | � t�� | j`W d Q R X d S )N)r2 �AttributeErrorr r< )r r r r � CheckDelIsolation_levelSegfault~ s z/RegressionTests.CheckDelIsolation_levelSegfaultN)$r> r? r@ r
r r r r r! r% �unittestZskipIfr Zsqlite_version_infor* r+ r- r0 r8 r: rN rT rV rY r[ r\ r^ r` rd rg rk rm rn rp ru rw 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 rO � s zUnhashableFunc.__init__c O s | j d7 _ | jS )NrD )rz r{ )r rq �kwargsr r r �__call__� s zUnhashableFunc.__call__)N)r> r? r@ �__hash__rO 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 �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 typerD zSELECT 1)ry rS r9 r Zset_progress_handlerr � assertFalserz )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())
ry rS 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�}| �td�� | j�|� W d Q R X | j�d� | �|j � d S )N)r{ zunhashable typezSELECT 1)
ry r Z SQLITE_DENYrS r9 r Zset_authorizerr r� rz )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? r@ r~ r r r r �UnhashableType� s r� � aggr_namezunhashable typer ZAggrr zno such function: %sz SELECT %s())�typerS 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 �td�} t �| t �t�f�S )NZCheck)rx Z makeSuiter Z TestSuiter )Zregression_suiter r r �suite� s r� c C s t �� } | �t� � d S )N)rx ZTextTestRunner�runr� )Zrunnerr r r �test� s r� �__main__)
r/ rx Zsqlite3r rs r� r ZTestCaser ry r r� r> r r r r �<module> s g 0