File: //usr/local/lib/python3.10/sqlite3/test/__pycache__/userfunctions.cpython-310.opt-2.pyc
o
�ixL � @ sv d dl Z d dlZ d dlZd dlmZ dd� Zdd� Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
dd� ZG dd� d�ZG dd� d�ZG dd� d�ZG dd � d �ZG d d � d �ZG d d � d �ZG d!d"� d"�ZG d#d$� d$�ZG d%d&� d&�ZG d'd(� d(e j�ZG d)d*� d*e j�ZG d+d,� d,e j�ZG d-d.� d.e�Z G d/d0� d0e�Z G d1d2� d2e�Z d3d4� Z d5d6� Z e!d7kr�e � dS dS )8� N)�
gc_collectc C � dS )N�foo� r r r �7/usr/local/lib/python3.10/sqlite3/test/userfunctions.py�func_returntext � r c C r )N�1 2r r r r r �func_returntextwithnull! r r
c C r )N�barr r r r r �func_returnunicode# r r c C r �N�* r r r r r �func_returnint% r r c C r )N� ��Q� @r r r r r �func_returnfloat' r r c C � d S �Nr r r r r �func_returnnull) r r c C r )N� blobr r r r r �func_returnblob+ r r c C r )N� r r r r r �func_returnlonglong- r r c C � dd d S �N� r r r r r r �func_raiseexception/ � r c @ � e Zd Zdd� Zdd� ZdS )�
AggrNoStepc C r r r ��selfr r r �__init__3 r zAggrNoStep.__init__c C r �N� r r r r r �finalize6 r zAggrNoStep.finalizeN)�__name__�
__module__� __qualname__r" r% r r r r r 2 � r c @ r )�AggrNoFinalizec C r r r r r r r r" : r zAggrNoFinalize.__init__c C r r r �r! �xr r r �step= r zAggrNoFinalize.stepN)r& r' r( r" r- r r r r r* 9 r) r* c @ �$ e Zd Zdd� Zdd� Zdd� ZdS )�AggrExceptionInInitc C r r r r r r r r" A r z AggrExceptionInInit.__init__c C r r r r+ r r r r- D r zAggrExceptionInInit.stepc C r r r r r r r r% G r z AggrExceptionInInit.finalizeN�r& r' r( r" r- r% r r r r r/ @ � r/ c @ r. )�AggrExceptionInStepc C r r r r r r r r" K r z AggrExceptionInStep.__init__c C r r r r+ r r r r- N r zAggrExceptionInStep.stepc C r r
r r r r r r% Q r z AggrExceptionInStep.finalizeNr0 r r r r r2 J r1 r2 c @ r. )�AggrExceptionInFinalizec C r r r r r r r r" U r z AggrExceptionInFinalize.__init__c C r r r r+ r r r r- X r z AggrExceptionInFinalize.stepc C r r r r r r r r% [ r z AggrExceptionInFinalize.finalizeNr0 r r r r r3 T r1 r3 c @ r. )�
AggrCheckTypec C s
d | _ d S r ��valr r r r r" _ �
zAggrCheckType.__init__c C s. t tttd �td�}t|| t|�u �| _d S �N)�str�int�float�NoneZblob�r9 r: r; �type�bytesr6 )r! � whichTyper6 �theTyper r r r- b s �zAggrCheckType.stepc C � | j S r r5 r r r r r% g � zAggrCheckType.finalizeNr0 r r r r r4 ^ s r4 c @ r. )�AggrCheckTypesc C �
d| _ d S )Nr r5 r r r r r" k r7 zAggrCheckTypes.__init__c G s@ t tttd �td�}|D ]}| jt|| t|�u �7 _q d S r8 r= )r! r@ ZvalsrA r6 r r r r- n s � �zAggrCheckTypes.stepc C rB r r5 r r r r r% t rC zAggrCheckTypes.finalizeNr0 r r r r rD j s rD c @ r. )�AggrSumc C rE )N� r5 r r r r r" x r7 zAggrSum.__init__c C s | j |7 _ d S r r5 )r! r6 r r r r- { s z AggrSum.stepc C rB r r5 r r r r r% ~ rC zAggrSum.finalizeNr0 r r r r rF w r1 rF c @ r. )�AggrTextc C rE )N� ��txtr r r r r" � r7 zAggrText.__init__c C s | j | | _ d S r rJ )r! rK r r r r- � s z
AggrText.stepc C rB r rJ r r r r r% � rC zAggrText.finalizeNr0 r r r r rH � s rH c @ 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,� Ze�ej d-k d.�d/d0� �Z e�ej d-k d.�d1d2� �Z e�ej d-kd3�d4d5� �Z d6d7� Z d8d9� Z!d:S );�
FunctionTestsc C s t �d�| _| j�ddt� | j�ddt� | j�ddt� | j�ddt� | j�ddt� | j�ddt � | j�d dt
� | j�d
dt � | j�d dd d
� � | j�dddd
� � | j�ddt � | j�dddd
� � | j�dddd
� � | j�dddd
� � | j�
d� d S )N�:memory:Z
returntextr ZreturntextwithnullZ
returnunicodeZ returnintZ returnfloatZ
returnnullZ
returnblobZreturnlonglongZ returnnanc S s t d�S )N�nan)r; r r r r �<lambda>� � z%FunctionTests.setUp.<locals>.<lambda>Zreturntoolargeintc S r )N� r r r r r rO � s ZraiseexceptionZisblobr$ c S s
t | t�S r )�
isinstancer? �r, r r r rO � s
Zisnonec S s | d u S r r rS r r r rO � rP Zspam���c W s t | �S r )�lenrS r r r rO � rP zcreate table test(t text))�sqlite�connect�con�create_functionr r
r r r r r r r �executer r r r �setUp� s zFunctionTests.setUpc C s | j �� d S r )rX �closer r r r �tearDown� s zFunctionTests.tearDownc C sF | � tj�� | j�dddd� � W d � d S 1 s w Y d S )N�bla���c S s d| S )N� r rS r r r rO � rP z9FunctionTests.test_func_error_on_create.<locals>.<lambda>)� assertRaisesrV �OperationalErrorrX rY r r r r �test_func_error_on_create� s "�z'FunctionTests.test_func_error_on_createc C s@ dd� }|� }|t � d< | j�dd|� | j�� }|�d� d S )Nc S s dd� } | S )Nc S r r# r r r r r �f� r z=FunctionTests.test_func_ref_count.<locals>.getfunc.<locals>.fr )rd r r r �getfunc� s z2FunctionTests.test_func_ref_count.<locals>.getfuncr Zreftestr zselect reftest())�globalsrX rY �cursorrZ )r! re rd �curr r r �test_func_ref_count� s
z!FunctionTests.test_func_ref_countc C �@ | j �� }|�d� |�� d }| �t|�t� | �|d� d S )Nzselect returntext()r r �rX rg rZ �fetchone� assertEqualr> r9 �r! rh r6 r r r �test_func_return_text� �
z#FunctionTests.test_func_return_textc C s< | j �� }|�d��� d }| �t|�t� | �|d� d S )Nzselect returntextwithnull()r r rk )r! rh �resr r r �$test_func_return_text_with_null_char� s
z2FunctionTests.test_func_return_text_with_null_charc C rj )Nzselect returnunicode()r r rk rn r r r �test_func_return_unicode� rp z&FunctionTests.test_func_return_unicodec C rj )Nzselect returnint()r r )rX rg rZ rl rm r> r: rn r r r �test_func_return_int� rp z"FunctionTests.test_func_return_intc C sR | j �� }|�d� |�� d }| �t|�t� |dk s |dkr'| �d� d S d S )Nzselect returnfloat()r g�&1� @gT㥛� @z wrong value)rX rg rZ rl rm r> r; �failrn r r r �test_func_return_float� s
�z$FunctionTests.test_func_return_floatc C sD | j �� }|�d� |�� d }| �t|�td �� | �|d � d S )Nzselect returnnull()r )rX rg rZ rl rm r> rn r r r �test_func_return_null� s
z#FunctionTests.test_func_return_nullc C rj )Nzselect returnblob()r r )rX rg rZ rl rm r> r? rn r r r �test_func_return_blob� rp z#FunctionTests.test_func_return_blobc C s0 | j �� }|�d� |�� d }| �|d� d S )Nzselect returnlonglong()r r �rX rg rZ rl rm rn r r r �test_func_return_long_long� s
z(FunctionTests.test_func_return_long_longc C s* | j �� }|�d� | �|�� d � d S )Nzselect returnnan()r )rX rg rZ � assertIsNonerl �r! rh r r r �test_func_return_nan� s
z"FunctionTests.test_func_return_nanc C sH | j �� }| �tj�� | j �d� W d � d S 1 s w Y d S )Nzselect returntoolargeint())rX rg ra rV rb rZ r| r r r � test_func_return_too_large_int� s
"�z,FunctionTests.test_func_return_too_large_intc C s^ | j �� }| �tj��}|�d� |�� W d � n1 s w Y | �t|j �d� d S )Nzselect raiseexception()z&user-defined function raised exception�
rX rg ra rV rb rZ rl rm r9 � exception�r! rh �cmr r r �test_func_exception� s
�z!FunctionTests.test_func_exceptionc C �2 | j �� }|�dd� |�� d }| �|d� d S )Nzselect spam(?, ?))r$ r` r r` ry rn r r r �test_any_arguments� �
z FunctionTests.test_any_argumentsc C s" | j �d�}| �|�� d � d S )Nzselect isblob(x'')r )rX rZ �
assertTruerl r| r r r �test_empty_blob s z FunctionTests.test_empty_blobc C s* | j �dtd�f�}| �|�� d � d S )Nzselect isnone(?)rN r )rX rZ r; r� rl r| r r r �test_nan_float s z FunctionTests.test_nan_floatc C s d}| � t|| jjdd� d S )Nz1Python int too large to convert to SQLite INTEGER�select spam(?))rQ )�assertRaisesRegex�
OverflowErrorrX rZ )r! �errr r r �test_too_large_int s �z FunctionTests.test_too_large_intc
C s* | � td| jjdtd�d d d� f� d S )Nz could not convert BLOB to bufferr� r r` )r� �
ValueErrorrX rZ �
memoryviewr r r r �test_non_contiguous_blob s �z&FunctionTests.test_non_contiguous_blobc C s | � td| jjdd� d S )Nzsurrogates not allowedr� )u ��)r� �UnicodeEncodeErrorrX rZ r r r r �test_param_surrogates s �z#FunctionTests.test_param_surrogatesc s� g � � fdd�}| j �dd|� dtfdtfdtfdtfd tftd
�tfd tfd tfd
tfdtfttd��tftd�tfd t d �fg
}|D ]\}}| j �
d|f�}|� � qE| � |� � d S )Nc s � � | t| �f� d S r )�appendr> )�arg��resultsr r �
append_result s z5FunctionTests.test_func_params.<locals>.append_resultZ test_paramsr$ r rT l �{�C l r �inf�textr u ˢqˡⁱᵗᵉr r` zselect test_params(?))
rX rY r: r; r9 r? � bytearray�ranger� r> rZ rl rm )r! r� Zdatasetr6 �_rh r r� r �test_func_params s*
�
z FunctionTests.test_func_params)� � r� z Requires SQLite 3.8.3 or higherc C s� t jjd d�}| jjdd|dd� tjdk r%| j�d� | �|j d� d S | �
tj �� | j�d � W d � d S 1 s=w Y d S )
N�Z return_valueZnondeterministicr F��
deterministic�r� � r z.select nondeterministic() = nondeterministic()r` z>create index t on test(t) where nondeterministic() is not null) �unittest�mock�MockrX rY rV �sqlite_version_inforZ rm �
call_countra rb �r! r� r r r �test_func_non_deterministic9 s
"�z)FunctionTests.test_func_non_deterministicc C s~ t jjd d�}| jjdd|dd� tjdk r%| j�d� | �|j d� d S z | j�d � W d S tj
y> | � d
� Y d S w ) Nr� r� r Tr� r� z(select deterministic() = deterministic()r$ z;create index t on test(t) where deterministic() is not nullz/Unexpected failure while creating partial index) r� r� r� rX rY rV r� rZ rm r� rb ru r� r r r �test_func_deterministicD s
�z%FunctionTests.test_func_deterministiczSQLite < 3.8.3 neededc C sF | � tj�� | jjddtdd� W d � d S 1 s w Y d S )Nr� r Tr� )ra rV ZNotSupportedErrorrX rY r: r r r r �%test_func_deterministic_not_supportedQ s "�z3FunctionTests.test_func_deterministic_not_supportedc C sB | � t�� | j�ddtd� W d � d S 1 sw Y d S )Nr� r T)ra � TypeErrorrX rY r: r r r r �$test_func_deterministic_keyword_onlyV s "�z2FunctionTests.test_func_deterministic_keyword_onlyc C sJ t �d�}dd� }|�dd|� |d�}~~|g}|�|� ~~t� d S )NrM c S r r r )�tr r r �md5sum^ r z=FunctionTests.test_function_destructor_via_gc.<locals>.md5sum�md5r$ z(create table lang (name, first_appeared))rV rW rY r� r )r! �destr� r, �yr r r � test_function_destructor_via_gcZ s
z-FunctionTests.test_function_destructor_via_gcN)"r& r' r( r[ r] rc ri ro rr rs rt rv rw rx rz r} r~ r� r� r� r� r� r� r� r� r� ZskipIfrV r� r� r� r� r� r� r r r r rL � s> !
rL c @ 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#S )$�AggregateTestsc C s� t �d�| _| j�� }|�d� |�ddddd td�f� | j�dd t� | j�d
d t� | j�d d t � | j�d d t
� | j�d
d t � | j�ddt � | j�ddt
� | j�dd t� | j�dd t� d S )NrM z�
create table test(
t text,
i integer,
f float,
n,
b blob
)
z6insert into test(t, i, f, n, b) values (?, ?, ?, ?, ?)r r r r Znostepr$ Z
nofinalizeZexcInitZexcStepZ excFinalizeZ checkTyper` Z
checkTypesrT ZmysumZaggtxt)rV rW rX rg rZ r� Zcreate_aggregater r* r/ r2 r3 r4 rD rF rH r| r r r r[ l s
�zAggregateTests.setUpc C r r r r r r r r] � s zAggregateTests.tearDownc C sB | � tj�� | j�ddt� W d � d S 1 sw Y d S )Nr^ r_ )ra rV rb rX rY rF r r r r �test_aggr_error_on_create� s "�z(AggregateTests.test_aggr_error_on_createc C sT | j �� }| �t��
}|�d� W d � n1 sw Y | �t|j�d� d S )Nzselect nostep(t) from testz+'AggrNoStep' object has no attribute 'step')rX rg ra �AttributeErrorrZ rm r9 r� r� r r r �test_aggr_no_step� s
�z AggregateTests.test_aggr_no_stepc C �b | j �� }| �tj��}|�d� |�� d }W d � n1 s!w Y | �t|j �d� d S )Nz select nofinalize(t) from testr �7user-defined aggregate's 'finalize' method raised errorr �r! rh r� r6 r r r �test_aggr_no_finalize� �
�z$AggregateTests.test_aggr_no_finalizec C r� )Nzselect excInit(t) from testr z7user-defined aggregate's '__init__' method raised errorr r� r r r �test_aggr_exception_in_init� r� z*AggregateTests.test_aggr_exception_in_initc C r� )Nzselect excStep(t) from testr z3user-defined aggregate's 'step' method raised errorr r� r r r �test_aggr_exception_in_step� r� z*AggregateTests.test_aggr_exception_in_stepc C r� )Nz select excFinalize(t) from testr r� r r� r r r � test_aggr_exception_in_finalize� r� z.AggregateTests.test_aggr_exception_in_finalizec C s8 | j �� }|�ddt� f� |�� d }| �|d� d S )Nz select checkTypes('str', ?, ?)r r r` )rX rg rZ r9 rl rm rn r r r �test_aggr_check_param_str� �
z(AggregateTests.test_aggr_check_param_strc C r� )Nzselect checkType('int', ?))r r r$ ry rn r r r �test_aggr_check_param_int� r� z(AggregateTests.test_aggr_check_param_intc C r� )Nz select checkTypes('int', ?, ?))r � r r` ry rn r r r �test_aggr_check_params_int� r� z)AggregateTests.test_aggr_check_params_intc C r� )Nz select checkType('float', ?))r r r$ ry rn r r r �test_aggr_check_param_float� r� z*AggregateTests.test_aggr_check_param_floatc C r� )Nzselect checkType('None', ?)r r r$ ry rn r r r �test_aggr_check_param_none� r� z)AggregateTests.test_aggr_check_param_nonec C s8 | j �� }|�dtd�f� |�� d }| �|d� d S )Nzselect checkType('blob', ?)r r r$ )rX rg rZ r� rl rm rn r r r �test_aggr_check_param_blob� r� z)AggregateTests.test_aggr_check_param_blobc C sJ | j �� }|�d� |�dg d�� |�d� |�� d }| �|d� d S )Nzdelete from testz insert into test(i) values (?)))�
)� )� zselect mysum(i) from testr �< )rX rg rZ Z executemanyrl rm rn r r r �test_aggr_check_aggr_sum� s
z'AggregateTests.test_aggr_check_aggr_sumc C s& | j �d�}|�� d }| �|� d S )Nz1select mysum(i) from (select 1 as i) where i == 0r )rX rZ rl r{ rn r r r �test_aggr_no_match� s z!AggregateTests.test_aggr_no_matchc C sj | j �� }dD ]+}| j|d�� |�d|f� |�� d }| �||� W d � n1 s-w Y qd S )N)r r rJ zselect aggtxt(?) from testr )rX rg ZsubTestrZ rl rm )r! rh rK r6 r r r �test_aggr_text� s
���z AggregateTests.test_aggr_textN)r& r' r( r[ r] r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� k s$ r� c @ s8 e Zd Zedd� �Zdd� Zdd� Zdd� Zd d
� Zd S ) �AuthorizerTestsc C s, | t jkrt jS |dks|dkrt jS t jS �N�c2�t2)rV �
SQLITE_SELECTZ SQLITE_DENY� SQLITE_OK��actionZarg1Zarg2Zdbname�sourcer r r �
authorizer_cb� s
z AuthorizerTests.authorizer_cbc C s6 t �d�| _| j�d� | j�d� | j�| j� d S )NrM z�
create table t1 (c1, c2);
create table t2 (c1, c2);
insert into t1 (c1, c2) values (1, 2);
insert into t2 (c1, c2) values (4, 5);
zselect c2 from t2)rV rW rX Z
executescriptrZ Zset_authorizerr� r r r r r[ � s zAuthorizerTests.setUpc C r r r r r r r r] r zAuthorizerTests.tearDownc C �N | � tj��}| j�d� W d � n1 sw Y | �dt|j�� d S )Nzselect * from t2�
prohibited�ra rV Z
DatabaseErrorrX rZ ZassertInr9 r� �r! r� r r r �test_table_access � �z!AuthorizerTests.test_table_accessc C r� )Nzselect c2 from t1r� r� r� r r r �test_column_access r� z"AuthorizerTests.test_column_accessN) r& r' r( � staticmethodr� r[ r] r� r� r r r r r� � s
r� c @ � e Zd Zedd� �ZdS )� AuthorizerRaiseExceptionTestsc C s( | t jkrt�|dks|dkrt�t jS r� )rV r� r� r� r� r r r r� �
z+AuthorizerRaiseExceptionTests.authorizer_cbN�r& r' r( r� r� r r r r r� � r� c @ r� )�AuthorizerIllegalTypeTestsc C �( | t jkrdS |dks|dkrdS t jS )NrG r� r� �rV r� r� r� r r r r� r� z(AuthorizerIllegalTypeTests.authorizer_cbNr� r r r r r� r� r� c @ r� )�AuthorizerLargeIntegerTestsc C r� )Nl r� r� r� r� r r r r� ! r� z)AuthorizerLargeIntegerTests.authorizer_cbNr� r r r r r� r� r� c C s$ t tttttg} t�dd� | D ��S )Nc S s g | ] }t �� �|��qS r )r� Z
TestLoaderZloadTestsFromTestCase)�.0r� r r r �
<listcomp>4 s zsuite.<locals>.<listcomp>)r� r� r� r� r� rL r� Z TestSuite)Ztestsr r r �suite* s � �r� c C s t �� } | �t� � d S r )r� ZTextTestRunner�runr� )Zrunnerr r r �test7 s r� �__main__)"r� Z
unittest.mockZsqlite3rV Z test.supportr r r
r r r r r r r r r* r/ r2 r3 r4 rD rF rH ZTestCaserL r� r� r� r� r� r� r� r& r r r r �<module> sD
b$
�