B ͦZ^@@sddlZddlZddlZddlZddlmZGdddejZGdddZ GdddejZ d d Z d d Ze d kr|edS)N)supportc@seZdZddZddZddZddZd d Zd d Zd dZ e e j dkdddZddZddZddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd5d6Z d7d8Z!d9d:Z"d;d<Z#d=S)>RegressionTestscCstd|_dS)Nz:memory:)sqliteconnectcon)selfrA/work/yifan.wang/lscsoft/opt/python37/lib/python3.7/regression.pysetUpszRegressionTests.setUpcCs|jdS)N)rclose)rrrr tearDown"szRegressionTests.tearDowncCs|j}|ddS)Nzpragma user_version)rcursorexecute)rcurrrr CheckPragmaUserVersion%s z&RegressionTests.CheckPragmaUserVersioncCs@tjdtjd}z|j}|dWd||XdS)Nz:memory:) detect_typeszpragma schema_version)rrZPARSE_COLNAMESrr rr )rrrrrr CheckPragmaSchemaVersion*s  z(RegressionTests.CheckPragmaSchemaVersioncstjdddfddtdD}|ddx.tdD]"}|dd d dtdDq9sz7RegressionTests.CheckStatementReset..rzcreate table test(x) zinsert into test(x) values (?)cSsg|] }|fqSrr)rrrrr r<s zselect x from test)rrranger executemanyrollback)rcursorsir)rr CheckStatementReset4s"z#RegressionTests.CheckStatementResetcCsN|j}|d||jddd|d||jddddS)Nz select 1 as "foo bar [datetime]"rzfoo barzselect 1 as "foo baz"zfoo baz)rr r assertEqual description)rrrrr CheckColumnNameWithSpacesCs    z)RegressionTests.CheckColumnNameWithSpacescCsPtd}g}x4tdD](}|}|||dt|qW|dS)Nz:memory:izselect 1 x union select )rrrr appendrstrr )rrrrrrrr #CheckStatementFinalizationOnCloseDbKs  z3RegressionTests.CheckStatementFinalizationOnCloseDb)r'zneeds sqlite 3.2.2 or newercCstd}|d|dy|dWntjk rBYnX|dy |Wn tjk rz|dYnXdS)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 DatabaseErrorcommitOperationalErrorZfail)rrrrr CheckOnConflictRollbackYs     z'RegressionTests.CheckOnConflictRollbackcCs(|jd|jd|jddS)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|jddS)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|dtjf|||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)rrPARSE_DECLTYPESrdatetimeZnow)rZSELECTrrrr CheckTypeMapUsagexs     z!RegressionTests.CheckTypeMapUsagec CsP|tj$}|jdttdgfWdQRXd}||t |j dS)Nz"select 'xxx' || ? || 'yyy' colnamez9Could not decode to UTF-8 column 'colname' with text 'xxx) assertRaisesrr)rrbytes bytearrayfetchoneZassertInr$Z exception)rcmmsgrrr CheckErrorMsgDecodeErrors  z(RegressionTests.CheckErrorMsgDecodeErrorcCs|ttjiddS)z! See issue 3312. N)r2 TypeErrorrZregister_adapter)rrrr CheckRegisterAdaptersz$RegressionTests.CheckRegisterAdapterc sGfdddt}tdd_xHdD]@}|j|d*|_|_|_||_WdQRXq*Wd_d_dtfdtfd tfd tfd tfg}xR|D]J\}}|j|d0| | |_WdQRX| jdWdQRXqWdS) Ncs eZdZddZfddZdS)z9RegressionTests.CheckSetIsolationLevel..CustomStrcSsdS)Nr)rrrr uppersz?RegressionTests.CheckSetIsolationLevel..CustomStr.uppercs d_dS)Nr,)isolation_level)r)rrr __del__szARegressionTests.CheckSetIsolationLevel..CustomStr.__del__N)__name__ __module__ __qualname__r;r=r)rrr CustomStrsrAz:memory:)r,DEFERREDZ IMMEDIATEZ EXCLUSIVE)levelrBabczIMMEDIATEEXCLUSIVEé) r$rrr<subTestlower capitalizer9 ValueErrorr2r)rrArCZpairsvalueexcr)rr CheckSetIsolationLevels$     z&RegressionTests.CheckSetIsolationLevelc CspGdddtj}td}||}|tj|dWdQRX|tjd|WdQRXdS)zd Verifies that cursor methods check whether base class __init__ was called. c@seZdZddZdS)z?RegressionTests.CheckCursorConstructorCallCheck..CursorcSsdS)Nr)rrrrr __init__szHRegressionTests.CheckCursorConstructorCallCheck..Cursor.__init__N)r>r?r@rOrrrr CursorsrPz:memory:z select 4+5Nz$^Base Cursor\.__init__ not called\.$) rrPrr2ProgrammingErrorrfetchallassertRaisesRegexr )rrPrrrrr CheckCursorConstructorCallChecks z/RegressionTests.CheckCursorConstructorCallCheckcCs(Gdddt}|jd|dfdS)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 MyStrsrUzselect ?rFN)r$rr)rrUrrr CheckStrSubclasssz RegressionTests.CheckStrSubclassc Cs>Gdddtj}|d}|tj|}WdQRXdS)zh Verifies that connection methods check whether base class __init__ was called. c@seZdZddZdS)zGRegressionTests.CheckConnectionConstructorCallCheck..ConnectioncSsdS)Nr)rnamerrr rOszPRegressionTests.CheckConnectionConstructorCallCheck..Connection.__init__N)r>r?r@rOrrrr ConnectionsrXz:memory:N)rrXr2rQr )rrXrrrrr #CheckConnectionConstructorCallChecksz3RegressionTests.CheckConnectionConstructorCallCheckc sGfdddtj}Gdddtj|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) cseZdZfddZdS)z;RegressionTests.CheckCursorRegistration..Connectioncs|S)Nr)r)rPrr r szBRegressionTests.CheckCursorRegistration..Connection.cursorN)r>r?r@r r)rPrr rXsrXc@seZdZddZdS)z7RegressionTests.CheckCursorRegistration..CursorcSstj||dS)N)rrPrO)rrrrr rOsz@RegressionTests.CheckCursorRegistration..Cursor.__init__N)r>r?r@rOrrrr rPsrPz:memory:zcreate table foo(x)zinsert into foo(x) values (?))r&))rzselect x from fooN) rrXrPr rrrr2ZInterfaceErrorrR)rrXrrr)rPr CheckCursorRegistrations  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 CheckAutoCommitszRegressionTests.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)rrrowrrr CheckPragmaAutocommits     z%RegressionTests.CheckPragmaAutocommitcCs|tj|jddS)z} Call a connection with a non-string SQL request: check error handling of the statement constructor. rDN)r2rWarningr)rrrr CheckConnectionCallsz#RegressionTests.CheckConnectionCallcCs"dd}|tj|jjd|dS)NcSsdS)NrDr)abrrr collation_cb sz4RegressionTests.CheckCollation..collation_cbu)r2rrQrZcreate_collation)rrcrrr CheckCollation szRegressionTests.CheckCollationc sftd}|ddfdd}|tjddd|DWd 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)c3sdddVdS)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 Cstjdtjd}|}|d|d|d|ddd|D}||td d d d d d dtd d d d d d dgdS)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|] }|dqS)rr)rrrrr r9szKRegressionTests.CheckConvertTimestampMicrosecondPadding..irZri@i@N)rrr.r rrRrr/)rrrvaluesrrr 'CheckConvertTimestampMicrosecondPadding&s    z7RegressionTests.CheckConvertTimestampMicrosecondPaddingcCs|jttjddddS)Nz:memory:{)r<)r2r9rr)rrrr CheckInvalidIsolationLevelType@sz.RegressionTests.CheckInvalidIsolationLevelTypecCsRtd}|t|d|t|d|}|t|jd|t|jddS)Nz:memory:z select 1z select 1z select 2z select 2)rrr2rKr r)rrrrrr CheckNullCharacterFs  z"RegressionTests.CheckNullCharacterc Cstd}|d||jdd}xt|dD]\}}|j||dp|d|f||dkr~||ddn2|dkr||ddn|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}td}t|}t||}||~~tdS)zm The interpreter shouldn't crash in case Cursor.__init__() is called more than once. cWsdS)Nr)argsrrr callbackssz/RegressionTests.CheckBpo31770..callbackz:memory:N)rrrPweakrefrefrOrZ gc_collect)rrrrrrtrrr CheckBpo31770ns    zRegressionTests.CheckBpo31770c Cs |t |j`WdQRXdS)N)r2AttributeErrorrr<)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"eZdZdZdddZddZdS)UnhashableFuncNcCsd|_||_dS)Nr)calls return_value)rr{rrr rOszUnhashableFunc.__init__cOs|jd7_|jS)NrD)rzr{)rrqkwargsrrr __call__szUnhashableFunc.__call__)N)r>r?r@__hash__rOr}rrrr rys ryc@s@eZdZdZddZddZddZdd 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. cCstd|_dS)Nz:memory:)rrr)rrrr r sz!UnhashableCallbacksTestCase.setUpcCs|jdS)N)rr )rrrr r sz$UnhashableCallbacksTestCase.tearDownc CsLtdd}|td|j|dWdQRX|jd||jdS)Nr)r{zunhashable typerDzSELECT 1)ryrSr9rZset_progress_handlerr assertFalserz)rfrrr test_progress_handlers   z1UnhashableCallbacksTestCase.test_progress_handlerc Cstd}t}|td|j|d|WdQRXd|}|tj||jd|WdQRX||j dS)N func_namezunhashable typerzno such function: %sz SELECT %s()) ryrSr9rZcreate_functionrr)rrrz)rrrr7rrr test_funcsz%UnhashableCallbacksTestCase.test_funcc CsLttjd}|td|j|WdQRX|jd||j dS)N)r{zunhashable typezSELECT 1) ryrZ SQLITE_DENYrSr9rZset_authorizerrrrz)rrrrr test_authorizers   z+UnhashableCallbacksTestCase.test_authorizerc CszGdddt}d}|td|j|d|ddiWdQRXd|}|tj||jd |WdQRXdS) Nc@seZdZdZdS)z=UnhashableCallbacksTestCase.test_aggr..UnhashableTypeN)r>r?r@r~rrrr UnhashableTypesr aggr_namezunhashable typerZAggrrzno such function: %sz SELECT %s())typerSr9rZcreate_aggregaterr)r)rrrr7rrr test_aggrs"z%UnhashableCallbacksTestCase.test_aggrN) r>r?r@__doc__r r rrrrrrrr rs rcCs ttd}t|ttfS)NZCheck)rxZ makeSuiterZ TestSuiter)Zregression_suiterrr suites rcCst}|tdS)N)rxZTextTestRunnerrunr)Zrunnerrrr testsr__main__) r/rxZsqlite3rrsrrZTestCaserryrrr>rrrr s g 0