%PDF-
%PDF-
Mini Shell
Mini Shell
U
p©ßaË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@kr†e#ƒ dS )Aé Nc C s dS )NÚfoo© r r r ú6/usr/local/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$
Zerion Mini Shell 1.0