EVOLUTION-NINJA
Edit File: regression.cpython-38.pyc
U �i�_�: � @ sh d dl Z d dlZd dlZd dlZd dlZd dlmZ G dd� dej�Z dd� Z dd� Ze d krde� dS ) � N)�supportc @ s e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z e � e j dk d�dd� �Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Z d/d0� Z d1d2� Z d3d4� Z d5d6� Z d7d8� Z!d9d:� Z"d;d<� Z#d=d>� Z$d?S )@�RegressionTestsc C s t �d�| _d S )N�:memory:)�sqlite�connect�con��self� r �6/opt/python38/lib/python3.8/sqlite3/test/regression.py�setUp s zRegressionTests.setUpc C s | j �� d S �N)r �closer r r r �tearDown# s zRegressionTests.tearDownc C s | j �� }|�d� d S )Nzpragma user_version)r �cursor�execute�r �curr r r �CheckPragmaUserVersion& s z&RegressionTests.CheckPragmaUserVersionc C s@ t jdt jd�}z| j�� }|�d� W 5 |�� |�� X d S )Nr �Z detect_typeszpragma schema_version)r r ZPARSE_COLNAMESr r r r �r r r r r r �CheckPragmaSchemaVersion+ s z(RegressionTests.CheckPragmaSchemaVersionc s� t jddd�� � fdd�td�D �}|d �d� td�D ]"}|d �d d d� td�D �� q:td�D ]}|| �d | d � qf� �� d S ) Nr � )Zcached_statementsc s g | ] }� � � �qS r )r ��.0�x�r r r � <listcomp>: s z7RegressionTests.CheckStatementReset.<locals>.<listcomp>r zcreate table test(x)� z insert into test(x) values (?)c S s g | ] }|f�qS r r r r r r r = s � zselect x from test)r r �ranger � executemany�rollback)r �cursors�ir r r �CheckStatementReset5 s z#RegressionTests.CheckStatementResetc C sN | j �� }|�d� | �|jd d d� |�d� | �|jd d d� d S )Nz select 1 as "foo bar [datetime]"r zfoo bar [datetime]zselect 1 as "foo baz"zfoo baz)r r r � assertEqual� descriptionr r r r �CheckColumnNameWithSpacesD s z)RegressionTests.CheckColumnNameWithSpacesc C sL t �d�}g }td�D ](}|�� }|�|� |�dt|� � q|�� d S )Nr �i zselect 1 x union select )r r r r �appendr �strr )r r r# r$ r r r r �#CheckStatementFinalizationOnCloseDbL s z3RegressionTests.CheckStatementFinalizationOnCloseDb)� � r. zneeds sqlite 3.2.2 or newerc C s� t �d�}|�d� |�d� z|�d� W n t jk rB Y nX |�d� z |�� W n t jk rz | �d� Y nX d S )Nr 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 �CheckOnConflictRollbackZ s z'RegressionTests.CheckOnConflictRollbackc C s( | j �d� | j �d� | j �d� dS )zm pysqlite would crash with older SQLite versions unless a workaround is implemented. �create table foo(bar)�drop table fooN�r r r r r r �-CheckWorkaroundForBuggySqliteTransferBindingsi s z=RegressionTests.CheckWorkaroundForBuggySqliteTransferBindingsc C s | j �d� dS )z pysqlite used to segfault with SQLite versions 3.5.x. These return NULL for "no-operation" statements � Nr5 r r r r �CheckEmptyStatementr s z#RegressionTests.CheckEmptyStatementc C sh d}t jdt jd�}|�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 foor r z create table foo(bar timestamp)z insert into foo(bar) values (?)r4 z create table foo(bar integer)� insert into foo(bar) values (5)N)r r �PARSE_DECLTYPESr �datetimeZnow)r ZSELECTr r r r �CheckTypeMapUsagey s z!RegressionTests.CheckTypeMapUsagec C sP | � tj��$}| j�dttdg��f��� W 5 Q R X d}| �|t |j �� d S )Nz"select 'xxx' || ? || 'yyy' colname� z9Could not decode to UTF-8 column 'colname' with text 'xxx) � assertRaisesr r0 r r �bytes� bytearray�fetchoneZassertInr+ � exception)r �cm�msgr r r �CheckErrorMsgDecodeError� s �z(RegressionTests.CheckErrorMsgDecodeErrorc C s | � ttji d� dS )z! See issue 3312. N)r> � TypeErrorr Zregister_adapterr r r r �CheckRegisterAdapter� s z$RegressionTests.CheckRegisterAdapterc s� G � fdd�dt �}t�d�� d � _dD ]@}| j|d��* |� _|�� � _|�� � _||�� _W 5 Q R X q(d � _d� _dtfdtfd tfd tfd tfg}|D ]J\}}| j|d��0 | � |�� |� _W 5 Q R X | � � jd� W 5 Q R X q�d S ) Nc s e Zd Zdd� Z� fdd�ZdS )z9RegressionTests.CheckSetIsolationLevel.<locals>.CustomStrc S s d S r r r r r r �upper� s z?RegressionTests.CheckSetIsolationLevel.<locals>.CustomStr.upperc s d� _ d S )Nr7 ��isolation_levelr r r r �__del__� s zARegressionTests.CheckSetIsolationLevel.<locals>.CustomStr.__del__N)�__name__� __module__� __qualname__rH rK r r r r � CustomStr� s rO r )r7 �DEFERREDZ IMMEDIATEZ EXCLUSIVE)�levelrP � � �abczIMMEDIATE EXCLUSIVE� é) r+ r r rJ �subTest�lower� capitalizerF � ValueErrorr> r& )r rO rQ Zpairs�value�excr r r �CheckSetIsolationLevel� s, � z&RegressionTests.CheckSetIsolationLevelc C sp G dd� dt j�}t �d�}||�}| �t j�� |�d��� W 5 Q R X | �t jd�� |�� W 5 Q R X dS )zd Verifies that cursor methods check whether base class __init__ was called. c @ s e Zd Zdd� ZdS )z?RegressionTests.CheckCursorConstructorCallCheck.<locals>.Cursorc S s d S r r r1 r r r �__init__� s zHRegressionTests.CheckCursorConstructorCallCheck.<locals>.Cursor.__init__N�rL rM rN r] r r r r �Cursor� s r_ r z select 4+5z$^Base Cursor\.__init__ not called\.$N) r r_ r r> �ProgrammingErrorr �fetchallZassertRaisesRegexr )r r_ r r r r r � CheckCursorConstructorCallCheck� s �z/RegressionTests.CheckCursorConstructorCallCheckc C s( G dd� 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 e Zd ZdS )z/RegressionTests.CheckStrSubclass.<locals>.MyStrN)rL rM rN r r r r �MyStr� s rc zselect ?rT N)r+ r r )r rc r r r �CheckStrSubclass� s z RegressionTests.CheckStrSubclassc C s> G dd� dt j�}|d�}| �t j�� |�� }W 5 Q R X dS )zh Verifies that connection methods check whether base class __init__ was called. c @ s e Zd Zdd� ZdS )zGRegressionTests.CheckConnectionConstructorCallCheck.<locals>.Connectionc S s d S r r )r �namer r r r] � s zPRegressionTests.CheckConnectionConstructorCallCheck.<locals>.Connection.__init__Nr^ r r r r � Connection� s rf r N)r rf r> r` r �r rf r r r r r �#CheckConnectionConstructorCallCheck� s z3RegressionTests.CheckConnectionConstructorCallCheckc s� G � fdd�dt j�}G dd� dt j�� |d�}|�� }|�d� |�ddd d g� |�d � |�� | �t j�� |� � W 5 Q R X d S ) z� Verifies that subclassed cursor classes are correctly registered with the connection object, too. (fetch-across-rollback problem) c s e Zd Z� fdd�ZdS )z;RegressionTests.CheckCursorRegistration.<locals>.Connectionc s � | �S r r r �r_ r r r � s zBRegressionTests.CheckCursorRegistration.<locals>.Connection.cursorN)rL rM rN r r ri r r rf � s rf c @ s e Zd Zdd� ZdS )z7RegressionTests.CheckCursorRegistration.<locals>.Cursorc S s t j�| |� d S r )r r_ r] r1 r r r r] � s z@RegressionTests.CheckCursorRegistration.<locals>.Cursor.__init__Nr^ r r r r r_ � s r_ r zcreate table foo(x)z insert into foo(x) values (?))r- )� )r zselect x from fooN) r rf r_ r r r! r" r> ZInterfaceErrorra rg r ri r �CheckCursorRegistration� s z'RegressionTests.CheckCursorRegistrationc C s t jddd�}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. r NrI )r r r1 r r r �CheckAutoCommit� s z RegressionTests.CheckAutoCommitc C s4 | 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. r3 r9 zpragma page_sizeN)r r r rA )r r �rowr r r �CheckPragmaAutocommit� s z%RegressionTests.CheckPragmaAutocommitc C s | � tj| jd� dS )z} Call a connection with a non-string SQL request: check error handling of the statement constructor. rR N)r> r �Warningr r r r r �CheckConnectionCall s z#RegressionTests.CheckConnectionCallc C s" dd� }| � tj| jjd|� d S )Nc S s dS )NrR r )�a�br r r � collation_cb s z4RegressionTests.CheckCollation.<locals>.collation_cbu �)r> r r` r Zcreate_collation)r rs r r r �CheckCollation s �z RegressionTests.CheckCollationc sf t �d�}|�� � � �d� � �d� � fdd�}| �t j�� � �ddd� |� D �� W 5 Q R X d 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. r zcreate table a (bar)zcreate table b (baz)c 3 s � � dd� dV d S )Nz insert into a (bar) values (?))rR rR )r r �r r r �foo s z4RegressionTests.CheckRecursiveCursorUse.<locals>.fooz insert into b (baz) values (?)c s s | ] }|fV qd S r r )r r$ r r r � <genexpr>% s z:RegressionTests.CheckRecursiveCursorUse.<locals>.<genexpr>N)r r r r r> r` r! )r r rv r ru r �CheckRecursiveCursorUse s �z'RegressionTests.CheckRecursiveCursorUsec C s� t jdt jd�}|�� }|�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". r 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 �qS )r r r r r r r : s zKRegressionTests.CheckConvertTimestampMicrosecondPadding.<locals>.<listcomp>i� rj � � r i@� i@� N)r r r: r r ra r& r; )r r r �valuesr r r �'CheckConvertTimestampMicrosecondPadding' s �z7RegressionTests.CheckConvertTimestampMicrosecondPaddingc C s | j ttjddd� d S )Nr �{ rI )r> rF r r r r r r � CheckInvalidIsolationLevelTypeA s �z.RegressionTests.CheckInvalidIsolationLevelTypec C sR t �d�}| �t|d� | �t|d� |�� }| �t|jd� | �t|jd� d S )Nr z select 1z select 1 z select 2z select 2 )r r r> rY r r r r r r �CheckNullCharacterG s z"RegressionTests.CheckNullCharacterc C s� t �d�}|�d� | �|jd� d}t|�d��D ]�\}}| j||d��p |�d|f� |�� |dkr|| �|d d� n2|dkr�| �|d d� n|d kr�| �|d d � |d7 }W 5 Q R X q4| �|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. r z� create table t(c); create table t2(c); insert into t values(0); insert into t values(1); insert into t values(2); r7 r zselect c from t)r$ rm z insert into t2(c) values (?)rR r. r- z'should have returned exactly three rowsN) r r Z executescriptr& rJ � enumerater rV r/ )r r Zcounterr$ rm r r r �CheckCommitCursorResetP s z&RegressionTests.CheckCommitCursorResetc C sB dd� }t �d�}t �|�}t�||�}|�|� ~~t�� dS )zm The interpreter shouldn't crash in case Cursor.__init__() is called more than once. c W s d S r r )�argsr r r �callbackt s z/RegressionTests.CheckBpo31770.<locals>.callbackr N)r r r_ �weakref�refr] r � gc_collect)r r� r r r� r r r � CheckBpo31770o s z RegressionTests.CheckBpo31770c C s | � t�� | j`W 5 Q R X d S r )r> �AttributeErrorr rJ r r r r � CheckDelIsolation_levelSegfault s z/RegressionTests.CheckDelIsolation_levelSegfaultc C sf G dd� d�}| j jtj| j jdd�| j jfD ]2}|� }||j� ||j� | j �d� |d � q.d S )Nc @ s e Zd Zdd� ZdS )z.RegressionTests.CheckBpo37347.<locals>.Printerc W s t jS r )r Z SQLITE_OK)r r� r r r �log� s z2RegressionTests.CheckBpo37347.<locals>.Printer.logN)rL rM rN r� r r r r �Printer� s r� rR )�nzselect 1)r Zset_trace_callback� functools�partialZset_progress_handlerZset_authorizerr� r )r r� �methodZprinter_instancer r r � CheckBpo37347� s � z RegressionTests.CheckBpo37347N)%rL rM rN r r r r r% r( r, �unittestZskipIfr Zsqlite_version_infor2 r6 r8 r< rE rG r\ rb rd rh rk rl rn rp rt rx r| r~ r r� r� r� r� r r r r r s> r c C s t �td�} t �| f�S )NZCheck)r� Z makeSuiter Z TestSuite)Zregression_suiter r r �suite� s �r� c C s t �� } | �t� � d S r )r� ZTextTestRunner�runr� )Zrunnerr r r �test� s r� �__main__) r; r� Zsqlite3r r� r� r� r ZTestCaser r� rL r r r r �<module> s v