File: //usr/local/lib/python3.10/sqlite3/test/__pycache__/regression.cpython-310.pyc
o
�i�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 krCe� dS dS )
� N)�support)�patchc @ s e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� 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 �4/usr/local/lib/python3.10/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 �test_pragma_user_version( s
z(RegressionTests.test_pragma_user_versionc C sN t jdt jd�}z| j�� }|�d� W |�� |�� d S |�� |�� w )Nr �Z detect_typeszpragma schema_version)r r �PARSE_COLNAMESr r r r �r
r r r r r �test_pragma_schema_version- s
�
z*RegressionTests.test_pragma_schema_versionc s� t jddd�� � fdd�td�D �}|d �d� td�D ]}|d �d d
d� td�D �� q td�D ]
}|| �d | d � q3� �� d S )
Nr � )Zcached_statementsc s g | ]}� � � �qS r )r ��.0�x�r r r �
<listcomp>< � z8RegressionTests.test_statement_reset.<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 �test_statement_reset7 s z$RegressionTests.test_statement_resetc 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 � test_column_name_with_spacesF s
z,RegressionTests.test_column_name_with_spacesc 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 �'test_statement_finalization_on_close_dbN s
z7RegressionTests.test_statement_finalization_on_close_dbc C s| t �d�}|�d� |�d� z|�d� W n
t jy Y nw |�d� z|�� W d S t jy= | �d� Y d S w )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 �test_on_conflict_rollback\ s
�
�z)RegressionTests.test_on_conflict_rollbackc 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 �2test_workaround_for_buggy_sqlite_transfer_bindingsj s zBRegressionTests.test_workaround_for_buggy_sqlite_transfer_bindingsc C s | j �d� dS )z
pysqlite used to segfault with SQLite versions 3.5.x. These return NULL
for "no-operation" statements
� Nr7 r r r r �test_empty_statements s z$RegressionTests.test_empty_statementc 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 (?)r6 z create table foo(bar integer)� insert into foo(bar) values (5)N)r r �PARSE_DECLTYPESr �datetimeZnowr )r
ZSELECTr r r r �test_type_map_usagez s
z#RegressionTests.test_type_map_usagec sr G � fdd�d�}|� dg� t jdt jd�}|�d� | �t�� |�d� � W d � d S 1 s2w Y d S )Nc � e Zd Z� fdd�ZdS )z2RegressionTests.test_bind_mutating_list.<locals>.Xc s � � � dS )Nz...)�clear)r
�protocol�Z
parametersr r � __conform__� s z>RegressionTests.test_bind_mutating_list.<locals>.X.__conform__N)�__name__�
__module__� __qualname__rC r rB r r �X� � rG 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
rG r r rB r �test_bind_mutating_list� s
"�z'RegressionTests.test_bind_mutating_listc C sd | � tj��}| j�dttdg��f��� W d � n1 s w Y d}| �|t |j
�� d S )Nz"select 'xxx' || ? || 'yyy' colname� z9Could not decode to UTF-8 column 'colname' with text 'xxx) rI r r2 r r �bytes� bytearray�fetchoneZassertInr/ � exception)r
�cm�msgr r r �test_error_msg_decode_error� s � �z+RegressionTests.test_error_msg_decode_errorc C s | � ttji d� dS )z!
See issue 3312.
N)rI � TypeErrorr Zregister_adapterr r r r �test_register_adapter� s z%RegressionTests.test_register_adapterc
s( G � fdd�dt �}t�d�� d � _dD ]*}| j|d�� |� _|�� � _|�� � _||�� _W d � n1 s9w Y qd � _d� _dtfdtfd tfd
tfd tfg}|D ]9\}}| j|d��' | � |�� |� _W d � n1 svw Y | �
� jd� W d � n1 s�w Y qXd S ) Nc s e Zd Zdd� Z� fdd�ZdS )z;RegressionTests.test_set_isolation_level.<locals>.CustomStrc S � d S r r r r r r �upper� � zARegressionTests.test_set_isolation_level.<locals>.CustomStr.upperc s
d� _ d S )Nr9 ��isolation_levelr r r r �__del__� s
zCRegressionTests.test_set_isolation_level.<locals>.CustomStr.__del__N)rD rE rF rW r[ r r r r � CustomStr� s r\ r )r9 �DEFERREDZ IMMEDIATEZ EXCLUSIVE)�levelr] � � �abczIMMEDIATE EXCLUSIVE� é) r/ r r rZ �subTest�lower�
capitalizerT �
ValueErrorrI r* )r
r\ r^ �pairs�value�excr r r �test_set_isolation_level� s2
�� � � ���z(RegressionTests.test_set_isolation_levelc C s� G dd� dt j�}t �d�}||�}| �t j�� |�d��� W d � n1 s*w Y | �t jd��
|�� W d � dS 1 sFw Y dS )zd
Verifies that cursor methods check whether base class __init__ was
called.
c @ � e Zd Zdd� ZdS )zBRegressionTests.test_cursor_constructor_call_check.<locals>.Cursorc S rV r r r3 r r r �__init__� rX zKRegressionTests.test_cursor_constructor_call_check.<locals>.Cursor.__init__N�rD rE rF rl r r r r �Cursor� � rn r z
select 4+5Nz$^Base Cursor\.__init__ not called\.$) r rn r rI �ProgrammingErrorr �fetchall�assertRaisesRegexr )r
rn r r r r r �"test_cursor_constructor_call_check� s
��
"�z2RegressionTests.test_cursor_constructor_call_checkc 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 )z0RegressionTests.test_str_subclass.<locals>.MyStrN)rD rE rF r r r r �MyStr� s rt zselect ?ra N)r/ r r )r
rt r r r �test_str_subclass� s z!RegressionTests.test_str_subclassc C sT G dd� dt j�}|d�}| �t j��
|�� }W d � dS 1 s#w Y dS )zh
Verifies that connection methods check whether base class __init__ was
called.
c @ rk )zJRegressionTests.test_connection_constructor_call_check.<locals>.Connectionc S rV r r )r
�namer r r rl � rX zSRegressionTests.test_connection_constructor_call_check.<locals>.Connection.__init__Nrm r r r r �
Connection� ro rw r N)r rw rI rp r �r
rw r r r r r �&test_connection_constructor_call_check� s
"�z6RegressionTests.test_connection_constructor_call_checkc s� G � fdd�dt j�}G dd� dt j�� |d�}|�� }|�d� |�dg d�� |�d � |�� | �t j��
|� � W d
� d
S 1 sHw Y d
S ) z�
Verifies that subclassed cursor classes are correctly registered with
the connection object, too. (fetch-across-rollback problem)
c r? )z<RegressionTests.test_cursor_registration.<locals>.Connectionc s � | �S r r r �rn r r r � s zCRegressionTests.test_cursor_registration.<locals>.Connection.cursorN)rD rE rF r r rz r r rw � rH rw c @ rk )z8RegressionTests.test_cursor_registration.<locals>.Cursorc S s t j�| |� d S r )r rn rl r3 r r r rl � s zARegressionTests.test_cursor_registration.<locals>.Cursor.__init__Nrm r r r r rn � ro rn r zcreate table foo(x)z insert into foo(x) values (?)))� )� )r zselect x from fooN)
r rw rn r r r% r& rI ZInterfaceErrorrq rx r rz r �test_cursor_registration� s
"�z(RegressionTests.test_cursor_registrationc 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 NrY )r r r3 r r r �test_auto_commit� s z RegressionTests.test_auto_commitc 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.
r5 r; zpragma page_sizeN)r r r rO )r
r �rowr r r �test_pragma_autocommit s
z&RegressionTests.test_pragma_autocommitc C s | � t| jd� dS )z}
Call a connection with a non-string SQL request: check error handling
of the statement constructor.
r_ N)rI rT r r r r r �test_connection_call s z$RegressionTests.test_connection_callc C s" dd� }| � tj| jjd|� d S )Nc S s dS )Nr_ r )�a�br r r � collation_cb rX z4RegressionTests.test_collation.<locals>.collation_cbu �)rI r rp r Zcreate_collation)r
r� r r r �test_collation s �z RegressionTests.test_collationc s| t �d�}|�� � � �d� � �d� � fdd�}| �t j�� � �ddd� |� D �� W d � d S 1 s7w 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 �
z6RegressionTests.test_recursive_cursor_use.<locals>.fooz insert into b (baz) values (?)c s s � | ]}|fV qd S r r )r r( r r r � <genexpr>3 s � z<RegressionTests.test_recursive_cursor_use.<locals>.<genexpr>N)r r r r rI rp r% )r
r r� r r� r �test_recursive_cursor_use s
�"�z)RegressionTests.test_recursive_cursor_usec
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 H r zNRegressionTests.test_convert_timestamp_microsecond_padding.<locals>.<listcomp>i� r| � � r i@� i@� N)r r r< r r rq r* r= )r
r r �valuesr r r �*test_convert_timestamp_microsecond_padding5 s
�z:RegressionTests.test_convert_timestamp_microsecond_paddingc C s | j ttjddd� d S )Nr �{ rY )rI rT r r r r r r �!test_invalid_isolation_level_typeO s
�z1RegressionTests.test_invalid_isolation_level_typec 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 rI rf r r r r r r �test_null_characterU s
z#RegressionTests.test_null_characterc C s� t �d�}|�d� | �|jd� d}t|�d��D ]P\}}| j||d��= |�d|f� |�� |dkr>| �|d d� n|dkrK| �|d d� n |d krW| �|d d � |d7 }W d
� n1 sew Y q| �|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);
r9 r zselect c from t)r( r z insert into t2(c) values (?)r_ � Nr{ z'should have returned exactly three rows) r r Z
executescriptr* rZ � enumerater rc r1 )r
r Zcounterr( r r r r �test_commit_cursor_reset^ s$
��
z(RegressionTests.test_commit_cursor_resetc 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 rV r r )�argsr r r �callback� rX z/RegressionTests.test_bpo31770.<locals>.callbackr N)r r rn �weakref�refrl r �
gc_collect)r
r� r r r� r r r �
test_bpo31770} s
z RegressionTests.test_bpo31770c C s6 | � t�� | j`W d � d S 1 sw Y d S r )rI �AttributeErrorr rZ r r r r �!test_del_isolation_level_segfault� s "�z1RegressionTests.test_del_isolation_level_segfaultc C sf G dd� d�}| j jtj| j jdd�| j jfD ]}|� }||j� ||j� | j �d� |d � qd S )Nc @ rk )z.RegressionTests.test_bpo37347.<locals>.Printerc W s t jS r )r Z SQLITE_OK)r
r� r r r �log� s z2RegressionTests.test_bpo37347.<locals>.Printer.logN)rD rE rF r� r r r r �Printer� ro r� r_ )�nzselect 1)r Zset_trace_callback� functools�partialZset_progress_handlerZset_authorizerr� r )r
r� �methodZprinter_instancer r r �
test_bpo37347� s �
�z RegressionTests.test_bpo37347c C s( | j �d�}|�� d }| �|d� d S )Nz
select X''r r` )r r rO r* )r
r �valr r r � test_return_empty_bytestring� s z,RegressionTests.test_return_empty_bytestringN)#rD rE rF r
r r r r) r, r0 r4 r8 r: r> rK rS rU rj rs ru ry r} r~ r� r� r� r� r� r� r� r� r� r� r� r� r r r r r ! sB
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 � � fdd�}t �tjd|i��/ � �tj� j�� � j�d� W d � n1 s)w Y W d � d S W d � d S 1 sAw Y d S )Nc s � j �� j�S r )r rl r �r r r r �<lambda>� s zBRecursiveUseOfCursors.test_recursive_cursor_init.<locals>.<lambda>ZINITz#select x as "x [INIT]", x from test� r �dictr �
convertersrr rp rR r r �r
�convr r r �test_recursive_cursor_init� � ��"�z0RecursiveUseOfCursors.test_recursive_cursor_initc r� )Nc s
� j �� S r )r r r� r r r r� � s
zCRecursiveUseOfCursors.test_recursive_cursor_close.<locals>.<lambda>ZCLOSEz$select x as "x [CLOSE]", x from testr� r� r r r �test_recursive_cursor_close� r� z1RecursiveUseOfCursors.test_recursive_cursor_closec s� g f� fdd� }t �tjd|i��4 � j�d� � �tj� j��
� j� � W d � n1 s0w Y W d � d S W d � d S 1 sHw Y d S )Nc s |r� j �� S |�d �S r )r rO r. )r �lr r r r� � s zCRecursiveUseOfCursors.test_recursive_cursor_fetch.<locals>.<lambda>ZITERz#select x as "x [ITER]", x from test)
r r� r r� r r rr rp rR rq r� r r r �test_recursive_cursor_fetch� s ��"�z1RecursiveUseOfCursors.test_recursive_cursor_fetchN) rD rE rF rR r
r r� r� r� r r r r r� � s r� c C s t tg} t�dd� | D ��S )Nc S s g | ] }t �� �|��qS r )�unittestZ
TestLoaderZloadTestsFromTestCase)r �tr r r r � s zsuite.<locals>.<listcomp>)r r� r� Z TestSuite)Ztestsr 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� rD r r r r �<module> s" $
�