B 18™\2)ã@sxddlZddlZddlmZmZGdd„dejƒZGdd„dejƒZGdd„dejƒZ d d „Z d d „Z e d krte ƒdS)éN)ÚTESTFNÚunlinkc@s^eZdZdd„Zdd„Zdd„Zdd„Ze e j d kd ¡d d „ƒZ d d„Z dd„Z dd„ZdS)ÚCollationTestsc Cs4t d¡}| t¡| ddd„¡WdQRXdS)Nz:memory:cSs||k||kS)N©)ÚxÚyrrúV/oak/stanford/groups/akundaje/marinovg/programs/Python-3.7.3/Lib/sqlite3/test/hooks.pyÚ!óz>CollationTests.CheckCreateCollationNotString..)ÚsqliteÚconnectÚ assertRaisesÚ TypeErrorÚcreate_collation)ÚselfÚconrrrÚCheckCreateCollationNotStrings  z,CollationTests.CheckCreateCollationNotStringc CsBt d¡}| t¡}| dd¡WdQRX| t|jƒd¡dS)Nz:memory:ÚXé*zparameter must be callable)r r r rrÚ assertEqualÚstrÚ exception)rrÚcmrrrÚCheckCreateCollationNotCallable#s  z.CollationTests.CheckCreateCollationNotCallablec Cs6t d¡}| tj¡| ddd„¡WdQRXdS)Nz:memory:ucolläcSs||k||kS)Nr)rrrrrr ,r z=CollationTests.CheckCreateCollationNotAscii..)r r r ÚProgrammingErrorr)rrrrrÚCheckCreateCollationNotAscii)s z+CollationTests.CheckCreateCollationNotAsciicCslGdd„dtƒ}t d¡}dd„}| |dƒ|¡| d¡ ¡}| |ddd ¡| |d dd ¡dS) Nc@seZdZdd„ZdS)z@CollationTests.CheckCreateCollationBadUpper..BadUpperStrcSsdS)Nr)rrrrÚupper0szFCollationTests.CheckCreateCollationBadUpper..BadUpperStr.upperN)Ú__name__Ú __module__Ú __qualname__rrrrrÚ BadUpperStr/sr z:memory:cSs||k||k S)Nr)rrrrrr 3r z=CollationTests.CheckCreateCollationBadUpper..Úmycollz› select x from ( select 'a' as x union select 'b' as x ) order by x collate mycoll rÚbéÚa)rr r rÚexecuteÚfetchallr)rr rr!ÚresultrrrÚCheckCreateCollationBadUpper.s  z+CollationTests.CheckCreateCollationBadUpper)éér#z&old SQLite versions crash on this testc CsŽdd„}t d¡}| d|¡d}| |¡ ¡}|j|dddgd d | dd¡| tj¡}| |¡ ¡}WdQRX| t|j ƒd ¡dS) NcSs||k||k S)Nr)rrrrrr!Bsz3CollationTests.CheckCollationIsUsed..mycollz:memory:r!zÉ select x from ( select 'a' as x union select 'b' as x union select 'c' as x ) order by x collate mycoll )Úc)r")r$z#the expected order was not returned)Úmsgz"no such collation sequence: mycoll) r r rr%r&rr ÚOperationalErrorrr)rr!rÚsqlr'rrrrÚCheckCollationIsUsed?s   z#CollationTests.CheckCollationIsUsedcCsJdd„}t d¡}| d|¡d}| |¡ ¡}|j|dddgd d dS) NcSs||k||k dS)Nlr)rrrrrr![sz@CollationTests.CheckCollationReturnsLargeInteger..mycollz:memory:r!zÉ select x from ( select 'a' as x union select 'b' as x union select 'c' as x ) order by x collate mycoll )r+)r")r$z#the expected order was not returned)r,)r r rr%r&r)rr!rr.r'rrrÚ!CheckCollationReturnsLargeIntegerZs  z0CollationTests.CheckCollationReturnsLargeIntegercCsdt d¡}| ddd„¡| ddd„¡| d¡ ¡}| |ddd¡| |d dd ¡d S) z„ Register two different collation functions under the same name. Verify that the last one is actually used. z:memory:r!cSs||k||kS)Nr)rrrrrr sr z.cSs||k||k S)Nr)rrrrrr tr zi select x from (select 'a' as x union select 'b' as x) order by x collate mycoll rr"r#r$N)r r rr%r&r)rrr'rrrÚCheckCollationRegisterTwicems  z*CollationTests.CheckCollationRegisterTwicec Cs^t d¡}| ddd„¡| dd¡| tj¡}| d¡WdQRX| t|jƒd¡dS)zu Register a collation, then deregister it. Make sure an error is raised if we try to use it. z:memory:r!cSs||k||kS)Nr)rrrrrr r z9CollationTests.CheckDeregisterCollation..Nz?select 'a' as x union select 'b' as x order by x collate mycollz"no such collation sequence: mycoll) r r rr r-r%rrr)rrrrrrÚCheckDeregisterCollation{s   z'CollationTests.CheckDeregisterCollationN)rrrrrrr(ÚunittestÚskipIfr Úsqlite_version_infor/r0r1r2rrrrrs rc@s,eZdZdd„Zdd„Zdd„Zdd„Zd S) Ú ProgressTestscs>t d¡}g‰‡fdd„}| |d¡| d¡| ˆ¡dS)zK Test that the progress handler is invoked once it is set. z:memory:csˆ d¡dS)Nr)Úappendr)Úprogress_callsrrÚprogressŽs z8ProgressTests.CheckProgressHandlerUsed..progressr#z0 create table foo(a, b) N)r r Úset_progress_handlerr%Ú assertTrue)rrr9r)r8rÚCheckProgressHandlerUsedˆs   z&ProgressTests.CheckProgressHandlerUsedcsrt d¡}g‰‡fdd„}| |d¡| ¡}| d¡tˆƒ}g‰| |d¡| d¡tˆƒ}| ||¡dS) z= Test that the opcode argument is respected. z:memory:csˆ d¡dS)Nr)r7r)r8rrr9žs z0ProgressTests.CheckOpcodeCount..progressr#z1 create table foo (a, b) r*z1 create table bar (a, b) N)r r r:Úcursorr%ÚlenÚassertGreaterEqual)rrr9ÚcursZ first_countZ second_countr)r8rÚCheckOpcodeCount˜s    zProgressTests.CheckOpcodeCountcCs<t d¡}dd„}| |d¡| ¡}| tj|jd¡dS)zW Test that returning a non-zero value stops the operation in progress. z:memory:cSsdS)Nr#rrrrrr9´sz4ProgressTests.CheckCancelOperation..progressr#zcreate table bar (a, b)N)r r r:r=r r-r%)rrr9r@rrrÚCheckCancelOperation¯s  z"ProgressTests.CheckCancelOperationcsRt d¡}d‰‡fdd„}| |d¡| dd¡| d¡ ¡| ˆdd¡dS) zc Test that setting the progress handler to None clears the previously set handler. z:memory:rcsd‰dS)Nr#rrr)Úactionrrr9Ãsz1ProgressTests.CheckClearHandler..progressr#Nz&select 1 union select 2 union select 3z progress handler was not cleared)r r r:r%r&r)rrr9r)rCrÚCheckClearHandler½s    zProgressTests.CheckClearHandlerN)rrrr<rArBrDrrrrr6‡sr6c@s>eZdZdd„Zdd„Zdd„Ze ej dkd¡d d „ƒZ d S) ÚTraceCallbackTestscsTt d¡}g‰‡fdd„}| |¡| d¡| ˆ¡| tdd„ˆDƒƒ¡dS)zI Test that the trace callback is invoked once it is set. z:memory:csˆ |¡dS)N)r7)Ú statement)Útraced_statementsrrÚtraceÓsz8TraceCallbackTests.CheckTraceCallbackUsed..tracezcreate table foo(a, b)css|]}d|kVqdS)zcreate table fooNr)Ú.0Ústmtrrrú Øsz.N)r r Úset_trace_callbackr%r;Úany)rrrHr)rGrÚCheckTraceCallbackUsedÍs     z)TraceCallbackTests.CheckTraceCallbackUsedcsHt d¡}g‰‡fdd„}| |¡| d¡| d¡| ˆd¡dS)zb Test that setting the trace callback to None clears the previously set callback. z:memory:csˆ |¡dS)N)r7)rF)rGrrrHàsz9TraceCallbackTests.CheckClearTraceCallback..traceNzcreate table foo(a, b)ztrace callback was not cleared)r r rLr%Ú assertFalse)rrrHr)rGrÚCheckClearTraceCallbackÚs     z*TraceCallbackTests.CheckClearTraceCallbackc s‚d‰t d¡}g‰‡fdd„}| |¡| d¡| dˆ¡| ¡| t‡fdd„ˆDƒƒd tˆƒd  t tˆƒ¡f¡d S) zG Test that the statement can contain unicode literals. uöäüÖÄÜ߀z:memory:csˆ |¡dS)N)r7)rF)rGrrrHîsz5TraceCallbackTests.CheckUnicodeContent..tracezcreate table foo(x)z insert into foo(x) values ("%s")c3s|]}ˆ|kVqdS)Nr)rIrJ)Ú unicode_valuerrrK÷sz9TraceCallbackTests.CheckUnicodeContent..z-Unicode data %s garbled in trace callback: %sz, N) r r rLr%Úcommitr;rMÚasciiÚjoinÚmap)rrrHr)rGrQrÚCheckUnicodeContentçs    z&TraceCallbackTests.CheckUnicodeContent)r)r)é z#sqlite3_prepare_v2 is not availablecs„g‰‡fdd„}ddg}| tt¡tjtdd}t t¡}| |¡| ¡}| |d¡| d¡| |d¡| ˆ|¡dS) Ncsˆ |¡dS)N)r7)rF)rGrrrHÿsz;TraceCallbackTests.CheckTraceCallbackContent..tracezcreate table foo(x)zinsert into foo(x) values(1))Úisolation_levelrzcreate table bar(x)r#) Ú addCleanuprrr r rLr=r%r)rrHZqueriesÚcon1Úcon2Úcurr)rGrÚCheckTraceCallbackContentûs     z,TraceCallbackTests.CheckTraceCallbackContentN) rrrrNrPrVr3r4r r5r]rrrrrEÌs  rEcCs4t td¡}t td¡}t td¡}t |||f¡S)NÚCheck)r3Ú makeSuiterr6rEÚ TestSuite)Zcollation_suiteZprogress_suiteZ trace_suiterrrÚsuites   racCst ¡}| tƒ¡dS)N)r3ÚTextTestRunnerÚrunra)ÚrunnerrrrÚtestsreÚ__main__) r3Úsqlite3r Ú test.supportrrÚTestCaserr6rErarerrrrrÚsjEC