Your IP : 216.73.216.85
o
0�i�* � @ s� d dl Z d dlZd dlZd dlmZmZ G dd� dej�ZG dd� dej�Z G dd� dej�Z
d d
� Z d d � Z e
d
kr@e � dS dS )� N)�TESTFN�unlinkc @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )�CollationTestsc C sJ t �d�}| �t�� |�d dd� � W d � d S 1 s w Y d S )N�:memory:c S � | |k| |k S �N� ��x�yr r �9/opt/alt/python310/lib64/python3.10/sqlite3/test/hooks.py�<lambda>" � zACollationTests.test_create_collation_not_string.<locals>.<lambda>)�sqlite�connect� assertRaises� TypeError�create_collation��self�conr r r � test_create_collation_not_string s
"�z/CollationTests.test_create_collation_not_stringc C sV t �d�}| �t��}|�dd� W d � n1 sw Y | �t|j�d� d S )Nr �X�* zparameter must be callable)r r r r r � assertEqual�str� exception�r r �cmr r r �"test_create_collation_not_callable$ s
�z1CollationTests.test_create_collation_not_callablec C sL t �d�}| �t j�� |�ddd� � W d � d S 1 s w Y d S )Nr u colläc S r r r r r r r r
- r z@CollationTests.test_create_collation_not_ascii.<locals>.<lambda>)r r r ZProgrammingErrorr r r r r � test_create_collation_not_ascii* s
"�z.CollationTests.test_create_collation_not_asciic C sl G dd� dt �}t�d�}dd� }|�|d�|� |�d��� }| �|d d d � | �|d
d d � d S ) Nc @ s e Zd Zdd� ZdS )zCCollationTests.test_create_collation_bad_upper.<locals>.BadUpperStrc S s d S r r )r r r r �upper1 � zICollationTests.test_create_collation_bad_upper.<locals>.BadUpperStr.upperN)�__name__�
__module__� __qualname__r! r r r r � BadUpperStr0 s r& r c S � | |k| |k S r r r r r r r
4 � z@CollationTests.test_create_collation_bad_upper.<locals>.<lambda>�mycollz�
select x from (
select 'a' as x
union
select 'b' as x
) order by x collate mycoll
r �b� �a)r r r r �execute�fetchallr )r r& r r) �resultr r r � test_create_collation_bad_upper/ s
�z.CollationTests.test_create_collation_bad_upperc C s� dd� }t �d�}|�d|� d}|�|��� }| j|g d�dd� |�dd � | �t j��}|�|��� }W d � n1 s@w Y | �t|j �d � d S )
Nc S r' r r r r r r r) A s z5CollationTests.test_collation_is_used.<locals>.mycollr r) ��
select x from (
select 'a' as x
union
select 'b' as x
union
select 'c' as x
) order by x collate mycoll
�)�c)r* )r, �#the expected order was not returned��msg�"no such collation sequence: mycoll)
r r r r- r. r r �OperationalErrorr r )r r) r �sqlr/ r r r r �test_collation_is_used@ s
� �z%CollationTests.test_collation_is_usedc C sH dd� }t �d�}|�d|� d}|�|��� }| j|g d�dd� d S ) Nc S s | |k| |k d S )Nl r r r r r r) Z s zCCollationTests.test_collation_returns_large_integer.<locals>.mycollr r) r1 r2 r4 r5 �r r r r- r. r )r r) r r9 r/ r r r �$test_collation_returns_large_integerY s
�z3CollationTests.test_collation_returns_large_integerc C sd t �d�}|�ddd� � |�ddd� � |�d��� }| �|d d d� | �|d d d
� d S ) z�
Register two different collation functions under the same name.
Verify that the last one is actually used.
r r) c S r r r r r r r r
r r z>CollationTests.test_collation_register_twice.<locals>.<lambda>c S r' r r r r r r r
s r( zi
select x from (select 'a' as x union select 'b' as x) order by x collate mycoll
r r* r+ r, Nr; )r r r/ r r r � test_collation_register_twicel s
�z,CollationTests.test_collation_register_twicec C sr t �d�}|�ddd� � |�dd� | �t j��
}|�d� W d � n1 s)w Y | �t|j�d� dS )zu
Register a collation, then deregister it. Make sure an error is raised if we try
to use it.
r r) c S r r r r r r r r
� r z:CollationTests.test_deregister_collation.<locals>.<lambda>Nz?select 'a' as x union select 'b' as x order by x collate mycollr7 ) r r r r r8 r- r r r r r r r �test_deregister_collationz s
�z(CollationTests.test_deregister_collationN) r# r$ r% r r r r0 r: r< r= r> r r r r r s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�
ProgressTestsc s> t �d�}g � � fdd�}|�|d� |�d� | �� � dS )zK
Test that the progress handler is invoked once it is set.
r c � � � d � dS �Nr ��appendr �Zprogress_callsr r �progress� �
z:ProgressTests.test_progress_handler_used.<locals>.progressr+ z0
create table foo(a, b)
N)r r �set_progress_handlerr- �
assertTrue�r r rE r rD r �test_progress_handler_used� s
z(ProgressTests.test_progress_handler_usedc sr t �d�}g � � fdd�}|�|d� |�� }|�d� t� �}g � |�|d� |�d� t� �}| �||� dS ) z=
Test that the opcode argument is respected.
r c r@ rA rB r rD r r rE � rF z1ProgressTests.test_opcode_count.<locals>.progressr+ z1
create table foo (a, b)
� z1
create table bar (a, b)
N)r r rG �cursorr- �lenZassertGreaterEqual)r r rE �cursZ first_countZ second_countr rD r �test_opcode_count� s
z ProgressTests.test_opcode_countc C s< t �d�}dd� }|�|d� |�� }| �t j|jd� dS )zW
Test that returning a non-zero value stops the operation in progress.
r c S s dS )Nr+ r r r r r rE � r" z5ProgressTests.test_cancel_operation.<locals>.progressr+ zcreate table bar (a, b)N)r r rG rL r r8 r- )r r rE rN r r r �test_cancel_operation� s
�z#ProgressTests.test_cancel_operationc sR t �d�}d� � fdd�}|�|d� |�dd� |�d��� | �� dd� dS ) zc
Test that setting the progress handler to None clears the previously set handler.
r r c s d� dS )Nr+ r r r ��actionr r rE � s z2ProgressTests.test_clear_handler.<locals>.progressr+ Nz&select 1 union select 2 union select 3z progress handler was not cleared)r r rG r- r. r rI r rQ r �test_clear_handler� s
z ProgressTests.test_clear_handlerN)r# r$ r% rJ rO rP rS r r r r r? � s
r? c @ sB e Zd Zejdd� �Zdd� Zdd� Zdd� Zd d
� Z d d � Z
d
S )�TraceCallbackTestsc
# sT � z g � |� � fdd�� d V W | �� |� |� d � d S | �� |� |� d � w )Nc s
� � | �S r rB )�stmt�Ztracedr r r
� s
z5TraceCallbackTests.check_stmt_trace.<locals>.<lambda>)�set_trace_callbackr )r �cx�expectedr rV r �check_stmt_trace� s � � z#TraceCallbackTests.check_stmt_tracec sT t �d�}g � � fdd�}|�|� |�d� | �� � | �tdd� � D ��� dS )zI
Test that the trace callback is invoked once it is set.
r c � � � | � d S r rB �Z statement��traced_statementsr r �trace� � z:TraceCallbackTests.test_trace_callback_used.<locals>.trace�create table foo(a, b)c s s � | ]}d |v V qdS )zcreate table fooNr ��.0rU r r r � <genexpr>� � � z>TraceCallbackTests.test_trace_callback_used.<locals>.<genexpr>N)r r rW r- rH �any�r r r_ r r] r �test_trace_callback_used� s
z+TraceCallbackTests.test_trace_callback_usedc sH t �d�}g � � fdd�}|�|� |�d� |�d� | �� d� dS )zb
Test that setting the trace callback to None clears the previously set callback.
r c r[ r rB r\ r] r r r_ � r` z;TraceCallbackTests.test_clear_trace_callback.<locals>.traceNra z trace callback was not cleared)r r rW r- Z assertFalserg r r] r �test_clear_trace_callback� s
z,TraceCallbackTests.test_clear_trace_callbackc
s� d�t �d�}g � � fdd�}|�|� |�d� |�d� � |�� | �t�fdd�� D ��d t��d
�t t� ��f � d S ) zG
Test that the statement can contain unicode literals.
u öäüÖÄÜ߀r c r[ r rB r\ r] r r r_ � r` z6TraceCallbackTests.test_unicode_content.<locals>.trace�create table foo(x)z insert into foo(x) values ('%s')c 3 s � | ]}� |v V qd S r r rb )�
unicode_valuer r rd � re z:TraceCallbackTests.test_unicode_content.<locals>.<genexpr>z-Unicode data %s garbled in trace callback: %sz, N)
r r rW r- ZcommitrH rf �ascii�join�maprg r )r^ rk r �test_unicode_content� s
��z'TraceCallbackTests.test_unicode_contentc s� g � � fdd�}ddg}| � tt� tjtd d�}t�t�}|�|� |�� }|�|d � |�d� |�|d � | �� |� d S ) Nc r[ r rB r\ r] r r r_ r` z=TraceCallbackTests.test_trace_callback_content.<locals>.tracerj z insert into foo(x) values(1))Zisolation_levelr zcreate table bar(x)r+ ) Z
addCleanupr r r r rW rL r- r )r r_ ZqueriesZcon1Zcon2Zcurr r] r �test_trace_callback_content s �
z.TraceCallbackTests.test_trace_callback_contentc C s� g d�}t �d�}| �||��5 |� |�d� |�ddd� td�D �� W d � n1 s/w Y W d � d S W d � d S 1 sGw Y d S )N)�create table t(t)zBEGIN zinsert into t values(0)zinsert into t values(1)zinsert into t values(2)ZCOMMITr rq zinsert into t values(?)c s s � | ]}|fV qd S r r )rc �vr r r rd s � z=TraceCallbackTests.test_trace_expanded_sql.<locals>.<genexpr>� )r r rZ r- Z executemany�range)r rY rX r r r �test_trace_expanded_sql s
��"�z*TraceCallbackTests.test_trace_expanded_sqlN) r# r$ r% �
contextlib�contextmanagerrZ rh ri ro rp ru r r r r rT � s
rT c C s t ttg} t�dd� | D ��S )Nc S s g | ] }t �� �|��qS r )�unittestZ
TestLoaderZloadTestsFromTestCase)rc �tr r r �
<listcomp>* s zsuite.<locals>.<listcomp>)r r? rT rx Z TestSuite)Ztestsr r r �suite# s � �r{ c C s t �� } | �t� � d S r )rx ZTextTestRunner�runr{ )Zrunnerr r r �test- s r} �__main__)rv rx Zsqlite3r Ztest.support.os_helperr r ZTestCaser r? rT r{ r} r# r r r r �<module> s hEX
�