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� dS )zm
pysqlite would crash with older SQLite versions unless
a workaround is implemented.
�create table foo(bar)�drop table fooN�r r r r r r �-CheckWorkaroundForBuggySqliteTransferBindingsl 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
� Nr9 r r r r �CheckEmptyStatementu s z#RegressionTests.CheckEmptyStatementc C sp 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 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)N)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__rE r rC r r �X� s rI 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
rI r r rC 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) rJ r r4 r r �bytes� bytearray�fetchoneZassertInr/ � exception)r
�cm�msgr r r �CheckErrorMsgDecodeError� s �&z(RegressionTests.CheckErrorMsgDecodeErrorc C s | � ttji d� dS )z!
See issue 3312.
N)rJ � 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 )Nr; ��isolation_levelr r r r �__del__� s zARegressionTests.CheckSetIsolationLevel.<locals>.CustomStr.__del__N)rF rG rH rW rZ r r r r � CustomStr� s r[ r )r; �DEFERREDZ IMMEDIATEZ EXCLUSIVE)�levelr\ � r �abczIMMEDIATE EXCLUSIVE� é) r/ r r rY �subTest�lower�
capitalizerU �
ValueErrorrJ 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 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 r r r5 r r r �__init__� s zHRegressionTests.CheckCursorConstructorCallCheck.<locals>.Cursor.__init__N�rF rG rH ri r r r r �Cursor� s rk r z
select 4+5Nz$^Base Cursor\.__init__ not called\.$) r rk r rJ �ProgrammingErrorr �fetchall�assertRaisesRegexr )r
rk 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)rF rG rH r r r r �MyStr� r rp zselect ?r_ N)r/ r r )r
rp r r r �CheckStrSubclass� s z RegressionTests.CheckStrSubclassc C sR G dd� dt j�}|d�}| �t j�� |�� }W d � n1 sD0 Y 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 r r )r
�namer r r ri � s zPRegressionTests.CheckConnectionConstructorCallCheck.<locals>.Connection.__init__Nrj r r r r �
Connection� s rs r N)r rs rJ rl r �r
rs 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 ) 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 r r r �rk r r r � s zBRegressionTests.CheckCursorRegistration.<locals>.Connection.cursorN)rF rG rH r r rv r r rs � s rs c @ s e Zd Zdd� ZdS )z7RegressionTests.CheckCursorRegistration.<locals>.Cursorc S s t j�| |� d S r )r rk ri r5 r r r ri � s z@RegressionTests.CheckCursorRegistration.<locals>.Cursor.__init__Nrj r r r r rk � s rk r zcreate table foo(x)z insert into foo(x) values (?)))r1 )� )r zselect x from fooN)
r rs rk r r r% r& rJ ZInterfaceErrorrm rt r rv 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.
r NrX )r r r5 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.
r7 r= zpragma page_sizeN)r r r rP )r
r �rowr r r �CheckPragmaAutocommit s
z%RegressionTests.CheckPragmaAutocommitc C s | � t| jd� dS )z}
Call a connection with a non-string SQL request: check error handling
of the statement constructor.
r^ N)rJ rU r r r r r �CheckConnectionCall s z#RegressionTests.CheckConnectionCallc C s" dd� }| � tj| jjd|� d S )Nc S s dS )Nr^ r )�a�br r r � collation_cb s z4RegressionTests.CheckCollation.<locals>.collation_cbu �)rJ r rl 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 )
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.
r 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>N)r r r r rJ rl 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� 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".
r 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� rw � � r i@� i@� N)r r r> r r rm r* r? )r
r r �valuesr r r �'CheckConvertTimestampMicrosecondPadding7 s
�z7RegressionTests.CheckConvertTimestampMicrosecondPaddingc C s | j ttjddd� d S )Nr �{ rX )rJ rU 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 rJ rd 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 )
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.
r 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( rz z insert into t2(c) values (?)r^ r2 Nr1 z'should have returned exactly three rows) r r Z
executescriptr* rY � enumerater ra r3 )r
r Zcounterr( rz r r r �CheckCommitCursorReset` 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 r r )�argsr r r �callback� s z/RegressionTests.CheckBpo31770.<locals>.callbackr N)r r rk �weakref�refri 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 )rJ �AttributeErrorr rY 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)rF rG rH 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)&rF rG rH r
r r r r) r, r0 �unittestZskipIfr Zsqlite_version_infor6 r: r< r@ rL rT rV rh ro rq ru rx ry 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 ri 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 �
convertersrn rl rS 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 rP 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 rn rl rS rm r� r r r �test_recursive_cursor_fetch� s
z1RecursiveUseOfCursors.test_recursive_cursor_fetchN) rF rG rH rS 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� rF r r r r �<module> s &
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists