File: //proc/self/root/opt/python38/lib/python3.8/sqlite3/test/__pycache__/userfunctions.cpython-38.pyc
U
�i�_�C � @ s� d dl Z d dlZ d dlZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd 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+�ZG d,d-� d-�ZG d.d/� d/�ZG d0d1� d1e j�Z G d2d3� d3e j�Z G d4d5� d5e j�Z G d6d7� d7e �Z G d8d9� d9e �Z G d:d;� d;e �Z!d<d=� Z"d>d?� Z#e$d@k�r�e#� dS )A� Nc C s dS )N�foo� r r r �9/opt/python38/lib/python3.8/sqlite3/test/userfunctions.py�func_returntext 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 C s t | �tkS r
)�type�str��vr r r �
func_isstring- s r c C s t | �tkS r
)r �intr r r r �
func_isint/ s r c C s t | �tkS r
)r �floatr r r r � func_isfloat1 s r c C s t | �t d �kS r
)r r r r r � func_isnone3 s r c C s t | ttf�S r
)�
isinstance�bytes�
memoryviewr r r r � func_isblob5 s r$ c C s t | t�o| dkS r )r! r r r r r �func_islonglong7 s r% c G s t | �S r
)�len)�argsr r r �func: s r( c @ s e Zd Zdd� Zdd� ZdS )�
AggrNoStepc C s d S r
r ��selfr r r �__init__> s zAggrNoStep.__init__c C s dS �N� r r* r r r �finalizeA 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, E s zAggrNoFinalize.__init__c C s d S r
r �r+ �xr r r �stepH s zAggrNoFinalize.stepN)r0 r1 r2 r, r6 r r r r r3 D s r3 c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�AggrExceptionInInitc C s dd d S r r r* r r r r, L s z AggrExceptionInInit.__init__c C s d S r
r r4 r r r r6 O s zAggrExceptionInInit.stepc C s d S r
r r* r r r r/ R s z AggrExceptionInInit.finalizeN�r0 r1 r2 r, r6 r/ r r r r r7 K s r7 c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�AggrExceptionInStepc C s d S r
r r* r r r r, V s z AggrExceptionInStep.__init__c C s dd d S r r r4 r r r r6 Y s zAggrExceptionInStep.stepc C s dS r r r* r r r r/ \ s z AggrExceptionInStep.finalizeNr8 r r r r r9 U s r9 c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�AggrExceptionInFinalizec C s d S r
r r* r r r r, ` s z AggrExceptionInFinalize.__init__c C s d S r
r r4 r r r r6 c s z AggrExceptionInFinalize.stepc C s dd d S r r r* r r r r/ f s z AggrExceptionInFinalize.finalizeNr8 r r r r r: _ s r: c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�
AggrCheckTypec C s
d | _ d S r
��valr* r r r r, j s zAggrCheckType.__init__c C s. t tttd �td�}t|| t|�k�| _d S �N)r r r �NoneZblob�r r r r r" r= )r+ � whichTyper= �theTyper r r r6 m s �zAggrCheckType.stepc C s | j S r
r< r* r r r r/ r s zAggrCheckType.finalizeNr8 r r r r r; i s r; c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�AggrCheckTypesc C s
d| _ d S )Nr r< r* r r r r, v s zAggrCheckTypes.__init__c G s@ t tttd �td�}|D ]"}| jt|| t|�k�7 _qd S r> r@ )r+ rA ZvalsrB r= r r r r6 y s
�zAggrCheckTypes.stepc C s | j S r
r< r* r r r r/ s zAggrCheckTypes.finalizeNr8 r r r r rC u s rC c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�AggrSumc C s
d| _ d S )N� r< r* r r r r, � s zAggrSum.__init__c C s | j |7 _ d S r
r< )r+ r= r r r r6 � s z AggrSum.stepc C s | j S r
r< r* r r r r/ � s zAggrSum.finalizeNr8 r r r r rD � s rD 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&� Ze�ejd'k d(�d)d*� �Ze�ejd'k d(�d+d,� �Ze�ejd'kd-�d.d/� �Z d0d1� Z d2S )3�
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 d t � | j�d
d t
� | j�dd t� | j�dd t� | j�dd t� | j�dd t� | j�ddt� | j�d� d S )N�:memory:Z
returntextr Z
returnunicodeZ returnintZ returnfloatZ
returnnullZ
returnblobZreturnlonglongZraiseexception�isstringr. ZisintZisfloatZisnoneZisblobZ
islonglongZspam���zcreate table test(t text))�sqlite�connect�con�create_functionr r r
r r r r r r r r r r$ r% r( �executer* r r r �setUp� s" zFunctionTests.setUpc C s | j �� d S r
)rL �closer* r r r �tearDown� s zFunctionTests.tearDownc C s0 | � tj�� | j�dddd� � W 5 Q R X d S )N�bla���c S s d| S )N� r )r5 r r r �<lambda>� � z6FunctionTests.CheckFuncErrorOnCreate.<locals>.<lambda>)� assertRaisesrJ �OperationalErrorrL rM 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 )rZ r r r �getfunc� s z0FunctionTests.CheckFuncRefCount.<locals>.getfuncr Zreftestr zselect reftest())�globalsrL rM �cursorrN )r+ r[ rZ �curr r r �CheckFuncRefCount� s
z FunctionTests.CheckFuncRefCountc C s@ | j �� }|�d� |�� d }| �t|�t� | �|d� d S )Nzselect returntext()r r �rL r] rN �fetchone� assertEqualr r �r+ r^ r= r r r �CheckFuncReturnText� s
z!FunctionTests.CheckFuncReturnTextc C s@ | j �� }|�d� |�� d }| �t|�t� | �|d� d S )Nzselect returnunicode()r r r` rc r r r �CheckFuncReturnUnicode� s
z$FunctionTests.CheckFuncReturnUnicodec C s@ | j �� }|�d� |�� d }| �t|�t� | �|d� d S )Nzselect returnint()r r )rL r] rN ra rb r r rc 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)rL r] rN ra rb r r �failrc r r r �CheckFuncReturnFloat� s
z"FunctionTests.CheckFuncReturnFloatc C sD | j �� }|�d� |�� d }| �t|�td �� | �|d � d S )Nzselect returnnull()r )rL r] rN ra rb r rc r r r �CheckFuncReturnNull� s
z!FunctionTests.CheckFuncReturnNullc C s@ | j �� }|�d� |�� d }| �t|�t� | �|d� d S )Nzselect returnblob()r r )rL r] rN ra rb r r" rc r r r �CheckFuncReturnBlob� s
z!FunctionTests.CheckFuncReturnBlobc C s0 | j �� }|�d� |�� d }| �|d� d S )Nzselect returnlonglong()r r �rL r] rN ra rb rc r r r �CheckFuncReturnLongLong� s
z%FunctionTests.CheckFuncReturnLongLongc C sJ | j �� }| �tj��}|�d� |�� W 5 Q R X | �t|j �d� d S )Nzselect raiseexception()z&user-defined function raised exception�
rL r] rW rJ rX rN ra rb r � exception�r+ r^ �cmr r r �CheckFuncException� s
z FunctionTests.CheckFuncExceptionc C s2 | j �� }|�dd� |�� d }| �|d� d S )Nzselect isstring(?)�r r r. rk rc r r r �CheckParamString� s
z FunctionTests.CheckParamStringc C s2 | j �� }|�dd� |�� d }| �|d� d S )Nzselect isint(?)�r r r. rk rc r r r �
CheckParamInt� s
zFunctionTests.CheckParamIntc C s2 | j �� }|�dd� |�� d }| �|d� d S )Nzselect isfloat(?)�r r r. rk rc r r r �CheckParamFloat� s
z FunctionTests.CheckParamFloatc C s2 | j �� }|�dd� |�� d }| �|d� d S )Nzselect isnone(?)r
r r. rk rc r r r �CheckParamNone� s
z FunctionTests.CheckParamNonec C s8 | j �� }|�dtd�f� |�� d }| �|d� d S )Nzselect isblob(?)r r r. �rL r] rN r# ra rb rc r r r �CheckParamBlob s
z FunctionTests.CheckParamBlobc C s2 | j �� }|�dd� |�� d }| �|d� d S )Nzselect islonglong(?))l r r. rk rc r r r �CheckParamLongLong s
z FunctionTests.CheckParamLongLongc C s2 | j �� }|�dd� |�� d }| �|d� d S )Nzselect spam(?, ?))r. rT r rT rk rc r r r �CheckAnyArguments s
z FunctionTests.CheckAnyArguments)� � r} z Requires SQLite 3.8.3 or higherc C sp t jjd d�}| jjdd|dd� tjdk rH| j�d� | �|j d� n$| �
tj �� | j�d � W 5 Q R X d S )
N�Z return_valueZnondeterministicr F��
deterministic�r} � r z.select nondeterministic() = nondeterministic()rT z>create index t on test(t) where nondeterministic() is not null) �unittest�mock�MockrL rM rJ �sqlite_version_inforN rb �
call_countrW rX �r+ r� r r r �CheckFuncNonDeterministic s
z'FunctionTests.CheckFuncNonDeterministicc C s~ t jjd d�}| jjdd|dd� tjdk rH| j�d� | �|j d� n2z| j�d � W n tj
k
rx | � d
� Y nX 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� rL rM rJ r� rN rb r� rX rg r� r r r �CheckFuncDeterministic) s
z$FunctionTests.CheckFuncDeterministiczSQLite < 3.8.3 neededc C s0 | � tj�� | jjddtdd� W 5 Q R X d S )Nr� r Tr� )rW rJ ZNotSupportedErrorrL rM r r* r r r �"CheckFuncDeterministicNotSupported6 s z0FunctionTests.CheckFuncDeterministicNotSupportedc C s, | � t�� | j�ddtd� W 5 Q R X d S )Nr� r T)rW � TypeErrorrL rM r r* r r r �!CheckFuncDeterministicKeywordOnly; s z/FunctionTests.CheckFuncDeterministicKeywordOnlyN) r0 r1 r2 rO rQ rY r_ rd re rf rh ri rj rl rq rs ru rw rx rz r{ r| r� ZskipIfrJ r� r� r� r� r� r r r r rF � s4
rF 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 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� d S )NrG 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 checkTyperT Z
checkTypesrI Zmysum)rJ rK rL r] rN r# Zcreate_aggregater) r3 r7 r9 r: r; rC rD )r+ r^ r r r rO A s
�zAggregateTests.setUpc C s d S r
r r* r r r rQ Y s zAggregateTests.tearDownc C s, | � tj�� | j�ddt� W 5 Q R X d S )NrR rS )rW rJ rX rL rM rD r* r r r �CheckAggrErrorOnCreate^ s z%AggregateTests.CheckAggrErrorOnCreatec C s@ | j �� }| �t��}|�d� W 5 Q R X | �t|j�d� d S )Nzselect nostep(t) from testz+'AggrNoStep' object has no attribute 'step')rL r] rW �AttributeErrorrN rb r rn ro r r r �CheckAggrNoStepb s
z AggregateTests.CheckAggrNoStepc C sN | j �� }| �tj�� }|�d� |�� d }W 5 Q R X | �t|j �d� d S )Nz select nofinalize(t) from testr �7user-defined aggregate's 'finalize' method raised errorrm �r+ r^ rp r= r r r �CheckAggrNoFinalizeh s
z"AggregateTests.CheckAggrNoFinalizec C sN | j �� }| �tj�� }|�d� |�� d }W 5 Q R X | �t|j �d� d S )Nzselect excInit(t) from testr z7user-defined aggregate's '__init__' method raised errorrm r� r r r �CheckAggrExceptionInInito s
z'AggregateTests.CheckAggrExceptionInInitc C sN | j �� }| �tj�� }|�d� |�� d }W 5 Q R X | �t|j �d� d S )Nzselect excStep(t) from testr z3user-defined aggregate's 'step' method raised errorrm r� r r r �CheckAggrExceptionInStepv s
z'AggregateTests.CheckAggrExceptionInStepc C sN | j �� }| �tj�� }|�d� |�� d }W 5 Q R X | �t|j �d� d S )Nz select excFinalize(t) from testr r� rm r� r r r � CheckAggrExceptionInFinalize} s
z+AggregateTests.CheckAggrExceptionInFinalizec C s2 | j �� }|�dd� |�� d }| �|d� d S )Nzselect checkType('str', ?)rr r r. rk rc r r r �CheckAggrCheckParamStr� s
z%AggregateTests.CheckAggrCheckParamStrc C s2 | j �� }|�dd� |�� d }| �|d� d S )Nzselect checkType('int', ?)rt r r. rk rc r r r �CheckAggrCheckParamInt� s
z%AggregateTests.CheckAggrCheckParamIntc C s2 | j �� }|�dd� |�� d }| �|d� d S )Nz select checkTypes('int', ?, ?))r � r rT rk rc r r r �CheckAggrCheckParamsInt� s
z&AggregateTests.CheckAggrCheckParamsIntc C s2 | j �� }|�dd� |�� d }| �|d� d S )Nz select checkType('float', ?)rv r r. rk rc r r r �CheckAggrCheckParamFloat� s
z'AggregateTests.CheckAggrCheckParamFloatc C s2 | j �� }|�dd� |�� d }| �|d� d S )Nzselect checkType('None', ?)r
r r. rk rc r r r �CheckAggrCheckParamNone� s
z&AggregateTests.CheckAggrCheckParamNonec C s8 | j �� }|�dtd�f� |�� d }| �|d� d S )Nzselect checkType('blob', ?)r r r. ry rc r r r �CheckAggrCheckParamBlob� s
z&AggregateTests.CheckAggrCheckParamBlobc C sL | j �� }|�d� |�ddddg� |�d� |�� d }| �|d� d S ) Nzdelete from testz insert into test(i) values (?))�
)� )� zselect mysum(i) from testr �< )rL r] rN Z executemanyra rb rc r r r �CheckAggrCheckAggrSum� s
z$AggregateTests.CheckAggrCheckAggrSumN)r0 r1 r2 rO rQ r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� @ 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)rJ �
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 )NrG 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)rJ rK rL Z
executescriptrN Zset_authorizerr� r* r r r rO � s zAuthorizerTests.setUpc C s d S r
r r* r r r rQ � s zAuthorizerTests.tearDownc C s: | � tj��}| j�d� W 5 Q R X | �dt|j�� d S )Nzselect * from t2�
prohibited�rW rJ Z
DatabaseErrorrL rN ZassertInr rn �r+ rp r r r �test_table_access� s z!AuthorizerTests.test_table_accessc C s: | � tj��}| j�d� W 5 Q R X | �dt|j�� d S )Nzselect c2 from t1r� r� r� r r r �test_column_access� s z"AuthorizerTests.test_column_accessN) r0 r1 r2 � staticmethodr� rO rQ 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� )rJ r� �
ValueErrorr� r� r r r r� � s
z+AuthorizerRaiseExceptionTests.authorizer_cbN�r0 r1 r2 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 )NrE r� r� �rJ 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 makeSuiterF 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.mockZsqlite3rJ r r r
r r r r r r r r r r$ r% r( r) r3 r7 r9 r: r; rC rD ZTestCaserF r� r� r� r� r� r� r� r0 r r r r �<module> sH
5p$