Your IP : 216.73.216.213
�
e f�: � @ s d d l Z d d l Z d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z Gd d � d � Z Gd d! � d! � Z Gd" d# � d# � Z Gd$ d% � d% � Z Gd& d' � d' � Z Gd( d) � d) � Z Gd* d+ � d+ � Z Gd, d- � d- e j � Z Gd. d/ � d/ e j � Z Gd0 d1 � d1 e j � Z Gd2 d3 � d3 e � Z Gd4 d5 � d5 e � Z Gd6 d7 � d7 e � Z d8 d9 � Z d: d; � Z e! d< k r e � n d S)=� Nc C s d S)N�foo� r r r �?/opt/alt/python34/lib64/python3.4/sqlite3/test/userfunctions.py�func_returntext s r c C s d S)N�barr r r r r �func_returnunicode s r c C s d S)N�* r r r r r �func_returnint s r c C s d S)Ng ��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 d S)Ns blobr r r r r �func_returnblob& s r c C s d S)N� � l r r r r r �func_returnlonglong( s r c C s d d d S)N� r r r r r r �func_raiseexception* s r c C s t | � t k S)N)�type�str)�vr r r �
func_isstring- s r c C s t | � t k S)N)r �int)r r r r �
func_isint/ s r c C s t | � t k S)N)r �float)r r r r � func_isfloat1 s r c C s t | � t d � k S)N)r )r r r r � func_isnone3 s r c C s t | t t f � S)N)�
isinstance�bytes�
memoryview)r r r r � func_isblob5 s r c C s t | t � o | d k S)Nr
r l )r r )r r r r �func_islonglong7 s r c @ s( e Z d Z d d � Z d d � Z d S)�
AggrNoStepc C s d S)Nr )�selfr r r �__init__; s zAggrNoStep.__init__c C s d S)Nr
r )r! r r r �finalize> s zAggrNoStep.finalizeN)�__name__�
__module__� __qualname__r" r# r r r r r : s r c @ s( e Z d Z d d � Z d d � Z d S)�AggrNoFinalizec C s d S)Nr )r! r r r r" B s zAggrNoFinalize.__init__c C s d S)Nr )r! �xr r r �stepE s zAggrNoFinalize.stepN)r$ r% r& r" r) r r r r r' A s r' c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�AggrExceptionInInitc C s d d d S)Nr r r )r! r r r r" I s z AggrExceptionInInit.__init__c C s d S)Nr )r! r( r r r r) L s zAggrExceptionInInit.stepc C s d S)Nr )r! r r r r# O s z AggrExceptionInInit.finalizeN)r$ r% r& r" r) r# r r r r r* H s r* c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�AggrExceptionInStepc C s d S)Nr )r! r r r r" S s z AggrExceptionInStep.__init__c C s d d d S)Nr r r )r! r( r r r r) V s zAggrExceptionInStep.stepc C s d S)Nr r )r! r r r r# Y s z AggrExceptionInStep.finalizeN)r$ r% r& r" r) r# r r r r r+ R s r+ c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�AggrExceptionInFinalizec C s d S)Nr )r! r r r r" ] s z AggrExceptionInFinalize.__init__c C s d S)Nr )r! r( r r r r) ` s z AggrExceptionInFinalize.stepc C s d d d S)Nr r r )r! r r r r# c s z AggrExceptionInFinalize.finalizeN)r$ r% r& r" r) r# r r r r r, \ s r, c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�
AggrCheckTypec C s
d | _ d S)N)�val)r! r r r r" g s zAggrCheckType.__init__c C sR i t d 6t d 6t d 6t d � d 6t d 6} t | | t | � k � | _ d S)Nr r r �NoneZblob)r r r r r r. )r! Z whichTyper. ZtheTyper r r r) j s %
zAggrCheckType.stepc C s | j S)N)r. )r! r r r r# o s zAggrCheckType.finalizeN)r$ r% r& r" r) r# r r r r r- f s r- c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�AggrSumc C s
d | _ d S)Ng )r. )r! r r r r" s s zAggrSum.__init__c C s | j | 7_ d S)N)r. )r! r. r r r r) v s z AggrSum.stepc C s | j S)N)r. )r! r r r r# y s zAggrSum.finalizeN)r$ r% r& r" r) r# r r r r r0 r s r0 c @ s� e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d � Z d! d" � Z d# d$ � Z d% S)&�
FunctionTestsc C sJ t j d � | _ | j j d d t � | j j d d t � | j j d d t � | j j d d t � | j j d d t � | j j d d t � | j j d d t
� | j j d
d t � | j j d d t � | j j d
d t
� | j j d d t � | j j d d t � | j j d d t � | j j d d t � d S)Nz:memory:Z
returntextr Z
returnunicodeZ returnintZ returnfloatZ
returnnullZ
returnblobZreturnlonglongZraiseexception�isstringr
ZisintZisfloatZisnoneZisblobZ
islonglong)�sqlite�connect�con�create_functionr r r r
r r r r r r r r r r )r! r r r �setUp} s zFunctionTests.setUpc C s | j j � d S)N)r5 �close)r! r r r �tearDown� s zFunctionTests.tearDownc C sI y- | j j d d d d � � | j d � Wn t j k
rD Yn Xd S)N�bla�d c S s d | S)N� r )r( r r r �<lambda>� s z6FunctionTests.CheckFuncErrorOnCreate.<locals>.<lambda>z&should have raised an OperationalErrori����)r5 r6 �failr3 �OperationalError)r! r r r �CheckFuncErrorOnCreate� s
z$FunctionTests.CheckFuncErrorOnCreatec C sX d d � } | � } | t � d <| j j d d | � | j j � } | j d � d S)Nc S s d d � } | S)Nc S s d S)Nr
r r r r r �f� s z;FunctionTests.CheckFuncRefCount.<locals>.getfunc.<locals>.fr )rA r r r �getfunc� s z0FunctionTests.CheckFuncRefCount.<locals>.getfuncr Zreftestr zselect reftest())�globalsr5 r6 �cursor�execute)r! rB rA �curr r r �CheckFuncRefCount� s
z FunctionTests.CheckFuncRefCountc C sV | j j � } | j d � | j � d } | j t | � t � | j | d � d S)Nzselect returntext()r r )r5 rD rE �fetchone� assertEqualr r )r! rF r. r r r �CheckFuncReturnText� s
z!FunctionTests.CheckFuncReturnTextc C sV | j j � } | j d � | j � d } | j t | � t � | j | d � d S)Nzselect returnunicode()r r )r5 rD rE rH rI r r )r! rF r. r r r �CheckFuncReturnUnicode� s
z$FunctionTests.CheckFuncReturnUnicodec C sV | j j � } | j d � | j � d } | j t | � t � | j | d � d S)Nzselect returnint()r r )r5 rD rE rH rI r r )r! rF r. r r r �CheckFuncReturnInt� s
z FunctionTests.CheckFuncReturnIntc C sn | j j � } | j d � | j � d } | j t | � t � | d k sZ | d k rj | j d � n d S)Nzselect returnfloat()r g�&1� @gT㥛� @z wrong value)r5 rD rE rH rI r r r> )r! rF r. r r r �CheckFuncReturnFloat� s
z"FunctionTests.CheckFuncReturnFloatc C s\ | j j � } | j d � | j � d } | j t | � t d � � | j | d � d S)Nzselect returnnull()r )r5 rD rE rH rI r )r! rF r. r r r �CheckFuncReturnNull� s
z!FunctionTests.CheckFuncReturnNullc C sV | j j � } | j d � | j � d } | j t | � t � | j | d � d S)Nzselect returnblob()r s blob)r5 rD rE rH rI r r )r! rF r. r r r �CheckFuncReturnBlob� s
z!FunctionTests.CheckFuncReturnBlobc C s@ | j j � } | j d � | j � d } | j | d � d S)Nzselect returnlonglong()r r
r l )r5 rD rE rH rI )r! rF r. r r r �CheckFuncReturnLongLong� s
z%FunctionTests.CheckFuncReturnLongLongc C s} | j j � } y( | j d � | j � | j d � Wn? t j k
rx } z | j | j d d � WYd d } ~ Xn Xd S)Nzselect raiseexception()z#should have raised OperationalErrorr z&user-defined function raised exception) r5 rD rE rH r> r3 r? rI �args)r! rF �er r r �CheckFuncException� s
z FunctionTests.CheckFuncExceptionc C sC | j j � } | j d d � | j � d } | j | d � d S)Nzselect isstring(?)r r r
)zfoo)r5 rD rE rH rI )r! rF r. r r r �CheckParamString� s z FunctionTests.CheckParamStringc C sC | j j � } | j d d � | j � d } | j | d � d S)Nzselect isint(?)r r r
)r )r5 rD rE rH rI )r! rF r. r r r �
CheckParamInt� s zFunctionTests.CheckParamIntc C sC | j j � } | j d d � | j � d } | j | d � d S)Nzselect isfloat(?)� ��Q� @r r
)rV )r5 rD rE rH rI )r! rF r. r r r �CheckParamFloat� s z FunctionTests.CheckParamFloatc C sC | j j � } | j d d � | j � d } | j | d � d S)Nzselect isnone(?)r r
)N)r5 rD rE rH rI )r! rF r. r r r �CheckParamNone� s z FunctionTests.CheckParamNonec C sL | j j � } | j d t d � f � | j � d } | j | d � d S)Nzselect isblob(?)s blobr r
)r5 rD rE r rH rI )r! rF r. r r r �CheckParamBlob� s z FunctionTests.CheckParamBlobc C sC | j j � } | j d d � | j � d } | j | d � d S)Nzselect islonglong(?)r
r r � )rZ )r5 rD rE rH rI )r! rF r. r r r �CheckParamLongLong� s z FunctionTests.CheckParamLongLongN)r$ r% r& r7 r9 r@ rG rJ rK rL rM rN rO rP rS rT rU rW rX rY r[ r r r r r1 | s$ r1 c @ s� e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d S) �AggregateTestsc C s� t j d � | _ | j j � } | j d � | j d d d d d t d � f � | j j d d t � | j j d
d t � | j j d d t � | j j d d t
� | j j d
d t � | j j d d t � | j j d d t
� d S)Nz:memory:z�
create table test(
t text,
i integer,
f float,
n,
b blob
)
z6insert into test(t, i, f, n, b) values (?, ?, ?, ?, ?)r r g ��Q� @s blobZnostepr
Z
nofinalizeZexcInitZexcStepZ excFinalizeZ checkTyper< Zmysum)r3 r4 r5 rD rE r Zcreate_aggregater r' r* r+ r, r- r0 )r! rF r r r r7 s zAggregateTests.setUpc C s d S)Nr )r! r r r r9 s zAggregateTests.tearDownc C sC y' | j j d d t � | j d � Wn t j k
r> Yn Xd S)Nr: r; z&should have raised an OperationalErrori����)r5 r6 r0 r> r3 r? )r! r r r �CheckAggrErrorOnCreate! s
z%AggregateTests.CheckAggrErrorOnCreatec C sp | j j � } y | j d � | j d � Wn< t k
rk } z | j | j d d � WYd d } ~ Xn Xd S)Nzselect nostep(t) from testz$should have raised an AttributeErrorr z+'AggrNoStep' object has no attribute 'step')r5 rD rE r> �AttributeErrorrI rQ )r! rF rR r r r �CheckAggrNoStep( s
z AggregateTests.CheckAggrNoStepc C s� | j j � } y. | j d � | j � d } | j d � Wn? t j k
r~ } z | j | j d d � WYd d } ~ Xn Xd S)Nz select nofinalize(t) from testr z&should have raised an OperationalErrorz7user-defined aggregate's 'finalize' method raised error) r5 rD rE rH r> r3 r? rI rQ )r! rF r. rR r r r �CheckAggrNoFinalize0 s
z"AggregateTests.CheckAggrNoFinalizec C s� | j j � } y. | j d � | j � d } | j d � Wn? t j k
r~ } z | j | j d d � WYd d } ~ Xn Xd S)Nzselect excInit(t) from testr z&should have raised an OperationalErrorz7user-defined aggregate's '__init__' method raised error) r5 rD rE rH r> r3 r? rI rQ )r! rF r. rR r r r �CheckAggrExceptionInInit9 s
z'AggregateTests.CheckAggrExceptionInInitc C s� | j j � } y. | j d � | j � d } | j d � Wn? t j k
r~ } z | j | j d d � WYd d } ~ Xn Xd S)Nzselect excStep(t) from testr z&should have raised an OperationalErrorz3user-defined aggregate's 'step' method raised error) r5 rD rE rH r> r3 r? rI rQ )r! rF r. rR r r r �CheckAggrExceptionInStepB s
z'AggregateTests.CheckAggrExceptionInStepc C s� | j j � } y. | j d � | j � d } | j d � Wn? t j k
r~ } z | j | j d d � WYd d } ~ Xn Xd S)Nz select excFinalize(t) from testr z&should have raised an OperationalErrorz7user-defined aggregate's 'finalize' method raised error) r5 rD rE rH r> r3 r? rI rQ )r! rF r. rR r r r � CheckAggrExceptionInFinalizeK s
z+AggregateTests.CheckAggrExceptionInFinalizec C sC | j j � } | j d d � | j � d } | j | d � d S)Nzselect checkType('str', ?)r r r
)zfoo)r5 rD rE rH rI )r! rF r. r r r �CheckAggrCheckParamStrT s z%AggregateTests.CheckAggrCheckParamStrc C sC | j j � } | j d d � | j � d } | j | d � d S)Nzselect checkType('int', ?)r r r
)r )r5 rD rE rH rI )r! rF r. r r r �CheckAggrCheckParamIntZ s z%AggregateTests.CheckAggrCheckParamIntc C sC | j j � } | j d d � | j � d } | j | d � d S)Nz select checkType('float', ?)� ��Q� @r r
)rf )r5 rD rE rH rI )r! rF r. r r r �CheckAggrCheckParamFloat` s z'AggregateTests.CheckAggrCheckParamFloatc C sC | j j � } | j d d � | j � d } | j | d � d S)Nzselect checkType('None', ?)r r
)N)r5 rD rE rH rI )r! rF r. r r r �CheckAggrCheckParamNonef s z&AggregateTests.CheckAggrCheckParamNonec C sL | j j � } | j d t d � f � | j � d } | j | d � d S)Nzselect checkType('blob', ?)s blobr r
)r5 rD rE r rH rI )r! rF r. r r r �CheckAggrCheckParamBlobl s z&AggregateTests.CheckAggrCheckParamBlobc C sf | j j � } | j d � | j d d d
d g � | j d � | j � d } | j | d � d S) Nzdelete from testz insert into test(i) values (?)�
� � zselect mysum(i) from testr �< )rj )rk )rl )r5 rD rE Z executemanyrH rI )r! rF r. r r r �CheckAggrCheckAggrSumr s
z$AggregateTests.CheckAggrCheckAggrSumN)r$ r% r& r7 r9 r] r_ r` ra rb rc rd re rg rh ri rn r r r r r\ s r\ c @ sR e Z d Z e d d � � Z d d � Z d d � Z d d � Z d d
� Z d S) �AuthorizerTestsc C s< | t j k r t j S| d k s. | d k r5 t j St j S)N�c2�t2)r3 �
SQLITE_SELECTZ SQLITE_DENY� SQLITE_OK)�action�arg1�arg2�dbname�sourcer r r �
authorizer_cb{ s
z AuthorizerTests.authorizer_cbc C sI t j d � | _ | j j d � | j j d � | j j | j � d S)Nz:memory: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)r3 r4 r5 Z
executescriptrE Zset_authorizerry )r! r r r r7 � s
zAuthorizerTests.setUpc C s d S)Nr )r! r r r r9 � s zAuthorizerTests.tearDownc C s� y | j j d � Wn] t j k
rs } z: | j d j d � s] | j d | j d � n d SWYd d } ~ Xn X| j d � d S)Nzselect * from t2r �
prohibitedzwrong exception text: %sz9should have raised an exception due to missing privileges)r5 rE r3 �
DatabaseErrorrQ �endswithr> )r! rR r r r �test_table_access� s z!AuthorizerTests.test_table_accessc C s� y | j j d � Wn] t j k
rs } z: | j d j d � s] | j d | j d � n d SWYd d } ~ Xn X| j d � d S)Nzselect c2 from t1r rz zwrong exception text: %sz9should have raised an exception due to missing privileges)r5 rE r3 r{ rQ r| r> )r! rR r r r �test_column_access� s z"AuthorizerTests.test_column_accessN) r$ r% r& � staticmethodry r7 r9 r} r~ r r r r ro z s
ro c @ s" e Z d Z e d d � � Z d S)� AuthorizerRaiseExceptionTestsc C s@ | t j k r t � n | d k s0 | d k r9 t � n t j S)Nrp rq )r3 rr �
ValueErrorrs )rt ru rv rw rx r r r ry � s
z+AuthorizerRaiseExceptionTests.authorizer_cbN)r$ r% r& r ry r r r r r� � s r� c @ s" e Z d Z e d d � � Z d S)�AuthorizerIllegalTypeTestsc C s6 | t j k r d S| d k s+ | d k r/ d St j S)Ng rp rq )r3 rr rs )rt ru rv rw rx r r r ry � s
z(AuthorizerIllegalTypeTests.authorizer_cbN)r$ r% r& r ry r r r r r� � s r� c @ s" e Z d Z e d d � � Z d S)�AuthorizerLargeIntegerTestsc C s6 | t j k r d S| d k s+ | d k r/ d St j S)Nr< � rp rq l l )r3 rr rs )rt ru rv rw rx r r r ry � s
z)AuthorizerLargeIntegerTests.authorizer_cbN)r$ r% r& r ry r r r r r� � s r� c C sm t j t d � } t j t d � } t j t � } t j | | | t j t � t j t � t j t � f � S)NZCheck) �unittestZ makeSuiter1 r\ ro Z TestSuiter� r� r� )Zfunction_suiteZaggregate_suiteZauthorizer_suiter r r �suite� s r� c C s t j � } | j t � � d S)N)r� ZTextTestRunnerZrunr� )Zrunnerr r r �test� s r� �__main__)"r� Zsqlite3r3 r r r r
r r r r r r r r r r r r' r* r+ r, r- r0 ZTestCaser1 r\ ro r� r� r� r� r� r$ r r r r �<module> s@
�v,