File: //usr/local/lib/python3.10/sqlite3/test/__pycache__/regression.cpython-310.opt-2.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� d S )N�create table foo(bar)�drop table foo�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� d S �N� r7 r r r r �test_empty_statements s z$RegressionTests.test_empty_statementc C sr 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 �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__rD r rC r r �X� � rH 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
rH r r rC 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) rJ 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 � d S r )rJ � 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 r9 ��isolation_levelr r r r �__del__� s
zCRegressionTests.test_set_isolation_level.<locals>.CustomStr.__del__N)rE rF rG rX r\ r r r r � CustomStr� s r] r )r: �DEFERREDZ IMMEDIATEZ EXCLUSIVE)�levelr^ � � �abczIMMEDIATE EXCLUSIVE� é) r/ r r r[ �subTest�lower�
capitalizerU �
ValueErrorrJ 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 � d S 1 sGw Y d S )Nc @ � e Zd Zdd� ZdS )zBRegressionTests.test_cursor_constructor_call_check.<locals>.Cursorc S rW r r r3 r r r �__init__� rY zKRegressionTests.test_cursor_constructor_call_check.<locals>.Cursor.__init__N�rE rF rG rm r r r r �Cursor� � ro r z
select 4+5z$^Base Cursor\.__init__ not called\.$) r ro r rJ �ProgrammingErrorr �fetchall�assertRaisesRegexr )r
ro 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� d S )Nc @ s e Zd ZdS )z0RegressionTests.test_str_subclass.<locals>.MyStrN)rE rF rG r r r r �MyStr� s ru zselect ?rb )r/ r r )r
ru r r r �test_str_subclass� s z!RegressionTests.test_str_subclassc C sV G dd� dt j�}|d�}| �t j��
|�� }W d � d S 1 s$w Y d S )Nc @ rl )zJRegressionTests.test_connection_constructor_call_check.<locals>.Connectionc S rW r r )r
�namer r r rm � rY zSRegressionTests.test_connection_constructor_call_check.<locals>.Connection.__init__Nrn r r r r �
Connection� rp rx r )r rx rJ rq r �r
rx 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 sIw Y d S )
Nc r@ )z<RegressionTests.test_cursor_registration.<locals>.Connectionc s � | �S r r r �ro r r r � s zCRegressionTests.test_cursor_registration.<locals>.Connection.cursorN)rE rF rG r r r{ r r rx � rI rx c @ rl )z8RegressionTests.test_cursor_registration.<locals>.Cursorc S s t j�| |� d S r )r ro rm r3 r r r rm � s zARegressionTests.test_cursor_registration.<locals>.Cursor.__init__Nrn r r r r ro � rp ro r zcreate table foo(x)z insert into foo(x) values (?)))� )� )r zselect x from foo)
r rx ro r r r% r& rJ ZInterfaceErrorrr ry r r{ r �test_cursor_registration� s
"�z(RegressionTests.test_cursor_registrationc C s t jdd d�}d S )Nr rZ )r r r3 r r r �test_auto_commit� s z RegressionTests.test_auto_commitc C s6 | j �� }|�d� |�d� |�d� |�� }d S )Nr5 r< zpragma page_size)r r r rP )r
r �rowr r r �test_pragma_autocommit s
z&RegressionTests.test_pragma_autocommitc C s | � t| jd� d S �Nr` )rJ rU 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 r� r )�a�br r r � collation_cb rY z4RegressionTests.test_collation.<locals>.collation_cbu �)rJ r rq 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 s8w 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 �
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>)r r r r rJ rq 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� 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 H r zNRegressionTests.test_convert_timestamp_microsecond_padding.<locals>.<listcomp>i� r} � � r i@� i@� )r r r= r r rr 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 �{ rZ )rJ rU 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 rJ rg 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|dkrL| �|d d� n |d krX| �|d d � |d7 }W d � n1 sfw Y q| �|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` � r| z'should have returned exactly three rows) r r Z
executescriptr* r[ � enumerater rd r1 )r
r Zcounterr( r� r r r �test_commit_cursor_reset^ s&
��
z(RegressionTests.test_commit_cursor_resetc C sD dd� }t �d�}t �|�}t�||�}|�|� ~~t�� d S )Nc W rW r r )�argsr r r �callback� rY z/RegressionTests.test_bpo31770.<locals>.callbackr )r r ro �weakref�refrm 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 )rJ �AttributeErrorr r[ 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 @ rl )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)rE rF rG r� r r r r �Printer� rp 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 ra )r r rP r* )r
r �valr r r � test_return_empty_bytestring� s z,RegressionTests.test_return_empty_bytestringN)#rE rF rG r
r r r r) r, r0 r4 r8 r; r? rL rT rV rk rt rv rz 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 rm 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 �
convertersrs rq rS 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 rP 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 rs rq rS rr r� r r r �test_recursive_cursor_fetch� s ��"�z1RecursiveUseOfCursors.test_recursive_cursor_fetchN) rE rF rG rS 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� rE r r r r �<module> s" $
�