Sindbad~EG File Manager
a
��g*J � @ sh d dl Z d dlZ d dlZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z G dd� d�Z
G 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�ZG d.d/� d/e�ZG d0d1� d1e�Z d2d3� Z d4d5� Z e d6k�rde � dS )7� Nc C s dS )N�foo� r r r �6/usr/local/lib/python3.9/sqlite3/test/userfunctions.py�func_returntext s r c C s dS )N�1 2r r r r r �func_returntextwithnull s r c C s dS )N�barr r r r r �func_returnunicode s r c C s dS �N�* r r r r r �func_returnint" s r c C s dS )N� ��Q� @r r r r r �func_returnfloat$ s r c C s d S �Nr r r r r �func_returnnull& s r c C s dS )N� blobr r r r r �func_returnblob( s r c C s dS )N� r r r r r �func_returnlonglong* s r c C s dd d S �N� r r r r r r �func_raiseexception, s r c @ s e Zd Zdd� Zdd� ZdS )�
AggrNoStepc C s d S r r ��selfr r r �__init__0 s zAggrNoStep.__init__c C s dS �N� r r r r r �finalize3 s zAggrNoStep.finalizeN)�__name__�
__module__� __qualname__r r r r r r r / s r c @ s e Zd Zdd� Zdd� ZdS )�AggrNoFinalizec C s d S r r r r r r r 7 s zAggrNoFinalize.__init__c C s d S r r �r �xr r r �step: s zAggrNoFinalize.stepN)r r r! r r% r r r r r" 6 s r" c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�AggrExceptionInInitc C s dd d S r r r r r r r > s z AggrExceptionInInit.__init__c C s d S r r r# r r r r% A s zAggrExceptionInInit.stepc C s d S r r r r r r r D s z AggrExceptionInInit.finalizeN�r r r! r r% r r r r r r& = s r& c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�AggrExceptionInStepc C s d S r r r r r r r H s z AggrExceptionInStep.__init__c C s dd d S r r r# r r r r% K s zAggrExceptionInStep.stepc C s dS r
r r r r r r N s z AggrExceptionInStep.finalizeNr' r r r r r( G s r( c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�AggrExceptionInFinalizec C s d S r r r r r r r R s z AggrExceptionInFinalize.__init__c C s d S r r r# r r r r% U s z AggrExceptionInFinalize.stepc C s dd d S r r r r r r r X s z AggrExceptionInFinalize.finalizeNr' r r r r r) Q s r) c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�
AggrCheckTypec C s
d | _ d S r ��valr r r r r \ s zAggrCheckType.__init__c C s. t tttd �td�}t|| t|�u �| _d S �N)�str�int�float�NoneZblob�r. r/ r0 �type�bytesr, )r � whichTyper, �theTyper r r r% _ s �zAggrCheckType.stepc C s | j S r r+ r r r r r d s zAggrCheckType.finalizeNr' r r r r r* [ s r* c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�AggrCheckTypesc C s
d| _ d S )Nr r+ r r r r r h s zAggrCheckTypes.__init__c G s@ t tttd �td�}|D ]"}| jt|| t|�u �7 _qd S r- r2 )r r5 Zvalsr6 r, r r r r% k s
�zAggrCheckTypes.stepc C s | j S r r+ r r r r r q s zAggrCheckTypes.finalizeNr' r r r r r7 g s r7 c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�AggrSumc C s
d| _ d S )N� r+ r r r r r u s zAggrSum.__init__c C s | j |7 _ d S r r+ )r r, r r r r% x s z AggrSum.stepc C s | j S r r+ r r r r r { s zAggrSum.finalizeNr' r r r r r8 t s r8 c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�AggrTextc C s
d| _ d S )N� ��txtr r r r r s zAggrText.__init__c C s | j | | _ d S r r< )r r= r r r r% � s z
AggrText.stepc C s | j S r r< r r r r r � s zAggrText.finalizeNr' r r r r r: ~ s r: 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 d8S )9�
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)r0 r r r r �<lambda>� � z%FunctionTests.setUp.<locals>.<lambda>Zreturntoolargeintc S s dS )N� r r r r r rA � rB ZraiseexceptionZisblobr c S s
t | t�S r )�
isinstancer4 �r$ r r r rA � rB Zisnonec S s | d u S r r rE r r r rA � rB Zspam���c W s t | �S r )�lenrE r r r rA � rB 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 )rJ �closer r r r �tearDown� s zFunctionTests.tearDownc C sD | � tj��$ | j�dddd� � W d � n1 s60 Y d S )N�bla���c S s d| S )N� r rE r r r rA � rB z6FunctionTests.CheckFuncErrorOnCreate.<locals>.<lambda>)� assertRaisesrH �OperationalErrorrJ rK r r r r �CheckFuncErrorOnCreate� s z$FunctionTests.CheckFuncErrorOnCreatec C s@ dd� }|� }|t � d< | j�dd|� | j�� }|�d� d S )Nc S s dd� } | S )Nc S s dS r r r r r r �f� s z;FunctionTests.CheckFuncRefCount.<locals>.getfunc.<locals>.fr )rV r r r �getfunc� s z0FunctionTests.CheckFuncRefCount.<locals>.getfuncr Zreftestr zselect reftest())�globalsrJ rK �cursorrL )r rW rV �curr r r �CheckFuncRefCount� s
z FunctionTests.CheckFuncRefCountc C s@ | j �� }|�d� |�� d }| �t|�t� | �|d� d S )Nzselect returntext()r r �rJ rY rL �fetchone� assertEqualr3 r. �r rZ r, r r r �CheckFuncReturnText� s
z!FunctionTests.CheckFuncReturnTextc C s< | j �� }|�d��� d }| �t|�t� | �|d� d S )Nzselect returntextwithnull()r r r\ )r rZ �resr r r � CheckFuncReturnTextWithNullChar� s
z-FunctionTests.CheckFuncReturnTextWithNullCharc C s@ | j �� }|�d� |�� d }| �t|�t� | �|d� d S )Nzselect returnunicode()r r r\ r_ r r r �CheckFuncReturnUnicode� s
z$FunctionTests.CheckFuncReturnUnicodec C s@ | j �� }|�d� |�� d }| �t|�t� | �|d� d S )Nzselect returnint()r r )rJ rY rL r] r^ r3 r/ r_ r r r �CheckFuncReturnInt� s
z FunctionTests.CheckFuncReturnIntc C sN | j �� }|�d� |�� d }| �t|�t� |dk s@|dkrJ| �d� d S )Nzselect returnfloat()r g�&1� @gT㥛� @z wrong value)rJ rY rL r] r^ r3 r0 �failr_ r r r �CheckFuncReturnFloat� s
z"FunctionTests.CheckFuncReturnFloatc C sD | j �� }|�d� |�� d }| �t|�td �� | �|d � d S )Nzselect returnnull()r )rJ rY rL r] r^ r3 r_ r r r �CheckFuncReturnNull� s
z!FunctionTests.CheckFuncReturnNullc C s@ | j �� }|�d� |�� d }| �t|�t� | �|d� d S )Nzselect returnblob()r r )rJ rY rL r] r^ r3 r4 r_ r r r �CheckFuncReturnBlob� s
z!FunctionTests.CheckFuncReturnBlobc C s0 | j �� }|�d� |�� d }| �|d� d S )Nzselect returnlonglong()r r �rJ rY rL r] r^ r_ r r r �CheckFuncReturnLongLong� s
z%FunctionTests.CheckFuncReturnLongLongc C s* | j �� }|�d� | �|�� d � d S )Nzselect returnnan()r )rJ rY rL Z assertIsNoner] �r rZ r r r �CheckFuncReturnNaN� s
z FunctionTests.CheckFuncReturnNaNc C sF | j �� }| �tj�� | j �d� W d � n1 s80 Y d S )Nzselect returntoolargeint())rJ rY rS rH rT rL rk r r r �CheckFuncReturnTooLargeInt� s
z(FunctionTests.CheckFuncReturnTooLargeIntc C s^ | j �� }| �tj��"}|�d� |�� W d � n1 s>0 Y | �t|j �d� d S )Nzselect raiseexception()z&user-defined function raised exception�
rJ rY rS rH rT rL r] r^ r. � exception�r rZ �cmr r r �CheckFuncException� s
&z FunctionTests.CheckFuncExceptionc C s2 | j �� }|�dd� |�� d }| �|d� d S )Nzselect spam(?, ?))r rR r rR ri r_ r r r �CheckAnyArguments� s
z FunctionTests.CheckAnyArgumentsc C s" | j �d�}| �|�� d � d S )Nzselect isblob(x'')r )rJ rL �
assertTruer] rk r r r �CheckEmptyBlob� s z FunctionTests.CheckEmptyBlobc C s* | j �dtd�f�}| �|�� d � d S )Nzselect isnone(?)r@ r )rJ rL r0 rt r] rk r r r �
CheckNaNFloat s zFunctionTests.CheckNaNFloatc C s d}| � t|| jjdd� d S )Nz1Python int too large to convert to SQLite INTEGER�select spam(?))rC )�assertRaisesRegex�
OverflowErrorrJ rL )r �errr r r �CheckTooLargeInt s �z FunctionTests.CheckTooLargeIntc
C s* | � td| jjdtd�d d d� f� d S )Nz could not convert BLOB to bufferrw r rR )rx �
ValueErrorrJ rL �
memoryviewr r r r �CheckNonContiguousBlob s �z$FunctionTests.CheckNonContiguousBlobc C s | � td| jjdd� d S )Nzsurrogates not allowedrw )u ��)rx �UnicodeEncodeErrorrJ rL r r r r �CheckParamSurrogates s �z"FunctionTests.CheckParamSurrogatesc s� g � � fdd�}| j �dd|� dtfdtfdtfdtfd tftd
�tfd tfd tfd
tfdtfttd��tftd�tfd t d �fg
}|D ] \}}| j �
d|f�}|� � q�| � |� � d S )Nc s � � | t| �f� d S r )�appendr3 )�arg��resultsr r �
append_result s z4FunctionTests.CheckFuncParams.<locals>.append_resultZ test_paramsr r rF l �{�C l r
�inf�textr u ˢqˡⁱᵗᵉr rR zselect test_params(?))
rJ rK r/ r0 r. r4 � bytearray�ranger} r3 rL r] r^ )r r� Zdatasetr, �_rZ r r� r �CheckFuncParams s*
�
z FunctionTests.CheckFuncParams)� � r� z Requires SQLite 3.8.3 or higherc C s� t jjd d�}| jjdd|dd� tjdk rH| j�d� | �|j d� n8| �
tj �� | j�d � W d � n1 sv0 Y d S )
N�Z return_valueZnondeterministicr F��
deterministic�r� � r z.select nondeterministic() = nondeterministic()rR z>create index t on test(t) where nondeterministic() is not null) �unittest�mock�MockrJ rK rH �sqlite_version_inforL r^ �
call_countrS rT �r r� r r r �CheckFuncNonDeterministic6 s
z'FunctionTests.CheckFuncNonDeterministicc C s| t jjd d�}| jjdd|dd� tjdk rH| j�d� | �|j d� n0z| j�d � W n tj
yv | � d
� Y n0 d S ) 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� rJ rK rH r� rL r^ r� rT re r� r r r �CheckFuncDeterministicA s
z$FunctionTests.CheckFuncDeterministiczSQLite < 3.8.3 neededc C sD | � tj��$ | jjddtdd� W d � n1 s60 Y d S )Nr� r Tr� )rS rH ZNotSupportedErrorrJ rK r/ r r r r �"CheckFuncDeterministicNotSupportedN s z0FunctionTests.CheckFuncDeterministicNotSupportedc C s@ | � t��" | j�ddtd� W d � n1 s20 Y d S )Nr� r T)rS � TypeErrorrJ rK r/ r r r r �!CheckFuncDeterministicKeywordOnlyS s z/FunctionTests.CheckFuncDeterministicKeywordOnlyN)!r r r! rM rO rU r[ r` rb rc rd rf rg rh rj rl rm rr rs ru rv r{ r~ r� r� r� ZskipIfrH r� r� r� r� r� r r r r r> � s: !
r> 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!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 )Nr? 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 checkTyperR Z
checkTypesrF ZmysumZaggtxt)rH rI rJ rY rL r} Zcreate_aggregater r" r& r( r) r* r7 r8 r: rk r r r rM Y s
�zAggregateTests.setUpc C s d S r r r r r r rO r s zAggregateTests.tearDownc C s@ | � tj�� | j�ddt� W d � n1 s20 Y d S )NrP rQ )rS rH rT rJ rK r8 r r r r �CheckAggrErrorOnCreatew s z%AggregateTests.CheckAggrErrorOnCreatec C sT | j �� }| �t��}|�d� W d � n1 s40 Y | �t|j�d� d S )Nzselect nostep(t) from testz+'AggrNoStep' object has no attribute 'step')rJ rY rS �AttributeErrorrL r^ r. ro rp r r r �CheckAggrNoStep{ s
(z AggregateTests.CheckAggrNoStepc C sb | j �� }| �tj��&}|�d� |�� d }W d � n1 sB0 Y | �t|j �d� d S )Nz select nofinalize(t) from testr �7user-defined aggregate's 'finalize' method raised errorrn �r rZ rq r, r r r �CheckAggrNoFinalize� s
*z"AggregateTests.CheckAggrNoFinalizec C sb | j �� }| �tj��&}|�d� |�� d }W d � n1 sB0 Y | �t|j �d� d S )Nzselect excInit(t) from testr z7user-defined aggregate's '__init__' method raised errorrn r� r r r �CheckAggrExceptionInInit� s
*z'AggregateTests.CheckAggrExceptionInInitc C sb | j �� }| �tj��&}|�d� |�� d }W d � n1 sB0 Y | �t|j �d� d S )Nzselect excStep(t) from testr z3user-defined aggregate's 'step' method raised errorrn r� r r r �CheckAggrExceptionInStep� s
*z'AggregateTests.CheckAggrExceptionInStepc C sb | j �� }| �tj��&}|�d� |�� d }W d � n1 sB0 Y | �t|j �d� d S )Nz select excFinalize(t) from testr r� rn r� r r r � CheckAggrExceptionInFinalize� s
*z+AggregateTests.CheckAggrExceptionInFinalizec C s8 | j �� }|�ddt� f� |�� d }| �|d� d S )Nz select checkTypes('str', ?, ?)r r rR )rJ rY rL r. r] r^ r_ r r r �CheckAggrCheckParamStr� s
z%AggregateTests.CheckAggrCheckParamStrc C s2 | j �� }|�dd� |�� d }| �|d� d S )Nzselect checkType('int', ?))r r r ri r_ r r r �CheckAggrCheckParamInt� s
z%AggregateTests.CheckAggrCheckParamIntc C s2 | j �� }|�dd� |�� d }| �|d� d S )Nz select checkTypes('int', ?, ?))r � r rR ri r_ r r r �CheckAggrCheckParamsInt� s
z&AggregateTests.CheckAggrCheckParamsIntc C s2 | j �� }|�dd� |�� d }| �|d� d S )Nz select checkType('float', ?))r
r r ri r_ r r r �CheckAggrCheckParamFloat� s
z'AggregateTests.CheckAggrCheckParamFloatc C s2 | j �� }|�dd� |�� d }| �|d� d S )Nzselect checkType('None', ?)r r r ri r_ r r r �CheckAggrCheckParamNone� s
z&AggregateTests.CheckAggrCheckParamNonec C s8 | j �� }|�dtd�f� |�� d }| �|d� d S )Nzselect checkType('blob', ?)r r r )rJ rY rL r} r] r^ r_ r r r �CheckAggrCheckParamBlob� s
z&AggregateTests.CheckAggrCheckParamBlobc C sJ | j �� }|�d� |�dg d�� |�d� |�� d }| �|d� d S )Nzdelete from testz insert into test(i) values (?)))�
)� )� zselect mysum(i) from testr �< )rJ rY rL Z executemanyr] r^ r_ r r r �CheckAggrCheckAggrSum� s
z$AggregateTests.CheckAggrCheckAggrSumc C sj | j �� }dD ]V}| j|d��6 |�d|f� |�� d }| �||� W d � q1 sZ0 Y qd S )N)r r r< zselect aggtxt(?) from testr )rJ rY ZsubTestrL r] r^ )r rZ r= r, r r r �
CheckAggrText� s
z AggregateTests.CheckAggrTextN)r r r! rM rO r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� X s r� c @ s8 e Zd Zedd� �Zdd� Zdd� Zdd� Zd d
� Zd S ) �AuthorizerTestsc C s, | t jkrt jS |dks |dkr&t jS t jS �N�c2�t2)rH �
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 )Nr? 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)rH rI rJ Z
executescriptrL Zset_authorizerr� r r r r rM � s zAuthorizerTests.setUpc C s d S r r r r r r rO � s zAuthorizerTests.tearDownc C sN | � tj�� }| j�d� W d � n1 s.0 Y | �dt|j�� d S )Nzselect * from t2�
prohibited�rS rH Z
DatabaseErrorrJ rL ZassertInr. ro �r rq r r r �test_table_access� s *z!AuthorizerTests.test_table_accessc C sN | � tj�� }| j�d� W d � n1 s.0 Y | �dt|j�� d S )Nzselect c2 from t1r� r� r� r r r �test_column_access� s *z"AuthorizerTests.test_column_accessN) r r r! � staticmethodr� rM rO r� r� r r r r r� � s
r� c @ s e Zd Zedd� �ZdS )� AuthorizerRaiseExceptionTestsc C s( | t jkrt�|dks |dkr"t�t jS r� )rH r� r| r� r� r r r r� � s
z+AuthorizerRaiseExceptionTests.authorizer_cbN�r r r! r� r� r r r r r� � s r� c @ s e Zd Zedd� �ZdS )�AuthorizerIllegalTypeTestsc C s( | t jkrdS |dks |dkr"dS t jS )Nr9 r� r� �rH r� r� r� r r r r� s
z(AuthorizerIllegalTypeTests.authorizer_cbNr� r r r r r� � s r� c @ s e Zd Zedd� �ZdS )�AuthorizerLargeIntegerTestsc C s( | t jkrdS |dks |dkr"dS t jS )Nl r� r� r� r� r r r r� s
z)AuthorizerLargeIntegerTests.authorizer_cbNr� r r r r r� s r� c
C sJ t �td�} t �td�}t �t�}t �| ||t �t�t �t�t �t�f�S )NZCheck) r� Z makeSuiter> r� r� Z TestSuiter� r� r� )Zfunction_suiteZaggregate_suiteZauthorizer_suiter r r �suite s
�r� c C s t �� } | �t� � d S r )r� ZTextTestRunner�runr� )Zrunnerr r r �test s r� �__main__) r� Z
unittest.mockZsqlite3rH r r r r r r r r r r r" r& r( r) r* r7 r8 r: ZTestCaser> r� r� r� r� r� r� r� r r r r r �<module> s>
Rz$
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists