Your IP : 216.73.216.213
�
e fx2 � @ so 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 rk e � n d S) � Nc @ sT 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 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 d- d. � Z d/ d0 � Z d1 d2 � Z d3 d4 � Z d5 d6 � Z d7 S)8�RegressionTestsc C s t j d � | _ d S)Nz:memory:)�sqlite�connect�con)�self� r �</opt/alt/python34/lib64/python3.4/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.CheckStatementFinalizationOnCloseDbc C s� t j d k r d St j d � } | j d � | j d � y | j d � Wn t j k
rd Yn X| j d � y | j � Wn" t j k
r� | j d � Yn Xd S) N� � z: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% )r Zsqlite_version_infor r
Z
DatabaseErrorZcommit�OperationalError�fail)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 �-CheckWorkaroundForBuggySqliteTransferBindingsg 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 �CheckEmptyStatementp 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 �CheckTypeMapUsagew s
z!RegressionTests.CheckTypeMapUsagec C s� d } y5 | j j d t t d g � � f � j � d } WnM t j k
r� } z* | j d } | j d � sx d } n WYd d } ~ Xn X| r� | j | � n d S)Nz"select 'xxx' || ? || 'yyy' colname� z@should have raised an OperationalError with detailed descriptionr z9Could not decode to UTF-8 column 'colname' with text 'xxxz7OperationalError did not have expected description text)
r r
�bytes� bytearray�fetchoner r& �args�
startswithr' )r �failure�e�msgr r r �CheckErrorMsgDecodeError� s
z(RegressionTests.CheckErrorMsgDecodeErrorc C s | j t t j i d � d S)z!
See issue 3312.
N)� assertRaises� TypeErrorr Zregister_adapter)r r r r �CheckRegisterAdapter� s z$RegressionTests.CheckRegisterAdapterc C s# t j d � } t | d d � d S)z!
See issue 3312.
z:memory:�isolation_level� éN)r r �setattr)r r r r r �CheckSetIsolationLevel� s z&RegressionTests.CheckSetIsolationLevelc C s� Gd d � d t j � } t j d � } | | � } y$ | j d � j � | j d � Wn) t j k
ro Yn | j d � Yn Xd 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)�__name__�
__module__� __qualname__r@ r r r r �Cursor� s rD z:memory:z
select 4+5z#should have raised ProgrammingErrorN)r rD r r
�fetchallr' �ProgrammingError)r rD 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)rA rB rC r r r r �MyStr� s rH zselect ?�abcN)r" r r
)r rH r r r �CheckStrSubclass� s z RegressionTests.CheckStrSubclassc C sr Gd d � d t j � } | d � } y | j � } | j d � Wn) t j k
rY Yn | j d � Yn Xd 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 r@ � s zPRegressionTests.CheckConnectionConstructorCallCheck.<locals>.Connection.__init__N)rA rB rC r@ r r r r �
Connection� s rL z:memory:z#should have raised ProgrammingErrorN)r rL r r' rF )r rL 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 � y | j � | j d � Wn) t j k
r� Yn | j d � Yn Xd
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 )rD r r r � s zBRegressionTests.CheckCursorRegistration.<locals>.Connection.cursorN)rA rB rC r r )rD r r rL � s rL c @ s e Z d Z d d � Z d S)z7RegressionTests.CheckCursorRegistration.<locals>.Cursorc S s t j j | | � d S)N)r rD r@ )r r r r r r@ � s z@RegressionTests.CheckCursorRegistration.<locals>.Cursor.__init__N)rA rB rC r@ r r r r rD � s rD z:memory:zcreate table foo(x)z insert into foo(x) values (?)r$ � r zselect x from fooz!should have raised InterfaceErrorN)r$ )rN )r )
r rL rD r r
r r rE r' ZInterfaceError)r rL r r r )rD 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
r2 )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 r3 �kwr r r �__call__ s z:RegressionTests.CheckSetDict.<locals>.NotHashable.__call__c S s
t � � d S)N)r: )r r r r �__hash__ s z:RegressionTests.CheckSetDict.<locals>.NotHashable.__hash__N)rA rB rC rT rU r r r r � NotHashable s rV N)r9 r: r Zcreate_functionZcreate_aggregateZset_authorizerZset_progress_handler)r rV �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.
� N)r9 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)NrY r )�a�br r r � collation_cb s z4RegressionTests.CheckCollation.<locals>.collation_cbu �)r9 r rF r Zcreate_collation)r r^ 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 QXd 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 (?)rY )rY )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
r9 rF r )r r r` 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� rN � � r i@� i@� N)r r r, r r
rE 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< �{ )r9 r: 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 r9 �
ValueErrorr r
)r r r r r r �CheckNullCharacterS s z"RegressionTests.CheckNullCharacterN) rA rB rC r r r r r r r# r( r) r+ r. r8 r; r? rG rJ rM rO rP rR rX r[ r_ rb rf rh rj r r r r r s6
r c C s" t j t d � } t j | f � S)NZCheck)�unittestZ makeSuiter Z TestSuite)Zregression_suiter r r �suite] s rl c C s t j � } | j t � � d S)N)rk ZTextTestRunnerZrunrl )Zrunnerr r r �testa s rm �__main__) r- rk Zsqlite3r ZTestCaser rl rm rA r r r r �<module> s � B