B 18™\š@ã@s€ddlZddlZddlZddlZddlmZGdd„dejƒZGdd„dƒZ Gdd„dejƒZ d d „Z d d „Ze d kr|eƒdS)éN)Úsupportc@seZdZdd„Zdd„Zdd„Zdd„Zd d „Zd d „Zd d„Z e   e j dkd¡dd„ƒZdd„Zdd„Zdd„Zdd„Zdd„Zdd„Zdd „Zd!d"„Zd#d$„Zd%d&„Zd'd(„Zd)d*„Zd+d,„Zd-d.„Zd/d0„Zd1d2„Zd3d4„Zd5d6„Z d7d8„Z!d9d:„Z"d;d<„Z#d=S)>ÚRegressionTestscCst d¡|_dS)Nz:memory:)ÚsqliteÚconnectÚcon)Úself©rú[/oak/stanford/groups/akundaje/marinovg/programs/Python-3.7.3/Lib/sqlite3/test/regression.pyÚsetUpszRegressionTests.setUpcCs|j ¡dS)N)rÚclose)rrrr ÚtearDown"szRegressionTests.tearDowncCs|j ¡}| d¡dS)Nzpragma user_version)rÚcursorÚexecute)rÚcurrrr ÚCheckPragmaUserVersion%s z&RegressionTests.CheckPragmaUserVersioncCs@tjdtjd}z|j ¡}| d¡Wd| ¡| ¡XdS)Nz:memory:)Ú detect_typeszpragma schema_version)rrÚPARSE_COLNAMESrr rr )rrrrrr ÚCheckPragmaSchemaVersion*s  z(RegressionTests.CheckPragmaSchemaVersioncs–tjddd‰‡fdd„tdƒDƒ}|d d¡x.tdƒD]"}|d d d d„tdƒDƒ¡q9sz7RegressionTests.CheckStatementReset..rzcreate table test(x)é zinsert into test(x) values (?)cSsg|] }|f‘qSrr)rrrrr r<sú zselect x from test)rrÚrangerÚ executemanyÚrollback)rÚcursorsÚir)rr ÚCheckStatementReset4s"z#RegressionTests.CheckStatementResetcCsN|j ¡}| d¡| |jddd¡| d¡| |jddd¡dS)Nz select 1 as "foo bar [datetime]"rzfoo barzselect 1 as "foo baz"zfoo baz)rr rÚ assertEqualÚ description)rrrrr ÚCheckColumnNameWithSpacesCs    z)RegressionTests.CheckColumnNameWithSpacescCsPt d¡}g}x4tdƒD](}| ¡}| |¡| dt|ƒ¡qW| ¡dS)Nz:memory:éizselect 1 x union select )rrrr ÚappendrÚstrr )rrrrrrrr Ú#CheckStatementFinalizationOnCloseDbKs  z3RegressionTests.CheckStatementFinalizationOnCloseDb)éér(zneeds sqlite 3.2.2 or newercCs€t d¡}| d¡| d¡y| d¡Wntjk rBYnX| d¡y | ¡Wn tjk rz| d¡YnXdS)Nz:memory:z3create table foo(x, unique(x) on conflict rollback)zinsert into foo(x) values (1)zinsert into foo(x) values (2)z1pysqlite knew nothing about the implicit ROLLBACK)rrrÚ DatabaseErrorÚcommitÚOperationalErrorÚfail)rrrrr ÚCheckOnConflictRollbackYs     z'RegressionTests.CheckOnConflictRollbackcCs(|j d¡|j d¡|j d¡dS)zm pysqlite would crash with older SQLite versions unless a workaround is implemented. zcreate table foo(bar)zdrop table fooN)rr)rrrr Ú-CheckWorkaroundForBuggySqliteTransferBindingshs  z=RegressionTests.CheckWorkaroundForBuggySqliteTransferBindingscCs|j d¡dS)z pysqlite used to segfault with SQLite versions 3.5.x. These return NULL for "no-operation" statements ÚN)rr)rrrr ÚCheckEmptyStatementqsz#RegressionTests.CheckEmptyStatementcCshd}tjdtjd}| d¡| dtj ¡f¡| |¡| d¡| d¡| d¡| |¡d S) z pysqlite until 2.4.1 did not rebuild the row_cast_map when recompiling a statement. This test exhibits the problem. zselect * from fooz:memory:)rzcreate table foo(bar timestamp)zinsert into foo(bar) values (?)zdrop table foozcreate table foo(bar integer)zinsert into foo(bar) values (5)N)rrÚPARSE_DECLTYPESrÚdatetimeÚnow)rZSELECTrrrr ÚCheckTypeMapUsagexs     z!RegressionTests.CheckTypeMapUsagec CsP| tj¡$}|j dttdgƒƒf¡ ¡WdQRXd}| |t |j ƒ¡dS)Nz"select 'xxx' || ? || 'yyy' colnameéúz9Could not decode to UTF-8 column 'colname' with text 'xxx) Ú assertRaisesrr+rrÚbytesÚ bytearrayÚfetchoneÚassertInr%Ú exception)rÚcmÚmsgrrr ÚCheckErrorMsgDecodeError‡s  z(RegressionTests.CheckErrorMsgDecodeErrorcCs| ttjid¡dS)z! See issue 3312. N)r6Ú TypeErrorrÚregister_adapter)rrrr ÚCheckRegisterAdaptersz$RegressionTests.CheckRegisterAdapterc sôG‡fdd„dtƒ}t d¡‰dˆ_xHdD]@}|j|d*|ˆ_| ¡ˆ_| ¡ˆ_||ƒˆ_WdQRXq*Wdˆ_dˆ_dtfdtfd tfd tfd tfg}xR|D]J\}}|j|d0|  |¡ |ˆ_WdQRX|  ˆjd¡WdQRXq¢WdS) Ncs eZdZdd„Z‡fdd„ZdS)z9RegressionTests.CheckSetIsolationLevel..CustomStrcSsdS)Nr)rrrr Úupper™sz?RegressionTests.CheckSetIsolationLevel..CustomStr.uppercs dˆ_dS)Nr/)Úisolation_level)r)rrr Ú__del__›szARegressionTests.CheckSetIsolationLevel..CustomStr.__del__N)Ú__name__Ú __module__Ú __qualname__rBrDr)rrr Ú CustomStr˜srHz:memory:)r/ÚDEFERREDZ IMMEDIATEZ EXCLUSIVE)ÚlevelrIéóÚabczIMMEDIATEEXCLUSIVEõé) r%rrrCÚsubTestÚlowerÚ capitalizer?Ú ValueErrorr6r )rrHrJÚpairsÚvalueÚexcr)rr ÚCheckSetIsolationLevel–s$     z&RegressionTests.CheckSetIsolationLevelc CspGdd„dtjƒ}t d¡}||ƒ}| tj¡| d¡ ¡WdQRX| tjd¡| ¡WdQRXdS)zd Verifies that cursor methods check whether base class __init__ was called. c@seZdZdd„ZdS)z?RegressionTests.CheckCursorConstructorCallCheck..CursorcSsdS)Nr)rrrrr Ú__init__ºszHRegressionTests.CheckCursorConstructorCallCheck..Cursor.__init__N)rErFrGrWrrrr ÚCursor¹srXz:memory:z select 4+5Nz$^Base Cursor\.__init__ not called\.$) rrXrr6ÚProgrammingErrorrÚfetchallÚassertRaisesRegexr )rrXrrrrr ÚCheckCursorConstructorCallCheck´s z/RegressionTests.CheckCursorConstructorCallCheckcCs(Gdd„dtƒ}|j d|dƒf¡dS)za The Python 3.0 port of the module didn't cope with values of subclasses of str. c@s eZdZdS)z/RegressionTests.CheckStrSubclass..MyStrN)rErFrGrrrr ÚMyStrÉsr]zselect ?rMN)r%rr)rr]rrr ÚCheckStrSubclassÅsz RegressionTests.CheckStrSubclassc Cs>Gdd„dtjƒ}|dƒ}| tj¡| ¡}WdQRXdS)zh Verifies that connection methods check whether base class __init__ was called. c@seZdZdd„ZdS)zGRegressionTests.CheckConnectionConstructorCallCheck..ConnectioncSsdS)Nr)rÚnamerrr rWÒszPRegressionTests.CheckConnectionConstructorCallCheck..Connection.__init__N)rErFrGrWrrrr Ú ConnectionÑsr`z:memory:N)rr`r6rYr )rr`rrrrr Ú#CheckConnectionConstructorCallCheckÌsz3RegressionTests.CheckConnectionConstructorCallCheckc sŠG‡fdd„dtjƒ}Gdd„dtjƒ‰|dƒ}| ¡}| d¡| ddd d g¡| d ¡| ¡| tj¡|  ¡Wd QRXd S) zœ Verifies that subclassed cursor classes are correctly registered with the connection object, too. (fetch-across-rollback problem) cseZdZ‡fdd„ZdS)z;RegressionTests.CheckCursorRegistration..Connectioncsˆ|ƒS)Nr)r)rXrr r ßszBRegressionTests.CheckCursorRegistration..Connection.cursorN)rErFrGr r)rXrr r`Þsr`c@seZdZdd„ZdS)z7RegressionTests.CheckCursorRegistration..CursorcSstj ||¡dS)N)rrXrW)rrrrr rWãsz@RegressionTests.CheckCursorRegistration..Cursor.__init__N)rErFrGrWrrrr rXâsrXz:memory:zcreate table foo(x)zinsert into foo(x) values (?))r')é)rzselect x from fooN) rr`rXr rrrr6ÚInterfaceErrorrZ)rr`rrr)rXr ÚCheckCursorRegistrationÙs  z'RegressionTests.CheckCursorRegistrationcCstjddd}dS)z§ Verifies that creating a connection in autocommit mode works. 2.5.3 introduced a regression so that these could no longer be created. z:memory:N)rC)rr)rrrrr ÚCheckAutoCommitïszRegressionTests.CheckAutoCommitcCs4|j ¡}| d¡| d¡| d¡| ¡}dS)z‡ Verifies that running a PRAGMA statement that does an autocommit does work. This did not work in 2.5.3/2.5.4. zcreate table foo(bar)zinsert into foo(bar) values (5)zpragma page_sizeN)rr rr9)rrÚrowrrr ÚCheckPragmaAutocommit÷s     z%RegressionTests.CheckPragmaAutocommitcCs| tj|jd¡dS)z} Call a connection with a non-string SQL request: check error handling of the statement constructor. rKN)r6rÚWarningr)rrrr ÚCheckConnectionCallsz#RegressionTests.CheckConnectionCallcCs"dd„}| tj|jjd|¡dS)NcSsdS)NrKr)ÚaÚbrrr Ú collation_cb sz4RegressionTests.CheckCollation..collation_cbuí²€)r6rrYrÚcreate_collation)rrlrrr ÚCheckCollation szRegressionTests.CheckCollationc sft d¡}| ¡‰ˆ d¡ˆ d¡‡fdd„}| tj¡ˆ ddd„|ƒDƒ¡Wd QRXd S) zÝ http://bugs.python.org/issue10811 Recursively using a cursor, such as when reusing it from a generator led to segfaults. Now we catch recursive cursor usage and raise a ProgrammingError. z:memory:zcreate table a (bar)zcreate table b (baz)c3sˆ dd¡dVdS)Nzinsert into a (bar) values (?))rKrK)rr)rrr Úfoos z4RegressionTests.CheckRecursiveCursorUse..foozinsert into b (baz) values (?)css|] }|fVqdS)Nr)rrrrr ú $sz:RegressionTests.CheckRecursiveCursorUse..N)rrr rr6rYr)rrror)rr ÚCheckRecursiveCursorUses    z'RegressionTests.CheckRecursiveCursorUsec CsŠtjdtjd}| ¡}| d¡| d¡| d¡| d¡dd„| ¡Dƒ}| |t d d d d d d d¡t d d d d d d d¡g¡dS)zË http://bugs.python.org/issue14720 The microsecond parsing of convert_timestamp() should pad with zeros, since the microsecond string "456" actually represents "456000". z:memory:)rzCREATE TABLE t (x TIMESTAMP)z4INSERT INTO t (x) VALUES ('2012-04-04 15:06:00.456')z:INSERT INTO t (x) VALUES ('2012-04-04 15:06:00.123456789')zSELECT * FROM tcSsg|] }|d‘qS)rr)rrrrr r9szKRegressionTests.CheckConvertTimestampMicrosecondPadding..iÜrbééri@õi@âN)rrr1r rrZr r2)rrrÚvaluesrrr Ú'CheckConvertTimestampMicrosecondPadding&s    z7RegressionTests.CheckConvertTimestampMicrosecondPaddingcCs|jttjddddS)Nz:memory:é{)rC)r6r?rr)rrrr ÚCheckInvalidIsolationLevelType@sz.RegressionTests.CheckInvalidIsolationLevelTypecCsRt d¡}| t|d¡| t|d¡| ¡}| t|jd¡| t|jd¡dS)Nz:memory:z select 1z select 1z select 2z select 2)rrr6rRr r)rrrrrr ÚCheckNullCharacterFs  z"RegressionTests.CheckNullCharacterc CsØt d¡}| d¡| |jd¡d}xžt| d¡ƒD]Œ\}}|j||dp| d|f¡| ¡|dkr~| |dd¡n2|dkr˜| |dd¡n|d kr°| |dd ¡|d7}Wd QRXq6W| |d d ¡d S) zÉ Connection.commit() did reset cursors, which made sqlite3 to return rows multiple times when fetched from cursors after commit. See issues 10513 and 23129 for details. z:memory:z£ create table t(c); create table t2(c); insert into t values(0); insert into t values(1); insert into t values(2); r/rzselect c from t)rrfzinsert into t2(c) values (?)rKr(Nr'z'should have returned exactly three rows) rrÚ executescriptr rCÚ enumeraterrOr*)rrZcounterrrfrrr ÚCheckCommitCursorResetOs" z&RegressionTests.CheckCommitCursorResetcCsBdd„}t d¡}t |¡}t ||¡}| |¡~~t ¡dS)zm The interpreter shouldn't crash in case Cursor.__init__() is called more than once. cWsdS)Nr)Úargsrrr Úcallbackssz/RegressionTests.CheckBpo31770..callbackz:memory:N)rrrXÚweakrefÚrefrWrÚ gc_collect)rr}rrrrrr Ú CheckBpo31770ns    zRegressionTests.CheckBpo31770c Cs | t¡ |j`WdQRXdS)N)r6ÚAttributeErrorrrC)rrrr ÚCheckDelIsolation_levelSegfault~s z/RegressionTests.CheckDelIsolation_levelSegfaultN)$rErFrGr r rrrr"r&ÚunittestÚskipIfrÚsqlite_version_infor-r.r0r4r>rArVr\r^rardrergrirnrqrurwrxr{rrƒrrrr rs:      rc@s"eZdZdZddd„Zdd„ZdS)ÚUnhashableFuncNcCsd|_||_dS)Nr)ÚcallsÚ return_value)rr‰rrr rW†szUnhashableFunc.__init__cOs|jd7_|jS)NrK)rˆr‰)rr|Úkwargsrrr Ú__call__ŠszUnhashableFunc.__call__)N)rErFrGÚ__hash__rWr‹rrrr r‡ƒs r‡c@s@eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dS)ÚUnhashableCallbacksTestCasez° https://bugs.python.org/issue34052 Registering unhashable callbacks raises TypeError, callbacks are not registered in SQLite after such registration attempt. cCst d¡|_dS)Nz:memory:)rrr)rrrr r –sz!UnhashableCallbacksTestCase.setUpcCs|j ¡dS)N)rr )rrrr r ™sz$UnhashableCallbacksTestCase.tearDownc CsLtdd}| td¡|j |d¡WdQRX|j d¡| |j¡dS)Nr)r‰zunhashable typerKzSELECT 1)r‡r[r?rÚset_progress_handlerrÚ assertFalserˆ)rÚfrrr Útest_progress_handlerœs   z1UnhashableCallbacksTestCase.test_progress_handlerc Cstd}tƒ}| td¡|j |d|¡WdQRXd|}| tj|¡|j d|¡WdQRX| |j ¡dS)NÚ func_namezunhashable typerzno such function: %sz SELECT %s()) r‡r[r?rÚcreate_functionrr+rrrˆ)rr’rr=rrr Ú test_func£sz%UnhashableCallbacksTestCase.test_funcc CsLttjd}| td¡|j |¡WdQRX|j d¡| |j ¡dS)N)r‰zunhashable typezSELECT 1) r‡rÚ SQLITE_DENYr[r?rÚset_authorizerrrrˆ)rrrrr Útest_authorizer­s   z+UnhashableCallbacksTestCase.test_authorizerc CszGdd„dtƒ}d}| td¡|j |d|ddiƒ¡WdQRXd|}| tj|¡|j d |¡WdQRXdS) Nc@seZdZdZdS)z=UnhashableCallbacksTestCase.test_aggr..UnhashableTypeN)rErFrGrŒrrrr ÚUnhashableTypeµsr˜Ú aggr_namezunhashable typerZAggrrzno such function: %sz SELECT %s())Útyper[r?rÚcreate_aggregaterr+r)rr˜r™r=rrr Ú test_aggr´s"z%UnhashableCallbacksTestCase.test_aggrN) rErFrGÚ__doc__r r r‘r”r—rœrrrr rs rcCs t td¡}t |t t¡f¡S)NÚCheck)r„Ú makeSuiterÚ TestSuiter)Zregression_suiterrr Úsuite¿s r¡cCst ¡}| tƒ¡dS)N)r„ÚTextTestRunnerÚrunr¡)Úrunnerrrr ÚtestÆsr¥Ú__main__) r2r„Úsqlite3rr~r¥rÚTestCaserr‡rr¡rErrrr Ús g 0