File: //opt/python38/lib/python3.8/sqlite3/test/__pycache__/regression.cpython-38.opt-2.pyc
U
�i�_�: � @ sh d dl Z d dlZd dlZd dlZd dlZd dlmZ G dd� dej�Z dd� Z
dd� Ze d krde� 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=d>� Z$d?S )@�RegressionTestsc C s t �d�| _d S )N�:memory:)�sqlite�connect�con��self� r
�6/opt/python38/lib/python3.8/sqlite3/test/regression.py�setUp s zRegressionTests.setUpc C s | j �� d S �N)r �closer 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 5 |�� |�� X d S )Nr �Z 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� td�D ]"}|d �d d
d� td�D �� q:td�D ]}|| �d | d � qf� �� d S )
Nr � )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 = s � zselect x from test)r r �ranger � executemany�rollback)r �cursors�ir
r r �CheckStatementReset5 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� descriptionr r
r
r �CheckColumnNameWithSpacesD s
z)RegressionTests.CheckColumnNameWithSpacesc C sL t �d�}g }td�D ](}|�� }|�|� |�dt|� � q|�� d S )Nr �i zselect 1 x union select )r r r r �appendr �strr )r r r# r$ r r
r
r �#CheckStatementFinalizationOnCloseDbL s
z3RegressionTests.CheckStatementFinalizationOnCloseDb)� � r. zneeds sqlite 3.2.2 or newerc C s� t �d�}|�d� |�d� z|�d� W n t jk
rB Y nX |�d� z |�� W n t jk
rz | �d� Y nX d S )Nr 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 �CheckOnConflictRollbackZ s
z'RegressionTests.CheckOnConflictRollbackc C s( | j �d� | j �d� | j �d� d S )N�create table foo(bar)�drop table foo�r r r r
r
r �-CheckWorkaroundForBuggySqliteTransferBindingsi s z=RegressionTests.CheckWorkaroundForBuggySqliteTransferBindingsc C s | j �d� d S �N� r5 r r
r
r �CheckEmptyStatementr s z#RegressionTests.CheckEmptyStatementc C sh d}t jdt jd�}|�d� |�dtj�� f� |�|� |�d� |�d� |�d� |�|� d S ) Nzselect * from foor r z create table foo(bar timestamp)z insert into foo(bar) values (?)r4 z create table foo(bar integer)� insert into foo(bar) values (5))r r �PARSE_DECLTYPESr �datetimeZnow)r ZSELECTr r
r
r �CheckTypeMapUsagey s
z!RegressionTests.CheckTypeMapUsagec C sP | � tj��$}| j�dttdg��f��� W 5 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 r0 r r �bytes� bytearray�fetchoneZassertInr+ � exception)r �cm�msgr
r
r �CheckErrorMsgDecodeError� s �z(RegressionTests.CheckErrorMsgDecodeErrorc C s | � ttji d � d S r
)r? � TypeErrorr Zregister_adapterr r
r
r �CheckRegisterAdapter� s z$RegressionTests.CheckRegisterAdapterc s� G � fdd�dt �}t�d�� d � _dD ]@}| j|d��* |� _|�� � _|�� � _||�� _W 5 Q R X q(d � _d� _dtfdtfd tfd
tfd tfg}|D ]J\}}| j|d��0 | � |�� |� _W 5 Q R X | �
� jd� W 5 Q R X q�d S ) Nc s e Zd Zdd� Z� fdd�ZdS )z9RegressionTests.CheckSetIsolationLevel.<locals>.CustomStrc S s d S r
r
r r
r
r �upper� s z?RegressionTests.CheckSetIsolationLevel.<locals>.CustomStr.upperc s
d� _ d S r7 ��isolation_levelr r r
r �__del__� s zARegressionTests.CheckSetIsolationLevel.<locals>.CustomStr.__del__N)�__name__�
__module__� __qualname__rI rL r
r r
r � CustomStr� s rP r )r8 �DEFERREDZ IMMEDIATEZ EXCLUSIVE)�levelrQ � � �abczIMMEDIATE EXCLUSIVE� é) r+ r r rK �subTest�lower�
capitalizerG �
ValueErrorr? r& )r rP rR Zpairs�value�excr
r r �CheckSetIsolationLevel� s,
� z&RegressionTests.CheckSetIsolationLevelc C sp G dd� dt j�}t �d�}||�}| �t j�� |�d��� W 5 Q R X | �t jd�� |�� W 5 Q R X d S )Nc @ s e Zd Zdd� ZdS )z?RegressionTests.CheckCursorConstructorCallCheck.<locals>.Cursorc S s d S r
r
r1 r
r
r �__init__� s zHRegressionTests.CheckCursorConstructorCallCheck.<locals>.Cursor.__init__N�rM rN rO r^ r
r
r
r �Cursor� s r` r z
select 4+5z$^Base Cursor\.__init__ not called\.$) r r` r r? �ProgrammingErrorr �fetchallZassertRaisesRegexr )r r` 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)rM rN rO r
r
r
r �MyStr� s rd zselect ?rU )r+ r r )r rd r
r
r �CheckStrSubclass� s z RegressionTests.CheckStrSubclassc C s> G dd� dt j�}|d�}| �t j�� |�� }W 5 Q R X d S )Nc @ s e Zd Zdd� ZdS )zGRegressionTests.CheckConnectionConstructorCallCheck.<locals>.Connectionc S s d S r
r
)r �namer
r
r r^ � s zPRegressionTests.CheckConnectionConstructorCallCheck.<locals>.Connection.__init__Nr_ r
r
r
r �
Connection� s rg r )r rg r? ra r �r rg 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 5 Q R X d S ) Nc s e Zd Z� fdd�ZdS )z;RegressionTests.CheckCursorRegistration.<locals>.Connectionc s � | �S r
r
r �r` r
r r � s zBRegressionTests.CheckCursorRegistration.<locals>.Connection.cursorN)rM rN rO r r
rj r
r rg � s rg c @ s e Zd Zdd� ZdS )z7RegressionTests.CheckCursorRegistration.<locals>.Cursorc S s t j�| |� d S r
)r r` r^ r1 r
r
r r^ � s z@RegressionTests.CheckCursorRegistration.<locals>.Cursor.__init__Nr_ r
r
r
r r` � s r` r zcreate table foo(x)z insert into foo(x) values (?))r- )� )r zselect x from foo)
r rg r` r r r! r" r? ZInterfaceErrorrb rh r
rj r �CheckCursorRegistration� s
z'RegressionTests.CheckCursorRegistrationc C s t jdd d�}d S )Nr rJ )r r r1 r
r
r �CheckAutoCommit� s z RegressionTests.CheckAutoCommitc C s4 | j �� }|�d� |�d� |�d� |�� }d S )Nr3 r: zpragma page_size)r r r rB )r r �rowr
r
r �CheckPragmaAutocommit� s
z%RegressionTests.CheckPragmaAutocommitc C s | � tj| jd� d S �NrS )r? r �Warningr r r
r
r �CheckConnectionCall s z#RegressionTests.CheckConnectionCallc C s" dd� }| � tj| jjd|� d S )Nc S s dS rp r
)�a�br
r
r � collation_cb s z4RegressionTests.CheckCollation.<locals>.collation_cbu �)r? r ra r Zcreate_collation)r ru r
r
r �CheckCollation s
�z RegressionTests.CheckCollationc sf t �d�}|�� � � �d� � �d� � fdd�}| �t j�� � �ddd� |� D �� W 5 Q R X d S ) Nr zcreate table a (bar)zcreate table b (baz)c 3 s � � dd� dV d S )Nz insert into a (bar) values (?))rS rS )r r
�r r
r �foo s z4RegressionTests.CheckRecursiveCursorUse.<locals>.fooz insert into b (baz) values (?)c s s | ] }|fV qd S r
r
)r r$ r
r
r � <genexpr>% s z:RegressionTests.CheckRecursiveCursorUse.<locals>.<genexpr>)r r r r r? ra r! )r r rx r
rw 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 )Nr 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 : s zKRegressionTests.CheckConvertTimestampMicrosecondPadding.<locals>.<listcomp>i� rk � � r i@� i@� )r r r; r r rb r& r< )r r r �valuesr
r
r �'CheckConvertTimestampMicrosecondPadding' s
�z7RegressionTests.CheckConvertTimestampMicrosecondPaddingc C s | j ttjddd� d S )Nr �{ rJ )r? rG r r r r
r
r � CheckInvalidIsolationLevelTypeA s
�z.RegressionTests.CheckInvalidIsolationLevelTypec C sR t �d�}| �t|d� | �t|d� |�� }| �t|jd� | �t|jd� d S )Nr z select 1z select 1 z
select 2z select 2 )r r r? rZ r r r r
r
r �CheckNullCharacterG s
z"RegressionTests.CheckNullCharacterc
C s� t �d�}|�d� | �|jd� d}t|�d��D ]�\}}| j||d��p |�d|f� |�� |dkr|| �|d d� n2|dkr�| �|d d� n|d kr�| �|d d � |d7 }W 5 Q R X q4| �|d
d � d S ) Nr z�
create table t(c);
create table t2(c);
insert into t values(0);
insert into t values(1);
insert into t values(2);
r8 r zselect c from t)r$ rn z insert into t2(c) values (?)rS r. r- z'should have returned exactly three rows) r r Z
executescriptr& rK � enumerater rW r/ )r r Zcounterr$ rn r
r
r �CheckCommitCursorResetP s
z&RegressionTests.CheckCommitCursorResetc C sB dd� }t �d�}t �|�}t�||�}|�|� ~~t�� d S )Nc W s d S r
r
)�argsr
r
r �callbackt s z/RegressionTests.CheckBpo31770.<locals>.callbackr )r r r` �weakref�refr^ r �
gc_collect)r r� r r r� r
r
r �
CheckBpo31770o s
z RegressionTests.CheckBpo31770c C s | � t�� | j`W 5 Q R X d S r
)r? �AttributeErrorr rK r r
r
r � CheckDelIsolation_levelSegfault s z/RegressionTests.CheckDelIsolation_levelSegfaultc C sf G dd� d�}| j jtj| j jdd�| j jfD ]2}|� }||j� ||j� | j �d� |d � q.d S )Nc @ s e Zd Zdd� ZdS )z.RegressionTests.CheckBpo37347.<locals>.Printerc W s t jS r
)r Z SQLITE_OK)r r� r
r
r �log� s z2RegressionTests.CheckBpo37347.<locals>.Printer.logN)rM rN rO r� r
r
r
r �Printer� s r� rS )�nzselect 1)r Zset_trace_callback� functools�partialZset_progress_handlerZset_authorizerr� r )r r� �methodZprinter_instancer
r
r �
CheckBpo37347� s �
z RegressionTests.CheckBpo37347N)%rM rN rO r r r r r% r( r, �unittestZskipIfr Zsqlite_version_infor2 r6 r9 r= rF rH r] rc re ri rl rm ro rr rv rz r~ r� r� r� r� r� r� r
r
r
r r s>
r c C s t �td�} t �| f�S )NZCheck)r� Z makeSuiter Z TestSuite)Zregression_suiter
r
r �suite� s �r� c C s t �� } | �t� � d S r
)r� ZTextTestRunner�runr� )Zrunnerr
r
r �test� s r� �__main__) r< r� Zsqlite3r r� r� r� r ZTestCaser r� rM r
r
r
r �<module> s v