File: //usr/local/lib/python3.7/sqlite3/test/__pycache__/hooks.cpython-37.pyc
B
��g2) � @ sx d dl Z d dlZd dlmZmZ G dd� de j�ZG dd� de j�ZG dd� de j�Z d d
� Z
d d � Z e d
krte � dS )� N)�TESTFN�unlinkc @ s^ e Zd Zdd� Zdd� Zdd� Zdd� Ze�e j
d k d
�d d � �Z d
d� Z dd� Z
dd� ZdS )�CollationTestsc C s4 t �d�}| �t�� |�d dd� � W d Q R X d S )Nz:memory:c S s | |k| |k S )N� )�x�yr r �./usr/local/lib/python3.7/sqlite3/test/hooks.py�<lambda>! � z>CollationTests.CheckCreateCollationNotString.<locals>.<lambda>)�sqlite�connect� assertRaises� TypeError�create_collation)�self�conr r r � CheckCreateCollationNotString s
z,CollationTests.CheckCreateCollationNotStringc C sB t �d�}| �t��}|�dd� W d Q R X | �t|j�d� d S )Nz:memory:�X�* zparameter must be callable)r r r
r r � assertEqual�str� exception)r r �cmr r r � CheckCreateCollationNotCallable# s
z.CollationTests.CheckCreateCollationNotCallablec C s6 t �d�}| �t j�� |�ddd� � W d Q R X d S )Nz:memory:u colläc S s | |k| |k S )Nr )r r r r r r , r
z=CollationTests.CheckCreateCollationNotAscii.<locals>.<lambda>)r r r
ZProgrammingErrorr )r r r r r � CheckCreateCollationNotAscii) s
z+CollationTests.CheckCreateCollationNotAsciic C sl G dd� dt �}t�d�}dd� }|�|d�|� |�d��� }| �|d d d � | �|d
d d � d S ) Nc @ s e Zd Zdd� ZdS )z@CollationTests.CheckCreateCollationBadUpper.<locals>.BadUpperStrc S s d S )Nr )r r r r �upper0 s zFCollationTests.CheckCreateCollationBadUpper.<locals>.BadUpperStr.upperN)�__name__�
__module__� __qualname__r r r r r � BadUpperStr/ s r z:memory:c S s | |k| |k S )Nr )r r r r r r 3 r
z=CollationTests.CheckCreateCollationBadUpper.<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 � CheckCreateCollationBadUpper. s
z+CollationTests.CheckCreateCollationBadUpper)� � r" z&old SQLite versions crash on this testc C s� dd� }t �d�}|�d|� d}|�|��� }| j|dddgd d
� |�dd � | �t j��}|�|��� }W d Q R X | �t|j �d � d S ) Nc S s | |k| |k S )Nr )r r r r r r B s z3CollationTests.CheckCollationIsUsed.<locals>.mycollz:memory:r z�
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# z#the expected order was not returned)�msgz"no such collation sequence: mycoll)
r r r r$ r% r r
�OperationalErrorr r )r r r �sqlr& r r r r �CheckCollationIsUsed? s
z#CollationTests.CheckCollationIsUsedc C sJ dd� }t �d�}|�d|� d}|�|��� }| j|dddgd d
� d S ) Nc S s | |k| |k d S )Nl r )r r r r r r [ s z@CollationTests.CheckCollationReturnsLargeInteger.<locals>.mycollz:memory:r z�
select x from (
select 'a' as x
union
select 'b' as x
union
select 'c' as x
) order by x collate mycoll
)r* )r! )r# z#the expected order was not returned)r+ )r r r r$ r% r )r r r r- r& r r r �!CheckCollationReturnsLargeIntegerZ s
z0CollationTests.CheckCollationReturnsLargeIntegerc 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.
z:memory:r c S s | |k| |k S )Nr )r r r r r r s r
z<CollationTests.CheckCollationRegisterTwice.<locals>.<lambda>c S s | |k| |k S )Nr )r r r r r r t r
zi
select x from (select 'a' as x union select 'b' as x) order by x collate mycoll
r r! r" r# N)r r r r$ r% r )r r r& r r r �CheckCollationRegisterTwicem s
z*CollationTests.CheckCollationRegisterTwicec C s^ t �d�}|�ddd� � |�dd� | �t j��}|�d� W dQ R X | �t|j�d� dS )zu
Register a collation, then deregister it. Make sure an error is raised if we try
to use it.
z:memory:r c S s | |k| |k S )Nr )r r r r r r � r
z9CollationTests.CheckDeregisterCollation.<locals>.<lambda>Nz?select 'a' as x union select 'b' as x order by x collate mycollz"no such collation sequence: mycoll) r r r r
r, r$ r r r )r r r r r r �CheckDeregisterCollation{ s
z'CollationTests.CheckDeregisterCollationN)r r r r r r r' �unittest�skipIfr �sqlite_version_infor. r/ r0 r1 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.
z:memory:c s � � d � dS )Nr )�appendr )�progress_callsr r �progress� s
z8ProgressTests.CheckProgressHandlerUsed.<locals>.progressr" z0
create table foo(a, b)
N)r r �set_progress_handlerr$ �
assertTrue)r r r8 r )r7 r �CheckProgressHandlerUsed� s
z&ProgressTests.CheckProgressHandlerUsedc sr t �d�}g � � fdd�}|�|d� |�� }|�d� t� �}g � |�|d� |�d� t� �}| �||� dS ) z=
Test that the opcode argument is respected.
z:memory:c s � � d � dS )Nr )r6 r )r7 r r r8 � s
z0ProgressTests.CheckOpcodeCount.<locals>.progressr" z1
create table foo (a, b)
r) z1
create table bar (a, b)
N)r r r9 �cursorr$ �lenZassertGreaterEqual)r r r8 �cursZ first_countZ second_countr )r7 r �CheckOpcodeCount� s
z ProgressTests.CheckOpcodeCountc C s< t �d�}dd� }|�|d� |�� }| �t j|jd� dS )zW
Test that returning a non-zero value stops the operation in progress.
z:memory:c S s dS )Nr" r r r r r r8 � s z4ProgressTests.CheckCancelOperation.<locals>.progressr" zcreate table bar (a, b)N)r r r9 r< r
r, r$ )r r r8 r> r r r �CheckCancelOperation� s
z"ProgressTests.CheckCancelOperationc sR t �d�}d� � fdd�}|�|d� |�dd� |�d��� | �� dd� dS ) zc
Test that setting the progress handler to None clears the previously set handler.
z:memory:r c s d� dS )Nr" r r r )�actionr r r8 � s z1ProgressTests.CheckClearHandler.<locals>.progressr" Nz&select 1 union select 2 union select 3z progress handler was not cleared)r r r9 r$ r% r )r r r8 r )rA r �CheckClearHandler� s
z ProgressTests.CheckClearHandlerN)r r r r; r? r@ rB r r r r r5 � s r5 c @ s> e Zd Zdd� Zdd� Zdd� Ze�ej dk d�d d
� �Z
d S ) �TraceCallbackTestsc sT t �d�}g � � fdd�}|�|� |�d� | �� � | �tdd� � D ��� dS )zI
Test that the trace callback is invoked once it is set.
z:memory:c s � � | � d S )N)r6 )� statement)�traced_statementsr r �trace� s z8TraceCallbackTests.CheckTraceCallbackUsed.<locals>.tracezcreate table foo(a, b)c s s | ]}d |kV qdS )zcreate table fooNr )�.0�stmtr r r � <genexpr>� s z<TraceCallbackTests.CheckTraceCallbackUsed.<locals>.<genexpr>N)r r �set_trace_callbackr$ r: �any)r r rF r )rE r �CheckTraceCallbackUsed� s
z)TraceCallbackTests.CheckTraceCallbackUsedc sH t �d�}g � � fdd�}|�|� |�d� |�d� | �� d� dS )zb
Test that setting the trace callback to None clears the previously set callback.
z:memory:c s � � | � d S )N)r6 )rD )rE r r rF � s z9TraceCallbackTests.CheckClearTraceCallback.<locals>.traceNzcreate table foo(a, b)z trace callback was not cleared)r r rJ r$ Z assertFalse)r r rF r )rE r �CheckClearTraceCallback� s
z*TraceCallbackTests.CheckClearTraceCallbackc
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 öäüÖÄÜ߀z:memory:c s � � | � d S )N)r6 )rD )rE r r rF � s z5TraceCallbackTests.CheckUnicodeContent.<locals>.tracezcreate table foo(x)z insert into foo(x) values ("%s")c 3 s | ]}� |kV qd S )Nr )rG rH )�
unicode_valuer r rI � s z9TraceCallbackTests.CheckUnicodeContent.<locals>.<genexpr>z-Unicode data %s garbled in trace callback: %sz, N)
r r rJ r$ Zcommitr: rK �ascii�join�map)r r rF r )rE rN r �CheckUnicodeContent� s
z&TraceCallbackTests.CheckUnicodeContent)r( r( � z#sqlite3_prepare_v2 is not availablec s� g � � fdd�}ddg}| � tt� tjtd d�}t�t�}|�|� |�� }|�|d � |�d� |�|d � | �� |� d S ) Nc s � � | � d S )N)r6 )rD )rE r r rF � s z;TraceCallbackTests.CheckTraceCallbackContent.<locals>.tracezcreate table foo(x)z insert into foo(x) values(1))Zisolation_levelr zcreate table bar(x)r" ) Z
addCleanupr r r r rJ r< r$ r )r rF ZqueriesZcon1Zcon2Zcurr )rE r �CheckTraceCallbackContent� s
z,TraceCallbackTests.CheckTraceCallbackContentN) r r r rL rM rR r2 r3 r r4 rT r r r r rC � s
rC c C s4 t �td�} t �td�}t �td�}t �| ||f�S )NZCheck)r2 Z makeSuiter r5 rC Z TestSuite)Zcollation_suiteZprogress_suiteZ trace_suiter r r �suite s rU c C s t �� } | �t� � d S )N)r2 ZTextTestRunner�runrU )Zrunnerr r r �test s rW �__main__)
r2 Zsqlite3r Z test.supportr r ZTestCaser r5 rC rU rW r r r r r �<module> s jEC