EVOLUTION-NINJA
Edit File: hooks.cpython-38.pyc
U �i�_2) � @ 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 �1/opt/python38/lib/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 � | �� |� 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' ) Z addCleanupr r r r r] rL r) r )r rX ZqueriesZcon1Zcon2Zcurr rV r �CheckTraceCallbackContent� s � z,TraceCallbackTests.CheckTraceCallbackContentN) r! r" r# r` ra rg r? r@ r rA ri 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 rj c C s t �� } | �t� � d S r )r? ZTextTestRunner�runrj )Zrunnerr r r �test s rl �__main__) r? Zsqlite3r Z test.supportr r ZTestCaser rB rT rj rl r! r r r r �<module> s jEC