EVOLUTION-NINJA
Edit File: 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$