Your IP : 216.73.216.213
U
i�f�C � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlm Z G dd� dej
�Z G dd� dej
�Z dd � Z
d
d � Zed kr�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=d>� Z$d?d@� Z%dAS )B�RegressionTestsc C s t �d�| _d S )N�:memory:)�sqlite�connect�con��self� r �</opt/alt/python38/lib64/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 �PARSE_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)�
� 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 �CheckStatementReset8 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 descriptionr r r r �CheckColumnNameWithSpacesG 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 �#CheckStatementFinalizationOnCloseDbO s
z3RegressionTests.CheckStatementFinalizationOnCloseDb)� � r0 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 �CheckOnConflictRollback] 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 �-CheckWorkaroundForBuggySqliteTransferBindingsl s z=RegressionTests.CheckWorkaroundForBuggySqliteTransferBindingsc C s | j �d� d S �N� r7 r r r r �CheckEmptyStatementu 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 (?)r6 z create table foo(bar integer)� insert into foo(bar) values (5))r r �PARSE_DECLTYPESr �datetimeZnow)r
ZSELECTr r r r �CheckTypeMapUsage| s
z!RegressionTests.CheckTypeMapUsagec s\ G � fdd�d�}|� dg� t jdt jd�}|�d� | �t�� |�d� � W 5 Q R X d S )Nc s e Zd Z� fdd�ZdS )z0RegressionTests.CheckBindMutatingList.<locals>.Xc s � � � dS )Nz...)�clear)r
Zprotocol�Z
parametersr r � __conform__� s z<RegressionTests.CheckBindMutatingList.<locals>.X.__conform__N)�__name__�
__module__� __qualname__rB r rA r r �X� s rF r r r z$create table foo(bar X, baz integer)z'insert into foo(bar, baz) values (?, ?))r r r= r � assertRaises�
IndexError)r
rF r r rA r �CheckBindMutatingList� s
z%RegressionTests.CheckBindMutatingListc 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) rG r r2 r r �bytes� bytearray�fetchoneZassertInr- Z exception)r
�cm�msgr r r �CheckErrorMsgDecodeError� s �z(RegressionTests.CheckErrorMsgDecodeErrorc C s | � ttji d � d S r )rG � 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 r9 ��isolation_levelr r r r �__del__� s zARegressionTests.CheckSetIsolationLevel.<locals>.CustomStr.__del__N)rC rD rE rS rV r r r r � CustomStr� s rW r )r: �DEFERREDZ IMMEDIATEZ EXCLUSIVE)�levelrX � � �abczIMMEDIATE EXCLUSIVE� é) r- r r rU �subTest�lower�
capitalizerQ �
ValueErrorrG r) )r
rW rY 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 r3 r r r �__init__� s zHRegressionTests.CheckCursorConstructorCallCheck.<locals>.Cursor.__init__N�rC rD rE re r r r r �Cursor� s rg r z
select 4+5z$^Base Cursor\.__init__ not called\.$) r rg r rG �ProgrammingErrorr �fetchall�assertRaisesRegexr )r
rg 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)rC rD rE r r r r �MyStr� s rl zselect ?r\ )r- r r )r
rl 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 re � s zPRegressionTests.CheckConnectionConstructorCallCheck.<locals>.Connection.__init__Nrf r r r r �
Connection� s ro r )r ro rG rh r �r
ro 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 �rg r r r � s zBRegressionTests.CheckCursorRegistration.<locals>.Connection.cursorN)rC rD rE r r rr r r ro � s ro c @ s e Zd Zdd� ZdS )z7RegressionTests.CheckCursorRegistration.<locals>.Cursorc S s t j�| |� d S r )r rg re r3 r r r re � s z@RegressionTests.CheckCursorRegistration.<locals>.Cursor.__init__Nrf r r r r rg � s rg r zcreate table foo(x)z insert into foo(x) values (?))r/ )� )r zselect x from foo)
r ro rg r r r$ r% rG ZInterfaceErrorri rp r rr r �CheckCursorRegistration� s
z'RegressionTests.CheckCursorRegistrationc C s t jdd d�}d S )Nr rT )r r r3 r r r �CheckAutoCommit s z RegressionTests.CheckAutoCommitc C s4 | j �� }|�d� |�d� |�d� |�� }d S )Nr5 r< zpragma page_size)r r r rM )r
r �rowr r r �CheckPragmaAutocommit s
z%RegressionTests.CheckPragmaAutocommitc C s | � tj| jd� d S �NrZ )rG r �Warningr r r r r �CheckConnectionCall s z#RegressionTests.CheckConnectionCallc C s" dd� }| � tj| jjd|� d S )Nc S s dS rx r )�a�br r r � collation_cb s z4RegressionTests.CheckCollation.<locals>.collation_cbu �)rG r rh r Zcreate_collation)r
r} 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 (?))rZ rZ )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>5 s z:RegressionTests.CheckRecursiveCursorUse.<locals>.<genexpr>)r r r r rG rh r$ )r
r r� 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 )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 J s zKRegressionTests.CheckConvertTimestampMicrosecondPadding.<locals>.<listcomp>i� rs � � r i@� i@� )r r r= r r ri r) r> )r
r r �valuesr r r �'CheckConvertTimestampMicrosecondPadding7 s
�z7RegressionTests.CheckConvertTimestampMicrosecondPaddingc C s | j ttjddd� d S )Nr �{ rT )rG rQ r r r r r r � CheckInvalidIsolationLevelTypeQ 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 rG ra r r r r r r �CheckNullCharacterW 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);
r: r zselect c from t)r' rv z insert into t2(c) values (?)rZ r0 r/ z'should have returned exactly three rows) r r Z
executescriptr) rU � enumerater r^ r1 )r
r Zcounterr' rv r r r �CheckCommitCursorReset` s
z&RegressionTests.CheckCommitCursorResetc C sB dd� }t �d�}t �|�}t�||�}|�|� ~~t�� d S )Nc W s d S r r )�argsr r r �callback� s z/RegressionTests.CheckBpo31770.<locals>.callbackr )r r rg �weakref�refre r Z
gc_collect)r
r� r r r� r r r �
CheckBpo31770 s
z RegressionTests.CheckBpo31770c C s | � t�� | j`W 5 Q R X d S r )rG �AttributeErrorr rU 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)rC rD rE r� r r r r �Printer� s r� rZ )�nzselect 1)r Zset_trace_callback� functools�partialZset_progress_handlerZset_authorizerr� r )r
r� �methodZprinter_instancer r r �
CheckBpo37347� s �
z RegressionTests.CheckBpo37347N)&rC rD rE r
r r r r( r* r. �unittestZskipIfr Zsqlite_version_infor4 r8 r; r? rI rP rR rd rk rm rq rt ru rw rz r~ r� r� 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 )Nr r zcreate table test(x foo)r! )r� )Zbar)r r r r r r r r$ r r r r r
� s � �zRecursiveUseOfCursors.setUpc C s | j �� | j�� | ` | `d S r )r r r r r r r r � s
z RecursiveUseOfCursors.tearDownc
sV � fdd�}t �tjd|i��. � �tj� j�� � j�d� W 5 Q R X W 5 Q R X d S )Nc s � j �� j�S r )r re r �r r r r �<lambda>� r[ zBRecursiveUseOfCursors.test_recursive_cursor_init.<locals>.<lambda>ZINITz#select x as "x [INIT]", x from test� r �dictr �
convertersrj rh rO r r �r
Zconvr r r �test_recursive_cursor_init� s z0RecursiveUseOfCursors.test_recursive_cursor_initc
sV � fdd�}t �tjd|i��. � �tj� j�� � j�d� W 5 Q R X W 5 Q R X d S )Nc s
� j �� S r )r r r� r r r r� � r[ zCRecursiveUseOfCursors.test_recursive_cursor_close.<locals>.<lambda>ZCLOSEz$select x as "x [CLOSE]", x from testr� 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 5 Q R X W 5 Q R X d S )Nc s |r� j �� S |�d �S r )r rM r, )r �lr r r r� � r[ zCRecursiveUseOfCursors.test_recursive_cursor_fetch.<locals>.<lambda>ZITERz#select x as "x [ITER]", x from test)
r r� r r� r r rj rh rO ri r� r r r �test_recursive_cursor_fetch� s
z1RecursiveUseOfCursors.test_recursive_cursor_fetchN) rC rD rE rO r
r r� r� r� r r r r r� � s r� c C s$ t �td�} t �t�}t �| |f�S )NZCheck)r� Z makeSuiter r� Z TestSuite)Zregression_suiteZrecursive_cursorr 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 Z
unittest.mockr ZTestCaser r� r� rC r r r r �<module> s &