𝗖𝗨𝗥𝗥𝗘𝗡𝗧 𝗙𝗜𝗟𝗘 : /opt/alt/python38/lib64/python3.8/sqlite3/test/__pycache__//hooks.cpython-38.pyc
U
i�fz* � @ 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 5 Q R X d S )N�:memory:c S s | |k| |k S �N� ��x�yr r �7/opt/alt/python38/lib64/python3.8/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 5 Q R X | �t|j�d� d S )Nr �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 5 Q R X d S )Nr u colläc S s | |k| |k S r r r r r r r , r
z=CollationTests.CheckCreateCollationNotAscii.<locals>.<lambda>)r r r ZProgrammingErrorr 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 r r )r r r r �upper0 s zFCollationTests.CheckCreateCollationBadUpper.<locals>.BadUpperStr.upperN)�__name__�
__module__� __qualname__r r r r r � BadUpperStr/ s r$ r c S s | |k| |k S r 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 5 Q R X | �t|j �d � d S ) Nc S s | |k| |k S r r r r r r r% B s z3CollationTests.CheckCollationIsUsed.<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 �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% [ s z@CollationTests.CheckCollationReturnsLargeInteger.<locals>.mycollr r% r/ r0 r2 r3 r4 r5 �r r r r) r* r )r r% r r9 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.
r r% c S s | |k| |k S r r r r r r r s r
z<CollationTests.CheckCollationRegisterTwice.<locals>.<lambda>c S s | |k| |k S r 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( Nr; )r r r+ r r r �CheckCollationRegisterTwicem s
z*CollationTests.CheckCollationRegisterTwicec C s^ t �d�}|�ddd� � |�dd� | �t j��}|�d� W 5 Q 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.
r r% c S s | |k| |k S r 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 mycollr7 ) r r r r r8 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< 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 s � � d � dS �Nr ��appendr �Zprogress_callsr r �progress� s
z8ProgressTests.CheckProgressHandlerUsed.<locals>.progressr' z0
create table foo(a, b)
N)r r �set_progress_handlerr) �
assertTrue�r r rG r rF 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.
r c s � � d � dS rC rD r rF r r rG � s
z0ProgressTests.CheckOpcodeCount.<locals>.progressr' z1
create table foo (a, b)
r. z1
create table bar (a, b)
N)r r rH �cursorr) �lenZassertGreaterEqual)r r rG �cursZ first_countZ second_countr rF 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.
r c S s dS )Nr' r r r r r rG � s z4ProgressTests.CheckCancelOperation.<locals>.progressr' zcreate table bar (a, b)N)r r rH rL r r8 r) )r r rG rN 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.
r r c s d� dS )Nr' r r r ��actionr r rG � s z1ProgressTests.CheckClearHandler.<locals>.progressr' Nz&select 1 union select 2 union select 3z progress handler was not cleared)r r rH r) r* r rJ r rQ r �CheckClearHandler� s
z ProgressTests.CheckClearHandlerN)r! r" r# rK rO rP rS r r r r rB � s rB 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.
r c s � � | � d S r rD �Z statement��traced_statementsr r �trace� s z8TraceCallbackTests.CheckTraceCallbackUsed.<locals>.trace�create table foo(a, b)c s s | ]}d |kV qdS )zcreate table fooNr ��.0Zstmtr r r � <genexpr>� s z<TraceCallbackTests.CheckTraceCallbackUsed.<locals>.<genexpr>N)r r �set_trace_callbackr) rI �any�r r rX r rV 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.
r c s � � | � d S r rD rU rV r r rX � s z9TraceCallbackTests.CheckClearTraceCallback.<locals>.traceNrY z trace callback was not cleared)r r r] r) Z assertFalser_ r rV 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 öäüÖÄÜ߀r c s � � | � d S r rD rU rV r r rX � s z5TraceCallbackTests.CheckUnicodeContent.<locals>.trace�create table foo(x)z insert into foo(x) values ("%s")c 3 s | ]}� |kV qd S r r rZ )�
unicode_valuer r r\ � s z9TraceCallbackTests.CheckUnicodeContent.<locals>.<genexpr>z-Unicode data %s garbled in trace callback: %sz, N)
r r r] r) ZcommitrI r^ �ascii�join�mapr_ r )rW rc 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 � tjd k r�|� |d
� | �
� |� d S ) Nc s � � | � d S r rD rU rV r r rX � s z;TraceCallbackTests.CheckTraceCallbackContent.<locals>.tracerb z insert into foo(x) values(1))Zisolation_levelr zcreate table bar(x)r' )r- � � ���) Z
addCleanupr r r r r] rL r) rA rE r )r rX ZqueriesZcon1Zcon2Zcurr rV r �CheckTraceCallbackContent� s �
z,TraceCallbackTests.CheckTraceCallbackContentN) r! r" r# r` ra rg r? r@ r rA rl r r r r rT � s
rT c C s4 t �td�} t �td�}t �td�}t �| ||f�S )NZCheck)r? Z makeSuiter rB rT Z TestSuite)Zcollation_suiteZprogress_suiteZ trace_suiter r r �suite s rm c C s t �� } | �t� � d S r )r? ZTextTestRunner�runrm )Zrunnerr r r �test s ro �__main__)
r? Zsqlite3r Z test.supportr r ZTestCaser rB rT rm ro r! r r r r �<module> s jEK