Sindbad~EG File Manager
a
��g�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 �3/usr/local/lib/python3.9/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 sP t jdt jd�}z(| j�� }|�d� W |�� |�� n|�� |�� 0 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>= � 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 @ r � 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 � assertEqual� 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)� � r2 zneeds sqlite 3.2.2 or newerc C s| t �d�}|�d� |�d� z|�d� W n t jy@ Y n0 |�d� z |�� W n t jyv | �d� Y n0 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� r9 r r r r �CheckEmptyStatementu s z#RegressionTests.CheckEmptyStatementc C sp 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 (?)r8 z create table foo(bar integer)� insert into foo(bar) values (5))r r �PARSE_DECLTYPESr �datetimeZnowr )r
ZSELECTr r r r �CheckTypeMapUsage| s
z!RegressionTests.CheckTypeMapUsagec sp G � fdd�d�}|� dg� t jdt jd�}|�d� | �t�� |�d� � W d � n1 sb0 Y d S )Nc s e Zd Z� fdd�ZdS )z0RegressionTests.CheckBindMutatingList.<locals>.Xc s � � � dS )Nz...)�clear)r
�protocol��
parametersr r � __conform__� s z<RegressionTests.CheckBindMutatingList.<locals>.X.__conform__N)�__name__�
__module__� __qualname__rF r rD r r �X� s rJ 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
rJ r r rD r �CheckBindMutatingList� s
z%RegressionTests.CheckBindMutatingListc C sd | � tj��.}| j�dttdg��f��� W d � n1 s@0 Y d}| �|t |j
�� d S )Nz"select 'xxx' || ? || 'yyy' colname� z9Could not decode to UTF-8 column 'colname' with text 'xxx) rK r r4 r r �bytes� bytearray�fetchoneZassertInr/ � exception)r
�cm�msgr r r �CheckErrorMsgDecodeError� s �&z(RegressionTests.CheckErrorMsgDecodeErrorc C s | � ttji d � d S r )rK � TypeErrorr Zregister_adapterr r r r �CheckRegisterAdapter� s z$RegressionTests.CheckRegisterAdapterc
s* G � fdd�dt �}t�d�� d � _dD ]T}| j|d��4 |� _|�� � _|�� � _||�� _W d � q(1 sr0 Y q(d � _d� _dtfdtfd tfd
tfd tfg}|D ]t\}}| j|d��N | � |�� |� _W d � n1 s�0 Y | �
� jd� W d � q�1 �s0 Y 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 r; ��isolation_levelr r r r �__del__� s zARegressionTests.CheckSetIsolationLevel.<locals>.CustomStr.__del__N)rG rH rI rX r[ r r r r � CustomStr� s r\ r )r< �DEFERREDZ IMMEDIATEZ EXCLUSIVE)�levelr] � r �abczIMMEDIATE EXCLUSIVE� é) r/ r r rZ �subTest�lower�
capitalizerV �
ValueErrorrK r* )r
r\ r^ �pairs�value�excr r r �CheckSetIsolationLevel� s&
* � $z&RegressionTests.CheckSetIsolationLevelc C s� G dd� dt j�}t �d�}||�}| �t j�� |�d��� W d � n1 sT0 Y | �t jd�� |�� W d � n1 s�0 Y d S )Nc @ s e Zd Zdd� ZdS )z?RegressionTests.CheckCursorConstructorCallCheck.<locals>.Cursorc S s d S r r r5 r r r �__init__� s zHRegressionTests.CheckCursorConstructorCallCheck.<locals>.Cursor.__init__N�rG rH rI rj r r r r �Cursor� s rl r z
select 4+5z$^Base Cursor\.__init__ not called\.$) r rl r rK �ProgrammingErrorr �fetchall�assertRaisesRegexr )r
rl 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)rG rH rI r r r r �MyStr� r rq zselect ?r` )r/ r r )r
rq r r r �CheckStrSubclass� s z RegressionTests.CheckStrSubclassc C sR G dd� dt j�}|d�}| �t j�� |�� }W d � n1 sD0 Y d S )Nc @ s e Zd Zdd� ZdS )zGRegressionTests.CheckConnectionConstructorCallCheck.<locals>.Connectionc S s d S r r )r
�namer r r rj � s zPRegressionTests.CheckConnectionConstructorCallCheck.<locals>.Connection.__init__Nrk r r r r �
Connection� s rt r )r rt rK rm r �r
rt r r r r r �#CheckConnectionConstructorCallCheck� s z3RegressionTests.CheckConnectionConstructorCallCheckc s� G � fdd�dt j�}G dd� dt j�� |d�}|�� }|�d� |�dg d�� |�d � |�� | �t j�� |� � W d � n1 s�0 Y d S )
Nc s e Zd Z� fdd�ZdS )z;RegressionTests.CheckCursorRegistration.<locals>.Connectionc s � | �S r r r �rl r r r � s zBRegressionTests.CheckCursorRegistration.<locals>.Connection.cursorN)rG rH rI r r rw r r rt � s rt c @ s e Zd Zdd� ZdS )z7RegressionTests.CheckCursorRegistration.<locals>.Cursorc S s t j�| |� d S r )r rl rj r5 r r r rj � s z@RegressionTests.CheckCursorRegistration.<locals>.Cursor.__init__Nrk r r r r rl � s rl r zcreate table foo(x)z insert into foo(x) values (?)))r1 )� )r zselect x from foo)
r rt rl r r r% r& rK ZInterfaceErrorrn ru r rw r �CheckCursorRegistration� s
z'RegressionTests.CheckCursorRegistrationc C s t jdd d�}d S )Nr rY )r r r5 r r r �CheckAutoCommit s z RegressionTests.CheckAutoCommitc C s4 | j �� }|�d� |�d� |�d� |�� }d S )Nr7 r> zpragma page_size)r r r rQ )r
r �rowr r r �CheckPragmaAutocommit s
z%RegressionTests.CheckPragmaAutocommitc C s | � t| jd� d S �Nr_ )rK rV r r r r r �CheckConnectionCall s z#RegressionTests.CheckConnectionCallc C s" dd� }| � tj| jjd|� d S )Nc S s dS r} r )�a�br r r � collation_cb s z4RegressionTests.CheckCollation.<locals>.collation_cbu �)rK r rm r Zcreate_collation)r
r� r r r �CheckCollation s �z RegressionTests.CheckCollationc sz t �d�}|�� � � �d� � �d� � fdd�}| �t j��( � �ddd� |� D �� W d � n1 sl0 Y d S ) Nr zcreate table a (bar)zcreate table b (baz)c 3 s � � dd� dV d S )Nz insert into a (bar) values (?))r_ r_ )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 r z:RegressionTests.CheckRecursiveCursorUse.<locals>.<genexpr>)r r r r rK rm 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 r zKRegressionTests.CheckConvertTimestampMicrosecondPadding.<locals>.<listcomp>i� rx � � r i@� i@� )r r r? r r rn r* r@ )r
r r �valuesr r r �'CheckConvertTimestampMicrosecondPadding7 s
�z7RegressionTests.CheckConvertTimestampMicrosecondPaddingc C s | j ttjddd� d S )Nr �{ rY )rK rV 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 rK re r r r r r r �CheckNullCharacterW s
z"RegressionTests.CheckNullCharacterc C s� t �d�}|�d� | �|jd� d}t|�d��D ]�\}}| j||d��z |�d|f� |�� |dkr|| �|d d� n2|dkr�| �|d d� n|d kr�| �|d d � |d7 }W d � q41 s�0 Y 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( r{ z insert into t2(c) values (?)r_ r2 r1 z'should have returned exactly three rows) r r Z
executescriptr* rZ � enumerater rb r3 )r
r Zcounterr( r{ 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 rl �weakref�refrj r �
gc_collect)r
r� r r r� r r r �
CheckBpo31770 s
z RegressionTests.CheckBpo31770c C s4 | � t�� | j`W d � n1 s&0 Y d S r )rK �AttributeErrorr rZ 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)rG rH rI r� r r r r �Printer� s r� r_ )�nzselect 1)r Zset_trace_callback� functools�partialZset_progress_handlerZset_authorizerr� r )r
r� �methodZprinter_instancer r r �
CheckBpo37347� s �
z RegressionTests.CheckBpo37347N)&rG rH rI r
r r r r) r, r0 �unittestZskipIfr Zsqlite_version_infor6 r: r= rA rM rU rW ri rp rr rv ry rz r| r~ 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 s~ � fdd�}t �tjd|i��L � �tj� j�� � j�d� W d � n1 sR0 Y W d � n1 sp0 Y d S )Nc s � j �� j�S r )r rj 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 �
convertersro rm rT r r �r
�convr r r �test_recursive_cursor_init� s z0RecursiveUseOfCursors.test_recursive_cursor_initc s~ � fdd�}t �tjd|i��L � �tj� j�� � j�d� W d � n1 sR0 Y W d � n1 sp0 Y 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 s� g f� fdd� }t �tjd|i��V � j�d� � �tj� j�� � j� � W d � n1 s`0 Y W d � n1 s~0 Y d S )Nc s |r� j �� S |�d �S r )r rQ 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 ro rm rT rn r� r r r �test_recursive_cursor_fetch� s
z1RecursiveUseOfCursors.test_recursive_cursor_fetchN) rG rH rI rT 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� rG r r r r �<module> s &
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists