Your IP : 216.73.216.213

Current Path : /opt/alt/python35/lib64/python3.5/sqlite3/test/__pycache__/
Upload File :
Current File : //opt/alt/python35/lib64/python3.5/sqlite3/test/__pycache__/userfunctions.cpython-35.pyc



��Yf;�@s ddlZddlZdd�Zdd�Zdd�Zdd	�Zd
d�Zdd
�Zdd�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd�ZGd d!�d!�ZGd"d#�d#�ZGd$d%�d%�ZGd&d'�d'�ZGd(d)�d)�ZGd*d+�d+�ZGd,d-�d-�ZGd.d/�d/�ZGd0d1�d1ej�ZGd2d3�d3ej�ZGd4d5�d5ej�ZGd6d7�d7e�ZGd8d9�d9e�ZGd:d;�d;e�Z d<d=�Z!d>d?�Z"e#d@kre"�dS)A�NcCsdS)N�foo�rrr�?/opt/alt/python35/lib64/python3.5/sqlite3/test/userfunctions.py�func_returntextsrcCsdS)N�barrrrrr�func_returnunicodesrcCsdS)N�*rrrrr�func_returnint sr	cCsdS)Ng��Q�	@rrrrr�func_returnfloat"sr
cCsdS)Nrrrrr�func_returnnull$srcCsdS)Nsblobrrrrr�func_returnblob&srcCsdS)N��lrrrrr�func_returnlonglong(srcCsdddS)N�rrrrrr�func_raiseexception*srcCst|�tkS)N)�type�str)�vrrr�
func_isstring-srcCst|�tkS)N)r�int)rrrr�
func_isint/srcCst|�tkS)N)r�float)rrrr�func_isfloat1srcCst|�td�kS)N)r)rrrr�func_isnone3srcCst|ttf�S)N)�
isinstance�bytes�
memoryview)rrrr�func_isblob5srcCst|t�o|dkS)Nr
rl)rr)rrrr�func_islonglong7srcGs
t|�S)N)�len)�argsrrr�func:sr"c@s(eZdZdd�Zdd�ZdS)�
AggrNoStepcCsdS)Nr)�selfrrr�__init__>szAggrNoStep.__init__cCsdS)Nr
r)r$rrr�finalizeAszAggrNoStep.finalizeN)�__name__�
__module__�__qualname__r%r&rrrrr#=sr#c@s(eZdZdd�Zdd�ZdS)�AggrNoFinalizecCsdS)Nr)r$rrrr%EszAggrNoFinalize.__init__cCsdS)Nr)r$�xrrr�stepHszAggrNoFinalize.stepN)r'r(r)r%r,rrrrr*Dsr*c@s4eZdZdd�Zdd�Zdd�ZdS)�AggrExceptionInInitcCsdddS)Nrrr)r$rrrr%LszAggrExceptionInInit.__init__cCsdS)Nr)r$r+rrrr,OszAggrExceptionInInit.stepcCsdS)Nr)r$rrrr&RszAggrExceptionInInit.finalizeN)r'r(r)r%r,r&rrrrr-Ksr-c@s4eZdZdd�Zdd�Zdd�ZdS)�AggrExceptionInStepcCsdS)Nr)r$rrrr%VszAggrExceptionInStep.__init__cCsdddS)Nrrr)r$r+rrrr,YszAggrExceptionInStep.stepcCsdS)Nrr)r$rrrr&\szAggrExceptionInStep.finalizeN)r'r(r)r%r,r&rrrrr.Usr.c@s4eZdZdd�Zdd�Zdd�ZdS)�AggrExceptionInFinalizecCsdS)Nr)r$rrrr%`sz AggrExceptionInFinalize.__init__cCsdS)Nr)r$r+rrrr,cszAggrExceptionInFinalize.stepcCsdddS)Nrrr)r$rrrr&fsz AggrExceptionInFinalize.finalizeN)r'r(r)r%r,r&rrrrr/_sr/c@s4eZdZdd�Zdd�Zdd�ZdS)�
AggrCheckTypecCs
d|_dS)N)�val)r$rrrr%jszAggrCheckType.__init__c
CsMdtdtdtdtd�dti}t||t|�k�|_dS)Nrrr�None�blob)rrrrrr1)r$�	whichTyper1�theTyperrrr,mszAggrCheckType.stepcCs|jS)N)r1)r$rrrr&rszAggrCheckType.finalizeN)r'r(r)r%r,r&rrrrr0isr0c@s4eZdZdd�Zdd�Zdd�ZdS)�AggrCheckTypescCs
d|_dS)Nr)r1)r$rrrr%vszAggrCheckTypes.__init__c
Gsddtdtdtdtd�dti}x3|D]+}|jt||t|�k�7_q1WdS)Nrrrr2r3)rrrrrr1)r$r4�valsr5r1rrrr,ys
zAggrCheckTypes.stepcCs|jS)N)r1)r$rrrr&szAggrCheckTypes.finalizeN)r'r(r)r%r,r&rrrrr6usr6c@s4eZdZdd�Zdd�Zdd�ZdS)�AggrSumcCs
d|_dS)Ng)r1)r$rrrr%�szAggrSum.__init__cCs|j|7_dS)N)r1)r$r1rrrr,�szAggrSum.stepcCs|jS)N)r1)r$rrrr&�szAggrSum.finalizeN)r'r(r)r%r,r&rrrrr8�sr8c@s�eZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd�Zdd �Zd!d"�Zd#d$�Zd%d&�Zd'S)(�
FunctionTestscCs`tjd�|_|jjddt�|jjddt�|jjddt�|jjddt�|jjddt�|jjddt	�|jjd	dt
�|jjd
dt�|jjddt�|jjd
dt
�|jjddt�|jjddt�|jjddt�|jjddt�|jjddt�dS)Nz:memory:Z
returntextrZ
returnunicodeZ	returnintZreturnfloatZ
returnnullZ
returnblobZreturnlonglongZraiseexception�isstringr
ZisintZisfloatZisnoneZisblobZ
islonglongZspam���)�sqlite�connect�con�create_functionrrr	r
rrrrrrrrrrr")r$rrr�setUp�s zFunctionTests.setUpcCs|jj�dS)N)r>�close)r$rrr�tearDown�szFunctionTests.tearDowncCs:|jtj��!|jjdddd��WdQRXdS)N�bla�dcSsd|S)N�r)r+rrr�<lambda>�sz6FunctionTests.CheckFuncErrorOnCreate.<locals>.<lambda>i����)�assertRaisesr<�OperationalErrorr>r?)r$rrr�CheckFuncErrorOnCreate�sz$FunctionTests.CheckFuncErrorOnCreatecCsXdd�}|�}|t�d<|jjdd|�|jj�}|jd�dS)NcSsdd�}|S)NcSsdS)Nr
rrrrr�f�sz;FunctionTests.CheckFuncRefCount.<locals>.getfunc.<locals>.fr)rJrrr�getfunc�sz0FunctionTests.CheckFuncRefCount.<locals>.getfuncrZreftestrzselect reftest())�globalsr>r?�cursor�execute)r$rKrJ�currrr�CheckFuncRefCount�s	
zFunctionTests.CheckFuncRefCountcCsV|jj�}|jd�|j�d}|jt|�t�|j|d�dS)Nzselect returntext()rr)r>rMrN�fetchone�assertEqualrr)r$rOr1rrr�CheckFuncReturnText�s

z!FunctionTests.CheckFuncReturnTextcCsV|jj�}|jd�|j�d}|jt|�t�|j|d�dS)Nzselect returnunicode()rr)r>rMrNrQrRrr)r$rOr1rrr�CheckFuncReturnUnicode�s

z$FunctionTests.CheckFuncReturnUnicodecCsV|jj�}|jd�|j�d}|jt|�t�|j|d�dS)Nzselect returnint()rr)r>rMrNrQrRrr)r$rOr1rrr�CheckFuncReturnInt�s

z FunctionTests.CheckFuncReturnIntcCsk|jj�}|jd�|j�d}|jt|�t�|dksZ|dkrg|jd�dS)Nzselect returnfloat()rg�&1�	@gT㥛� 	@zwrong value)r>rMrNrQrRrrZfail)r$rOr1rrr�CheckFuncReturnFloat�s
z"FunctionTests.CheckFuncReturnFloatcCs\|jj�}|jd�|j�d}|jt|�td��|j|d�dS)Nzselect returnnull()r)r>rMrNrQrRr)r$rOr1rrr�CheckFuncReturnNull�s

z!FunctionTests.CheckFuncReturnNullcCsV|jj�}|jd�|j�d}|jt|�t�|j|d�dS)Nzselect returnblob()rsblob)r>rMrNrQrRrr)r$rOr1rrr�CheckFuncReturnBlob�s

z!FunctionTests.CheckFuncReturnBlobcCs@|jj�}|jd�|j�d}|j|d�dS)Nzselect returnlonglong()rr
rl)r>rMrNrQrR)r$rOr1rrr�CheckFuncReturnLongLong�s
z%FunctionTests.CheckFuncReturnLongLongc
Cs_|jj�}|jtj��}|jd�|j�WdQRX|jt|j	�d�dS)Nzselect raiseexception()z&user-defined function raised exception)
r>rMrGr<rHrNrQrRr�	exception)r$rO�cmrrr�CheckFuncException�s

z FunctionTests.CheckFuncExceptioncCsC|jj�}|jdd�|j�d}|j|d�dS)Nzselect isstring(?)rrr
)r)r>rMrNrQrR)r$rOr1rrr�CheckParamString�szFunctionTests.CheckParamStringcCsC|jj�}|jdd�|j�d}|j|d�dS)Nzselect isint(?)rrr
)r)r>rMrNrQrR)r$rOr1rrr�
CheckParamInt�szFunctionTests.CheckParamIntcCsC|jj�}|jdd�|j�d}|j|d�dS)Nzselect isfloat(?)���Q�	@rr
)r_)r>rMrNrQrR)r$rOr1rrr�CheckParamFloat�szFunctionTests.CheckParamFloatcCsC|jj�}|jdd�|j�d}|j|d�dS)Nzselect isnone(?)rr
)N)r>rMrNrQrR)r$rOr1rrr�CheckParamNone�szFunctionTests.CheckParamNonecCsL|jj�}|jdtd�f�|j�d}|j|d�dS)Nzselect isblob(?)sblobrr
)r>rMrNrrQrR)r$rOr1rrr�CheckParamBlobszFunctionTests.CheckParamBlobcCsC|jj�}|jdd�|j�d}|j|d�dS)Nzselect islonglong(?)r
rr�)rc)r>rMrNrQrR)r$rOr1rrr�CheckParamLongLong
sz FunctionTests.CheckParamLongLongcCsC|jj�}|jdd�|j�d}|j|d�dS)Nzselect spam(?, ?)r
rEr)r
rE)r>rMrNrQrR)r$rOr1rrr�CheckAnyArgumentsszFunctionTests.CheckAnyArgumentsN)r'r(r)r@rBrIrPrSrTrUrVrWrXrYr\r]r^r`rarbrdrerrrrr9�s&r9c@s�eZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd�ZdS) �AggregateTestscCstjd�|_|jj�}|jd�|jdddddtd�f�|jjdd	t�|jjd
d	t�|jjdd	t	�|jjdd	t
�|jjd
d	t�|jjddt�|jjddt
�|jjdd	t�dS)Nz:memory:z�
            create table test(
                t text,
                i integer,
                f float,
                n,
                b blob
                )
            z6insert into test(t, i, f, n, b) values (?, ?, ?, ?, ?)rrg��Q�	@sblobZnostepr
Z
nofinalizeZexcInitZexcStepZexcFinalizeZ	checkTyperEZ
checkTypesZmysumr;)r<r=r>rMrNrZcreate_aggregater#r*r-r.r/r0r6r8)r$rOrrrr@s	zAggregateTests.setUpcCsdS)Nr)r$rrrrB0szAggregateTests.tearDowncCs4|jtj��|jjddt�WdQRXdS)NrCrDi����)rGr<rHr>r?r8)r$rrr�CheckAggrErrorOnCreate5sz%AggregateTests.CheckAggrErrorOnCreatec
CsR|jj�}|jt��}|jd�WdQRX|jt|j�d�dS)Nzselect nostep(t) from testz+'AggrNoStep' object has no attribute 'step')r>rMrG�AttributeErrorrNrRrrZ)r$rOr[rrr�CheckAggrNoStep9szAggregateTests.CheckAggrNoStepc
Cse|jj�}|jtj��$}|jd�|j�d}WdQRX|jt|j	�d�dS)Nzselect nofinalize(t) from testrz7user-defined aggregate's 'finalize' method raised error)
r>rMrGr<rHrNrQrRrrZ)r$rOr[r1rrr�CheckAggrNoFinalize?s

z"AggregateTests.CheckAggrNoFinalizec
Cse|jj�}|jtj��$}|jd�|j�d}WdQRX|jt|j	�d�dS)Nzselect excInit(t) from testrz7user-defined aggregate's '__init__' method raised error)
r>rMrGr<rHrNrQrRrrZ)r$rOr[r1rrr�CheckAggrExceptionInInitFs

z'AggregateTests.CheckAggrExceptionInInitc
Cse|jj�}|jtj��$}|jd�|j�d}WdQRX|jt|j	�d�dS)Nzselect excStep(t) from testrz3user-defined aggregate's 'step' method raised error)
r>rMrGr<rHrNrQrRrrZ)r$rOr[r1rrr�CheckAggrExceptionInStepMs

z'AggregateTests.CheckAggrExceptionInStepc
Cse|jj�}|jtj��$}|jd�|j�d}WdQRX|jt|j	�d�dS)Nzselect excFinalize(t) from testrz7user-defined aggregate's 'finalize' method raised error)
r>rMrGr<rHrNrQrRrrZ)r$rOr[r1rrr�CheckAggrExceptionInFinalizeTs

z+AggregateTests.CheckAggrExceptionInFinalizecCsC|jj�}|jdd�|j�d}|j|d�dS)Nzselect checkType('str', ?)rrr
)r)r>rMrNrQrR)r$rOr1rrr�CheckAggrCheckParamStr[sz%AggregateTests.CheckAggrCheckParamStrcCsC|jj�}|jdd�|j�d}|j|d�dS)Nzselect checkType('int', ?)rrr
)r)r>rMrNrQrR)r$rOr1rrr�CheckAggrCheckParamIntasz%AggregateTests.CheckAggrCheckParamIntcCsC|jj�}|jdd�|j�d}|j|d�dS)Nzselect checkTypes('int', ?, ?)r�rrE)rrp)r>rMrNrQrR)r$rOr1rrr�CheckAggrCheckParamsIntgsz&AggregateTests.CheckAggrCheckParamsIntcCsC|jj�}|jdd�|j�d}|j|d�dS)Nzselect checkType('float', ?)���Q�	@rr
)rr)r>rMrNrQrR)r$rOr1rrr�CheckAggrCheckParamFloatmsz'AggregateTests.CheckAggrCheckParamFloatcCsC|jj�}|jdd�|j�d}|j|d�dS)Nzselect checkType('None', ?)rr
)N)r>rMrNrQrR)r$rOr1rrr�CheckAggrCheckParamNonessz&AggregateTests.CheckAggrCheckParamNonecCsL|jj�}|jdtd�f�|j�d}|j|d�dS)Nzselect checkType('blob', ?)sblobrr
)r>rMrNrrQrR)r$rOr1rrr�CheckAggrCheckParamBlobysz&AggregateTests.CheckAggrCheckParamBlobcCsf|jj�}|jd�|jdd	d
dg�|jd�|j�d}|j|d�dS)Nzdelete from testzinsert into test(i) values (?)�
��zselect mysum(i) from testr�<)rv)rw)rx)r>rMrNZexecutemanyrQrR)r$rOr1rrr�CheckAggrCheckAggrSums

z$AggregateTests.CheckAggrCheckAggrSumN)r'r(r)r@rBrgrirjrkrlrmrnrorqrsrtrurzrrrrrfsrfc@sReZdZedd��Zdd�Zdd�Zdd�Zd	d
�ZdS)�AuthorizerTestscCs<|tjkrtjS|dks.|dkr5tjStjS)N�c2�t2)r<�
SQLITE_SELECTZSQLITE_DENY�	SQLITE_OK)�action�arg1�arg2�dbname�sourcerrr�
authorizer_cb�s
zAuthorizerTests.authorizer_cbcCsItjd�|_|jjd�|jjd�|jj|j�dS)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)r<r=r>Z
executescriptrNZset_authorizerr�)r$rrrr@�s
	zAuthorizerTests.setUpcCsdS)Nr)r$rrrrB�szAuthorizerTests.tearDowncCsI|jtj��}|jjd�WdQRX|jdt|j��dS)Nzselect * from t2�
prohibited)rGr<�
DatabaseErrorr>rN�assertInrrZ)r$r[rrr�test_table_access�sz!AuthorizerTests.test_table_accesscCsI|jtj��}|jjd�WdQRX|jdt|j��dS)Nzselect c2 from t1r�)rGr<r�r>rNr�rrZ)r$r[rrr�test_column_access�sz"AuthorizerTests.test_column_accessN)	r'r(r)�staticmethodr�r@rBr�r�rrrrr{�s
r{c@s"eZdZedd��ZdS)�AuthorizerRaiseExceptionTestscCs:|tjkrt�|dks-|dkr3t�tjS)Nr|r})r<r~�
ValueErrorr)r�r�r�r�r�rrrr��s
z+AuthorizerRaiseExceptionTests.authorizer_cbN)r'r(r)r�r�rrrrr��sr�c@s"eZdZedd��ZdS)�AuthorizerIllegalTypeTestscCs6|tjkrdS|dks+|dkr/dStjS)Ngr|r})r<r~r)r�r�r�r�r�rrrr��s
z(AuthorizerIllegalTypeTests.authorizer_cbN)r'r(r)r�r�rrrrr��sr�c@s"eZdZedd��ZdS)�AuthorizerLargeIntegerTestscCs6|tjkrdS|dks+|dkr/dStjS)NrE� r|r}ll)r<r~r)r�r�r�r�r�rrrr��s
z)AuthorizerLargeIntegerTests.authorizer_cbN)r'r(r)r�r�rrrrr��sr�cCsmtjtd�}tjtd�}tjt�}tj|||tjt�tjt�tjt�f�S)NZCheck)	�unittestZ	makeSuiter9rfr{Z	TestSuiter�r�r�)Zfunction_suiteZaggregate_suiteZauthorizer_suiterrr�suite�sr�cCs tj�}|jt��dS)N)r�ZTextTestRunnerZrunr�)Zrunnerrrr�test�sr��__main__)$r�Zsqlite3r<rrr	r
rrrrrrrrrrr"r#r*r-r.r/r0r6r8ZTestCaser9rfr{r�r�r�r�r�r'rrrr�<module>sD




�p$