Your IP : 216.73.216.213
��Yf 8 � @ sl d d l Z d d l Z d d l Z Gd d � d e j � Z d d � Z d d � Z e d k rh e � d S) � Nc @ s{ e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z e
j e j
d= k d � d d � � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d! � Z d" d# � Z d$ d% � Z d& d' � Z d( d) � Z d* d+ � Z d, d- � Z d. d/ � Z d0 d1 � Z d2 d3 � Z d4 d5 � Z d6 d7 � Z d8 d9 � Z! d: d; � Z" d< S)>�RegressionTestsc C s t j d � | _ d S)Nz:memory:)�sqlite�connect�con)�self� r �</opt/alt/python35/lib64/python3.5/sqlite3/test/regression.py�setUp s zRegressionTests.setUpc C s | j j � d S)N)r �close)r r r r �tearDown s zRegressionTests.tearDownc C s | j j � } | j d � d S)Nzpragma user_version)r �cursor�execute)r �curr r r �CheckPragmaUserVersion# s z&RegressionTests.CheckPragmaUserVersionc
C sT t j d d t j �} z | j j � } | j d � Wd | j � | j � Xd S)Nz:memory:� detect_typeszpragma schema_version)r r ZPARSE_COLNAMESr r r
r
)r r r r r r �CheckPragmaSchemaVersion( s
z(RegressionTests.CheckPragmaSchemaVersionc s� t j d d d �� � f d d � t d � D� } | d j d � x; t d � D]- } | d j d d
d � t d � D� � qR Wx- t d � D] } | | j d | d � q� W� j � d S)
Nz:memory:Zcached_statements� c s g | ] } � j � � q Sr )r )�.0�x)r r r �
<listcomp>7 s z7RegressionTests.CheckStatementReset.<locals>.<listcomp>r zcreate table test(x)�
z insert into test(x) values (?)c S s g | ] } | f � q Sr r )r r r r r r : s � zselect x from test)r r �ranger
� executemany�rollback)r �cursors�ir )r r �CheckStatementReset2 s + z#RegressionTests.CheckStatementResetc C sc | j j � } | j d � | j | j d d d � | j d � | j | j d d d � d S)Nz select 1 as "foo bar [datetime]"r zfoo barzselect 1 as "foo baz"zfoo baz)r r r
� assertEqualZ description)r r r r r �CheckColumnNameWithSpacesA s
z)RegressionTests.CheckColumnNameWithSpacesc C sj t j d � } g } xD t d � D]6 } | j � } | j | � | j d t | � � q" W| j � d S)Nz:memory:�i zselect 1 x union select )r r r r �appendr
�strr
)r r r r r r r r �#CheckStatementFinalizationOnCloseDbI s
z3RegressionTests.CheckStatementFinalizationOnCloseDb� � zneeds sqlite 3.2.2 or newerc C s� t j d � } | j d � | j d � y | j d � Wn t j k
rQ Yn X| j d � y | j � Wn" t j k
r� | j d � Yn Xd S)Nz:memory:z3create table foo(x, unique(x) on conflict rollback)z insert into foo(x) values (1)z insert into foo(x) values (2)z1pysqlite knew nothing about the implicit ROLLBACK)r r r
Z
DatabaseError�commit�OperationalErrorZfail)r r r r r �CheckOnConflictRollbackW s
z'RegressionTests.CheckOnConflictRollbackc C s4 | j j d � | j j d � | j j d � d S)zm
pysqlite would crash with older SQLite versions unless
a workaround is implemented.
zcreate table foo(bar)zdrop table fooN)r r
)r r r r �-CheckWorkaroundForBuggySqliteTransferBindingsf s z=RegressionTests.CheckWorkaroundForBuggySqliteTransferBindingsc C s | j j d � d S)z
pysqlite used to segfault with SQLite versions 3.5.x. These return NULL
for "no-operation" statements
� N)r r
)r r r r �CheckEmptyStatemento s z#RegressionTests.CheckEmptyStatementc C s� d } t j d d t j �} | j d � | j d t j j � f � | j | � | j d � | j d � | j d � | j | � 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:r z create table foo(bar timestamp)z insert into foo(bar) values (?)zdrop table fooz create table foo(bar integer)z insert into foo(bar) values (5)N)r r �PARSE_DECLTYPESr
�datetimeZnow)r ZSELECTr r r r �CheckTypeMapUsagev s
z!RegressionTests.CheckTypeMapUsagec C sj | j t j � �2 } | j j d t t d g � � f � j � Wd QRXd } | j | t | j
� � d S)Nz"select 'xxx' || ? || 'yyy' colname� z9Could not decode to UTF-8 column 'colname' with text 'xxx) � assertRaisesr r' r r
�bytes� bytearray�fetchoneZassertInr" Z exception)r �cm�msgr r r �CheckErrorMsgDecodeError� s
&z(RegressionTests.CheckErrorMsgDecodeErrorc C s | j t t j i d � d S)z!
See issue 3312.
N)r0 � TypeErrorr Zregister_adapter)r r r r �CheckRegisterAdapter� s z$RegressionTests.CheckRegisterAdapterc sB G� f d d � d t � } t j d � � d � _ x^ d D]V } | j d | � �; | � _ | j � � _ | j � � _ | | � � _ Wd QRXq; Wd � _ d � _ d t f d
t f d t f d t f d
t f g } xa | D]Y \ } } | j d | � �8 | j | � � | � _ Wd QRX| j
� j d � Wd QRXq� Wd S)Nc s. e Z d Z d d � Z � f d d � Z d S)z9RegressionTests.CheckSetIsolationLevel.<locals>.CustomStrc S s d S)Nr )r r r r �upper� s z?RegressionTests.CheckSetIsolationLevel.<locals>.CustomStr.upperc s
d � _ d S)Nr* )�isolation_level)r )r r r �__del__� s zARegressionTests.CheckSetIsolationLevel.<locals>.CustomStr.__del__N)�__name__�
__module__� __qualname__r9 r; r )r r r � CustomStr� s r? z:memory:r* �DEFERRED� IMMEDIATE� EXCLUSIVE�level� � �abczIMMEDIATE EXCLUSIVE� é)r* r@ rA rB ) r" r r r: �subTest�lower�
capitalizer7 �
ValueErrorr0 r )r r? rC Zpairs�value�excr )r r �CheckSetIsolationLevel� s$
z&RegressionTests.CheckSetIsolationLevelc C se Gd d � d t j � } t j d � } | | � } | j t j � � | j d � j � Wd QRXd S)zd
Verifies that cursor methods check whether base class __init__ was
called.
c @ s e Z d Z d d � Z d S)z?RegressionTests.CheckCursorConstructorCallCheck.<locals>.Cursorc S s d S)Nr )r r r r r �__init__� s zHRegressionTests.CheckCursorConstructorCallCheck.<locals>.Cursor.__init__N)r< r= r> rO r r r r �Cursor� s rP z:memory:z
select 4+5N)r rP r r0 �ProgrammingErrorr
�fetchall)r rP r r r r r � CheckCursorConstructorCallCheck� s
z/RegressionTests.CheckCursorConstructorCallCheckc C s6 Gd d � d t � } | j j d | d � f � d S)za
The Python 3.0 port of the module didn't cope with values of subclasses of str.
c @ s e Z d Z d S)z/RegressionTests.CheckStrSubclass.<locals>.MyStrN)r< r= r> r r r r �MyStr� s rT zselect ?rF N)r" r r
)r rT r r r �CheckStrSubclass� s z RegressionTests.CheckStrSubclassc C sO Gd d � d t j � } | d � } | j t j � � | j � } Wd QRXd S)zh
Verifies that connection methods check whether base class __init__ was
called.
c @ s e Z d Z d d � Z d S)zGRegressionTests.CheckConnectionConstructorCallCheck.<locals>.Connectionc S s d S)Nr )r �namer r r rO � s zPRegressionTests.CheckConnectionConstructorCallCheck.<locals>.Connection.__init__N)r< r= r> rO r r r r �
Connection� s rW z:memory:N)r rW r0 rQ r )r rW r r r r r �#CheckConnectionConstructorCallCheck� s z3RegressionTests.CheckConnectionConstructorCallCheckc s� G� f d d � d t j � } Gd d � d t j � � | d � } | j � } | j d � | j d d
d d g � | j d � | j � | j t j � � | j � Wd QRXd S)z�
Verifies that subclassed cursor classes are correctly registered with
the connection object, too. (fetch-across-rollback problem)
c s" e Z d Z � f d d � Z d S)z;RegressionTests.CheckCursorRegistration.<locals>.Connectionc s
� | � S)Nr )r )rP r r r � s zBRegressionTests.CheckCursorRegistration.<locals>.Connection.cursorN)r< r= r> r r )rP r r rW � s rW c @ s e Z d Z d d � Z d S)z7RegressionTests.CheckCursorRegistration.<locals>.Cursorc S s t j j | | � d S)N)r rP rO )r r r r r rO � s z@RegressionTests.CheckCursorRegistration.<locals>.Cursor.__init__N)r< r= r> rO r r r r rP � s rP z:memory:zcreate table foo(x)z insert into foo(x) values (?)r$ � r zselect x from fooN)r$ )rY )r )
r rW rP r r
r r r0 ZInterfaceErrorrR )r rW r r r )rP r �CheckCursorRegistration� s
z'RegressionTests.CheckCursorRegistrationc C s t j d d d �} d S)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:r: N)r r )r r r r r �CheckAutoCommit� s z RegressionTests.CheckAutoCommitc C sF | j j � } | j d � | j d � | j d � | j � } d S)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)z insert into foo(bar) values (5)zpragma page_sizeN)r r r
r3 )r r �rowr r r �CheckPragmaAutocommit� s
z%RegressionTests.CheckPragmaAutocommitc C s� Gd d � d � } | � } | j t | j j | � | j t | j j | � | j t | j j | � | j t | j j | � d S)z�
See http://bugs.python.org/issue7478
It was possible to successfully register callbacks that could not be
hashed. Return codes of PyDict_SetItem were not checked properly.
c @ s( e Z d Z d d � Z d d � Z d S)z1RegressionTests.CheckSetDict.<locals>.NotHashablec _ s d S)Nr )r �args�kwr r r �__call__ s z:RegressionTests.CheckSetDict.<locals>.NotHashable.__call__c S s
t � � d S)N)r7 )r r r r �__hash__ s z:RegressionTests.CheckSetDict.<locals>.NotHashable.__hash__N)r< r= r> r` ra r r r r � NotHashable s rb N)r0 r7 r Zcreate_functionZcreate_aggregateZset_authorizerZset_progress_handler)r rb �varr r r � CheckSetDict� s z RegressionTests.CheckSetDictc C s | j t j | j d � d S)z}
Call a connection with a non-string SQL request: check error handling
of the statement constructor.
rD N)r0 r �Warningr )r r r r �CheckConnectionCall s z#RegressionTests.CheckConnectionCallc C s/ d d � } | j t j | j j d | � d S)Nc S s d S)NrD r )�a�br r r � collation_cb s z4RegressionTests.CheckCollation.<locals>.collation_cbu �)r0 r rQ r Zcreate_collation)r ri r r r �CheckCollation s z RegressionTests.CheckCollationc s� t j d � } | j � � � j d � � j d � � f d d � } | j t j � �% � j d d d � | � 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)c 3 s � j d d � d Vd S)Nz insert into a (bar) values (?)rD )rD )r
r )r r r �foo+ s z4RegressionTests.CheckRecursiveCursorUse.<locals>.fooz insert into b (baz) values (?)c s s | ] } | f Vq d S)Nr )r r r r r � <genexpr>1 s z:RegressionTests.CheckRecursiveCursorUse.<locals>.<genexpr>N)r r r r
r0 rQ r )r r rk r )r r �CheckRecursiveCursorUse s
z'RegressionTests.CheckRecursiveCursorUsec C s� t j d d t j �} | j � } | j d � | j d � | j d � | j d � d d � | j � D� } | j | t j d d
d
d d d
d � t j d d
d
d d d
d � g � d S)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:r z CREATE 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 tc S s g | ] } | d � q S)r r )r r r r r r F s zKRegressionTests.CheckConvertTimestampMicrosecondPadding.<locals>.<listcomp>i� rY � � r i@� i@� N)r r r, r r
rR r r- )r r r �valuesr r r �'CheckConvertTimestampMicrosecondPadding3 s
z7RegressionTests.CheckConvertTimestampMicrosecondPaddingc C s | j t t j d d d �d S)Nz:memory:r: �{ )r0 r7 r r )r r r r � CheckInvalidIsolationLevelTypeM s z.RegressionTests.CheckInvalidIsolationLevelTypec C sq t j d � } | j t | d � | j t | d � | j � } | j t | j d � | j t | j d � d S)Nz:memory:z select 1z select 1 z
select 2z select 2 )r r r0 rK r r
)r r r r r r �CheckNullCharacterS s z"RegressionTests.CheckNullCharacterc C s t j d � } | j d � | j | j d � d } x� t | j d � � D]� \ } } | j d | d | � �� | j d | f � | j � | d k r� | j | d d � nC | d k r� | j | d d � n | d
k r� | j | d d
� | d 7} Wd QRXqK W| j | 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* r zselect c from tr r\ z insert into t2(c) values (?)rD r% Nr$ z'should have returned exactly three rows) r r Z
executescriptr r: � enumerater
rH r& )r r Zcounterr r\ r r r �CheckCommitCursorReset\ s" "
z&RegressionTests.CheckCommitCursorResetN)r$ r% r% )#r< r= r> r r r r r r r# �unittestZskipIfr Zsqlite_version_infor( r) r+ r. r6 r8 rN rS rU rX rZ r[ r] rd rf rj rm rq rs rt rv r r r r r s8
'
r c C s" t j t d � } t j | f � S)NZCheck)rw Z makeSuiter Z TestSuite)Zregression_suiter r r �suite| s rx c C s t j � } | j t � � d S)N)rw ZTextTestRunnerZrunrx )Zrunnerr r r �test� s ry �__main__) r- rw Zsqlite3r ZTestCaser rx ry r< r r r r �<module> s � a