��Qc@s�dd
lZdd
l
Z
dd
lZde
jf
d��YZde
jf
d��YZd�Zd�Ze	dkr~e�
nd
S(	i�NtCollationTestsc
BsPeZ
d�Zd
�Zd�Zd�Zd�Zd�Zd�Zd�Z	RS(c


CsdS(
N((
tself((s(/sys/lib/python2.7/sqlite3/test/hooks.pytsetUps
c


CsdS(
N((
R
((s(/sys/lib/python2.7/sqlite3/test/hooks.pyttearDowns
c
Csatj
d
�}
y!|
jdd�
|jd�
Wn*tk
r\
}
|j|jdd�
n
XdS(Ns:memory:t
Xi*sshould have raised a TypeErrorisparameter must be callable(tsqlitetconnecttcreate_collationtfailt	TypeErrortassertEqualtargs(R
tcont
e((s(/sys/lib/python2.7/sqlite3/test/hooks.pytCheckCreateCollationNotCallable"s





c
CsMtj
d
�}
y!|
jdt�
|jd�
Wntjk
rH
}
n
XdS(Ns:memory:scoll�should have raised a ProgrammingError(RRRtcmpRtProgrammingError(R
RR
((s(/sys/lib/python2.7/sqlite3/test/hooks.pytCheckCreateCollationNotAscii*s





c
Cs
tj
dkrdSd�}
tjd�}|jd|
�
d}|j|�j�}|ddd	ks�|ddd
ks�|dddkr�|jd�
n|jdd�
y&|j|�j�}|jd
�
Wn3tjk
r


}
|j	|j
dj�d�
n
XdS(Niii
cSst||
�S(
N(
R(t
xt
y((s(/sys/lib/python2.7/sqlite3/test/hooks.pytmycoll5ss:memory:Rs�
            select x from (
            select 'a' as x
            union
            select 'b' as x
            union
            select 'c' as x
            ) order by x collate mycoll
            it
ct
bt
as#the expected order was not returneds&should have raised an OperationalErrors"no such collation sequence: mycoll(iii
(Rtversion_infoRRtexecutetfetchallRtNonetOperationalErrorR
Rtlower(R
RRtsqltresultR
((s(/sys/lib/python2.7/sqlite3/test/hooks.pytCheckCollationIsUsed2s


	
	

<





c
Csfd
�}
tj
d�}|jd|
�
d}|j|�j�}|j|d
ddgdd	�

dS(
NcSs||
k||
kdS(Nii l((RR((s(/sys/lib/python2.7/sqlite3/test/hooks.pyRPss:memory:Rs�
            select x from (
            select 'a' as x
            union
            select 'b' as x
            union
            select 'c' as x
            ) order by x collate mycoll
            RRRtmsgs#the expected order was not returned(
R(
R(
R(RRRRRR
(R
RRRR((s(/sys/lib/python2.7/sqlite3/test/hooks.pyt!CheckCollationReturnsLargeIntegerOs
	
	


c
Cs�tj
d
�}
|
jdt�
|
jdd��
|
jd�j�}|dddkso|dddkr|jd	�
nd
S(s�
        Register two different collation functions under the same name.
        Verify that the last one is actually used.
        s:memory:RcSst||
�S(
N(
R(RR((s(/sys/lib/python2.7/sqlite3/test/hooks.pyt<lambda>issi
            select x from (select 'a' as x union select 'b' as x) order by x collate mycoll
            iRi
Rs wrong collation function is usedN(RRRRRRR(R
RR((s(/sys/lib/python2.7/sqlite3/test/hooks.pytCheckCollationRegisterTwicebs



(
c
Cs�tj
d
�}
|
jdt�
|
jdd�
y|
jd�
|jd�
Wn<tjk
r�
}
|jdj	d�s�|jd�
q�n
XdS(	su
        Register a collation, then deregister it. Make sure an error is raised if we try
        to use it.
        s:memory:Rs?select 'a' as x union select 'b' as x order by x collate mycolls&should have raised an OperationalErrorisno such collation sequenceswrong OperationalError raisedN(
RRRRRRRRRt
startswith(R
RR
((s(/sys/lib/python2.7/sqlite3/test/hooks.pytCheckDeregisterCollationps








(
t__name__t
__module__RRRRR R"R$R&(((s(/sys/lib/python2.7/sqlite3/test/hooks.pyRs
							t
ProgressTestsc
Bs,eZ
d�Zd
�Zd�Zd�ZRS(c
sRtj
d
�}
g��f
d�}|
j|d�
|
jd�
|j��
dS(sK
        Test that the progress handler is invoked once it is set.
        s:memory:cs�jd�
d
S(Ni(tappendR((
tprogress_calls(s(/sys/lib/python2.7/sqlite3/test/hooks.pytprogress�s


i
s0
            create table foo(a, b)
            N(RRtset_progress_handlerRt
assertTrue(R
RR,((
R+s(/sys/lib/python2.7/sqlite3/test/hooks.pytCheckProgressHandlerUsed�s



c
s�tj
d
�}
g��f
d�}|
j|d�
|
j�}|jd�
t��}g�|
j|d�
|jd�
t��}|j||k�
dS(s=
        Test that the opcode argument is respected.
        s:memory:cs�jd�
d
S(Ni(R*R((
R+(s(/sys/lib/python2.7/sqlite3/test/hooks.pyR,�s


i
s1
            create table foo (a, b)
            is1
            create table bar (a, b)
            N(RRR-tcursorRtlenR.(R
RR,tcurstfirst_counttsecond_count((
R+s(/sys/lib/python2.7/sqlite3/test/hooks.pytCheckOpcodeCount�s









c
s]tj
d
�}
g��f
d�}|
j|d�
|
j�}|jtj|jd�
dS(sW
        Test that returning a non-zero value stops the operation in progress.
        s:memory:cs�jd�
d
S(Ni
(R*R((
R+(s(/sys/lib/python2.7/sqlite3/test/hooks.pyR,�s


i
screate table bar (a, b)N(RRR-R0tassertRaisesRR(R
RR,R2((
R+s(/sys/lib/python2.7/sqlite3/test/hooks.pytCheckCancelOperation�s






c
sttj
d
�}
g��f
d�}|
j|d�
|
jdd�
|
jd�j�
|jt��dd�
dS(sc
        Test that setting the progress handler to None clears the previously set handler.
        s:memory:cs�jd
�
dS(Ni
i(
R*((
taction(s(/sys/lib/python2.7/sqlite3/test/hooks.pyR,�s


i
s&select 1 union select 2 union select 3is progress handler was not clearedN(RRR-RRRR
R1(R
RR,((
R8s(/sys/lib/python2.7/sqlite3/test/hooks.pytCheckClearHandler�s




(R'R(R/R5R7R9(((s(/sys/lib/python2.7/sqlite3/test/hooks.pyR)s
			cCs7tj
td
�}tj
td
�}
tj||
f�S(NtCheck(tunittestt	makeSuiteRR)t	TestSuite(tcollation_suitetprogress_suite((s(/sys/lib/python2.7/sqlite3/test/hooks.pytsuite�s


c
Cs tj
�}|jt��
dS(
N(R;tTextTestRunnertrunR@(
trunner((s(/sys/lib/python2.7/sqlite3/test/hooks.pyttest�s

t__main__(
tosR;tsqlite3RtTestCaseRR)R@RDR'(((s(/sys/lib/python2.7/sqlite3/test/hooks.pyt<module>s
dF