B 2*™\eBã@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úBC:\ALexclude\prg\programme\Python37\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)rrZPARSE_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)rrrZ DatabaseErrorÚcommitÚOperationalErrorZfail)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ÚdatetimeZnow)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ÚfetchoneZassertInr$Z exception)rÚcmÚmsgrrr ÚCheckErrorMsgDecodeError‡s  z(RegressionTests.CheckErrorMsgDecodeErrorcCs| ttjid¡dS)z! See issue 3312. N)r2Ú TypeErrorrZregister_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__r;r=r)rrr Ú CustomStr˜srAz:memory:)r,ÚDEFERREDZ IMMEDIATEZ EXCLUSIVE)ÚlevelrBéóÚabczIMMEDIATEEXCLUSIVEõé) r$rrr<ÚsubTestÚlowerÚ capitalizer9Ú ValueErrorr2r)rrArCZpairsÚ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)r>r?r@rOrrrr ÚCursor¹srPz:memory:z select 4+5Nz$^Base Cursor\.__init__ not called\.$) rrPrr2ÚProgrammingErrorrÚfetchallÚassertRaisesRegexr )rrPrrrrr Ú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)r>r?r@rrrr ÚMyStrÉsrUzselect ?rFN)r$rr)rrUrrr Ú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 rOÒszPRegressionTests.CheckConnectionConstructorCallCheck..Connection.__init__N)r>r?r@rOrrrr Ú ConnectionÑsrXz:memory:N)rrXr2rQr )rrXrrrrr Ú#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)rPrr r ßszBRegressionTests.CheckCursorRegistration..Connection.cursorN)r>r?r@r r)rPrr rXÞsrXc@seZdZdd„ZdS)z7RegressionTests.CheckCursorRegistration..CursorcSstj ||¡dS)N)rrPrO)rrrrr rOãsz@RegressionTests.CheckCursorRegistration..Cursor.__init__N)r>r?r@rOrrrr rPâsrPz:memory:zcreate table foo(x)zinsert into foo(x) values (?))r&)é)rzselect x from fooN) rrXrPr rrrr2ZInterfaceErrorrR)rrXrrr)rPr Ú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)r<)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 rr5)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. rDN)r2rÚWarningr)rrrr ÚCheckConnectionCallsz#RegressionTests.CheckConnectionCallcCs"dd„}| tj|jjd|¡dS)NcSsdS)NrDr)ÚaÚbrrr Ú collation_cb sz4RegressionTests.CheckCollation..collation_cbuí²€)r2rrQrZcreate_collation)rrcrrr Ú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 (?))rDrD)rr)rrr Úfoos z4RegressionTests.CheckRecursiveCursorUse..foozinsert into b (baz) values (?)css|] }|fVqdS)Nr)rrrrr ú $sz:RegressionTests.CheckRecursiveCursorUse..N)rrr rr2rQr)rrrer)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ÜrZééri@õi@âN)rrr.r rrRrr/)rrrÚvaluesrrr Ú'CheckConvertTimestampMicrosecondPadding&s    z7RegressionTests.CheckConvertTimestampMicrosecondPaddingcCs|jttjddddS)Nz:memory:é{)r<)r2r9rr)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)rrr2rKr 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)rr]zinsert into t2(c) values (?)rDr'Nr&z'should have returned exactly three rows) rrZ executescriptrr<Ú enumeraterrHr()rrZcounterrr]rrr Ú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)rrrPÚweakrefÚrefrOrZ gc_collect)rrrrrrtrrr Ú CheckBpo31770ns    zRegressionTests.CheckBpo31770c Cs | t¡ |j`WdQRXdS)N)r2ÚAttributeErrorrr<)rrrr ÚCheckDelIsolation_levelSegfault~s z/RegressionTests.CheckDelIsolation_levelSegfaultN)$r>r?r@r r rrrr!r%ÚunittestZskipIfrZsqlite_version_infor*r+r-r0r8r:rNrTrVrYr[r\r^r`rdrgrkrmrnrprurwrrrr rs:      rc@s"eZdZdZddd„Zdd„ZdS)ÚUnhashableFuncNcCsd|_||_dS)Nr)ÚcallsÚ return_value)rr{rrr rO†szUnhashableFunc.__init__cOs|jd7_|jS)NrD)rzr{)rrqÚkwargsrrr Ú__call__ŠszUnhashableFunc.__call__)N)r>r?r@Ú__hash__rOr}rrrr ryƒs ryc@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 typerDzSELECT 1)ryrSr9rZset_progress_handlerrÚ assertFalserz)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()) ryrSr9rZcreate_functionrr)rr€rz)rrƒrr7rrr Ú test_func£sz%UnhashableCallbacksTestCase.test_funcc CsLttjd}| td¡|j |¡WdQRX|j d¡| |j ¡dS)N)r{zunhashable typezSELECT 1) ryrZ SQLITE_DENYrSr9rZset_authorizerrr€rz)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)r>r?r@r~rrrr ÚUnhashableTypeµsr†Ú aggr_namezunhashable typerZAggrrzno such function: %sz SELECT %s())ÚtyperSr9rZcreate_aggregaterr)r)rr†r‡r7rrr Ú test_aggr´s"z%UnhashableCallbacksTestCase.test_aggrN) r>r?r@Ú__doc__r r r‚r„r…r‰rrrr rs rcCs t td¡}t |t t¡f¡S)NZCheck)rxZ makeSuiterZ TestSuiter)Zregression_suiterrr Úsuite¿s r‹cCst ¡}| tƒ¡dS)N)rxZTextTestRunnerÚrunr‹)Zrunnerrrr ÚtestÆsrÚ__main__) r/rxZsqlite3rrsrrZTestCaserryrr‹r>rrrr Ús g 0