Your IP : 216.73.216.213
���] 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/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
� assertEqual� 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)Nzcreate table foo(bar)zdrop table foo)r r
)r r r r �-CheckWorkaroundForBuggySqliteTransferBindingsf s z=RegressionTests.CheckWorkaroundForBuggySqliteTransferBindingsc C s | j j d � d S)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) Nzselect * 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))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# � exception)r �cm�msgr r r �CheckErrorMsgDecodeError� s
&z(RegressionTests.CheckErrorMsgDecodeErrorc C s | j t t j i d � d S)N)r1 � 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__r; r= r )r r r � CustomStr� s rA z:memory:r+ �DEFERRED� IMMEDIATE� EXCLUSIVE�level� � �abczIMMEDIATE EXCLUSIVE� é)r+ rB rC rD ) r# r r r< �subTest�lower�
capitalizer9 �
ValueErrorr1 r )r rA rE 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)Nc @ 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@ rQ r r r r �Cursor� s rR z:memory:z
select 4+5)r rR r r1 �ProgrammingErrorr
�fetchall)r rR r r r r r � CheckCursorConstructorCallCheck� s
z/RegressionTests.CheckCursorConstructorCallCheckc C s6 Gd d � d t � } | j j d | d � f � d S)Nc @ s e Z d Z d S)z/RegressionTests.CheckStrSubclass.<locals>.MyStrN)r> r? r@ r r r r �MyStr� s rV zselect ?rH )r# r r
)r rV r r r �CheckStrSubclass� s z RegressionTests.CheckStrSubclassc C sO Gd d � d t j � } | d � } | j t j � � | j � } Wd QRXd S)Nc @ s e Z d Z d d � Z d S)zGRegressionTests.CheckConnectionConstructorCallCheck.<locals>.Connectionc S s d S)Nr )r �namer r r rQ � s zPRegressionTests.CheckConnectionConstructorCallCheck.<locals>.Connection.__init__N)r> r? r@ rQ r r r r �
Connection� s rY z:memory:)r rY r1 rS r )r rY 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)Nc s" e Z d Z � f d d � Z d S)z;RegressionTests.CheckCursorRegistration.<locals>.Connectionc s
� | � S)Nr )r )rR r r r � s zBRegressionTests.CheckCursorRegistration.<locals>.Connection.cursorN)r> r? r@ r r )rR r r rY � s rY c @ s e Z d Z d d � Z d S)z7RegressionTests.CheckCursorRegistration.<locals>.Cursorc S s t j j | | � d S)N)r rR rQ )r r r r r rQ � s z@RegressionTests.CheckCursorRegistration.<locals>.Cursor.__init__N)r> r? r@ rQ r r r r rR � s rR z:memory:zcreate table foo(x)z insert into foo(x) values (?)r% � r zselect x from foo)r% )r[ )r )
r rY rR r r
r r r1 ZInterfaceErrorrT )r rY r r r )rR r �CheckCursorRegistration� s
z'RegressionTests.CheckCursorRegistrationc C s t j d d d �} d S)Nz:memory:r< )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)Nzcreate table foo(bar)z insert into foo(bar) values (5)zpragma page_size)r r r
r4 )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)Nc @ 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)r9 )r r r r �__hash__ s z:RegressionTests.CheckSetDict.<locals>.NotHashable.__hash__N)r> r? r@ rb rc r r r r � NotHashable s rd )r1 r9 r Zcreate_functionZcreate_aggregateZset_authorizerZset_progress_handler)r rd �varr r r � CheckSetDict� s z RegressionTests.CheckSetDictc C s | j t j | j d � d S)NrF )r1 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)NrF r )�a�br r r � collation_cb s z4RegressionTests.CheckCollation.<locals>.collation_cbu �)r1 r rS r Zcreate_collation)r rk 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) Nz:memory:zcreate table a (bar)zcreate table b (baz)c 3 s � j d d � d Vd S)Nz insert into a (bar) values (?)rF )rF )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>)r r r r
r1 rS r )r r rm 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)Nz: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� r[ � � r i@� i@� )r r r- r r
rT 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< �{ )r1 r9 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 r1 rM 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)
Nz: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 (?)rF r& r% z'should have returned exactly three rows) r r Z
executescriptr r< � enumerater
rJ 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/ r8 r: rP rU rW rZ r\ r] r_ rf rh rl ro rs ru rv rx r r r r r s8
'
r c C s" t j t d � } t j | f � S)NZCheck)ry Z makeSuiter Z TestSuite)Zregression_suiter r r �suite| s rz c C s t j � } | j t � � d S)N)ry ZTextTestRunner�runrz )Zrunnerr r r �test� s r| �__main__) r. ry Zsqlite3r ZTestCaser rz r| r> r r r r �<module> s � a