Your IP : 216.73.216.213
B
� f0 � @ s� d dl Z d dlZd dlmZ G dd� dej�Zdd� ZG dd� dej�Z G d d
� d
e j
�Z G d d � d e j
�Z G d
d� de j
�Z
G dd� de j
�ZG dd� de j
�ZG dd� de j
�Zdd� Zdd� Zedkr�e� dS )� N)�Sequencec @ s e Zd Zdd� ZdS )� MyConnectionc O s t jj| f|�|� d S )N)�sqlite�
Connection�__init__)�self�args�kwargs� r
�9/opt/alt/python37/lib64/python3.7/sqlite3/test/factory.pyr s zMyConnection.__init__N)�__name__�
__module__� __qualname__r r
r
r
r r s r c C s0 i }x&t | j�D ]\}}|| ||d < qW |S )Nr )� enumerateZ description)�cursor�row�d�idx�colr
r
r � dict_factory s r c @ s e Zd Zdd� ZdS )�MyCursorc O s t jj| f|�|� t| _d S )N)r �Cursorr r � row_factory)r r r r
r
r r ' s zMyCursor.__init__N)r r
r r r
r
r
r r & s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�ConnectionFactoryTestsc C s t jdtd�| _d S )Nz:memory:)�factory)r �connectr �con)r r
r
r �setUp, s z ConnectionFactoryTests.setUpc C s | j �� d S )N)r �close)r r
r
r �tearDown/ s z ConnectionFactoryTests.tearDownc C s | � | jt� d S )N)�assertIsInstancer r )r r
r
r �CheckIsInstance2 s z&ConnectionFactoryTests.CheckIsInstanceN)r r
r r r r! r
r
r
r r + s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�CursorFactoryTestsc C s t �d�| _d S )Nz:memory:)r r r )r r
r
r r 6 s zCursorFactoryTests.setUpc C s | j �� d S )N)r r )r r
r
r r 9 s zCursorFactoryTests.tearDownc C sR | j �� }| �|tj� | j �t�}| �|t� | j jdd� d�}| �|t� d S )Nc S s t | �S )N)r )r r
r
r �<lambda>A � z4CursorFactoryTests.CheckIsInstance.<locals>.<lambda>)r )r r r r r r )r �curr
r
r r! < s
z"CursorFactoryTests.CheckIsInstancec C sB | � t| jjd � | � t| jjdd� � | � t| jjdd� � d S )Nc S s d S )Nr
r
r
r
r r# H r$ z8CursorFactoryTests.CheckInvalidFactory.<locals>.<lambda>c S s d S )Nr
)r r
r
r r# J r$ )� assertRaises� TypeErrorr r )r r
r
r �CheckInvalidFactoryD s z&CursorFactoryTests.CheckInvalidFactoryN)r r
r r r r! r( r
r
r
r r" 5 s r" c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� RowFactoryTestsBackwardsCompatc C s t �d�| _d S )Nz:memory:)r r r )r r
r
r r M s z$RowFactoryTestsBackwardsCompat.setUpc C s8 | j jtd�}|�d� |�� }| �|t� |�� d S )N)r zselect 4+5 as foo)r r r �execute�fetchoner �dictr )r r% r r
r
r �CheckIsProducedByFactoryP s
z7RowFactoryTestsBackwardsCompat.CheckIsProducedByFactoryc C s | j �� d S )N)r r )r r
r
r r W s z'RowFactoryTestsBackwardsCompat.tearDownN)r r
r r r- r r
r
r
r r) L s r) c @ sl e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )�RowFactoryTestsc C s t �d�| _d S )Nz:memory:)r r r )r r
r
r r [ s zRowFactoryTests.setUpc C s, dd� | j _| j �d��� }| �|t� d S )Nc S s t |�S )N)�list)r% r r
r
r r# _ r$ z4RowFactoryTests.CheckCustomFactory.<locals>.<lambda>z select 1, 2)r r r* r+ r r/ )r r r
r
r �CheckCustomFactory^ s z"RowFactoryTests.CheckCustomFactoryc C st t j| j_| j�d��� }| �|t j� | �|d dd� | �|d dd� | �|d dd � | �|d
dd � | �|d dd
� | �|d dd� | �|d dd� | �|d dd� | �t �� |d W d Q R X | �t �� |d W d Q R X | �t �� |d W d Q R X | �t �� |d W d Q R X | �t �� |d W d Q R X | �t �� |dd W d Q R X d S )Nzselect 1 as a_1, 2 as bZa_1� z&by name: wrong result for column 'a_1'�b� z$by name: wrong result for column 'b'ZA_1z&by name: wrong result for column 'A_1'�Bz$by name: wrong result for column 'B'r z#by index: wrong result for column 0z#by index: wrong result for column 1���z$by index: wrong result for column -1���z$by index: wrong result for column -2�cza_za1���i� )
r �Rowr r r* r+ r � assertEqualr& �
IndexError)r r r
r
r �CheckSqliteRowIndexc s.
z#RowFactoryTests.CheckSqliteRowIndexc C sj t j| j_| j�d��� }| �|d d� | �t�� |d W d Q R X | �t�� |d W d Q R X d S )Nu select 1 as ÿ� ÿr1 u Ÿ� ß) r r9 r r r* r+ r: r& r; )r r r
r
r �CheckSqliteRowIndexUnicode� s
z*RowFactoryTests.CheckSqliteRowIndexUnicodec C s� t j| j_| j�d��� }| �|dd� d� | �|dd� d� | �|dd� d� | �|dd� d� | �|dd � d� | �|d d� d � | �|d
d � d � | �|d
d � d
� | �|ddd� d� | �|ddd
� d� d S )Nzselect 1, 2, 3, 4r r
r1 )r1 � )r3 r@ )r3 r@ � )r1 r3 r@ r6 r5 )r@ )r@ rA rA r3 )r1 r@ )rA r3 )r r9 r r r* r+ r: )r r r
r
r �CheckSqliteRowSlice� s
z#RowFactoryTests.CheckSqliteRowSlicec C s, t j| j_| j�d��� }x |D ]}q W dS )z$Checks if the row object is iterablezselect 1 as a, 2 as bN)r r9 r r r* r+ )r r r r
r
r �CheckSqliteRowIter� s
z"RowFactoryTests.CheckSqliteRowIterc C s> t j| j_| j�d��� }t|�}| �||d |d f� dS )z4Checks if the row object can be converted to a tuplezselect 1 as a, 2 as b�ar2 N)r r9 r r r* r+ �tupler: )r r �tr
r
r �CheckSqliteRowAsTuple� s
z%RowFactoryTests.CheckSqliteRowAsTuplec C sN t j| j_| j�d��� }t|�}| �|d |d � | �|d |d � dS )zCChecks if the row object can be correctly converted to a dictionaryzselect 1 as a, 2 as brD r2 N)r r9 r r r* r+ r, r: )r r r r
r
r �CheckSqliteRowAsDict� s
z$RowFactoryTests.CheckSqliteRowAsDictc C s� t j| j_| j�d��� }| j�d��� }| j�d��� }| j�d��� }| j�d��� }| �||k� | �||k� | �||k� | �||k� | �||k� | �|t� k� | �||k� | �||k� | �||k� | �||k� | �||k� | �|t� k� | � t
�� ||k W dQ R X | � t
�� ||k W dQ R X | � t
�� ||k W dQ R X | � t
�� ||k W dQ R X | � t |�t |�� dS )z6Checks if the row object compares and hashes correctlyzselect 1 as a, 2 as bzselect 1 as a, 3 as bzselect 1 as b, 2 as azselect 2 as b, 1 as aN)
r r9 r r r* r+ �
assertTrueZ assertFalse�objectr& r' r: �hash)r Zrow_1Zrow_2Zrow_3Zrow_4Zrow_5r
r
r �CheckSqliteRowHashCmp� s6
z%RowFactoryTests.CheckSqliteRowHashCmpc C sN t j| j_| j�d��� }t|�}| �tt |��tt |��� | �
|t � dS )z2 Checks if the row object can act like a sequence zselect 1 as a, 2 as bN) r r9 r r r* r+ rE r: r/ �reversedr r )r r Zas_tupler
r
r �CheckSqliteRowAsSequence� s
z(RowFactoryTests.CheckSqliteRowAsSequencec C sD G dd� dt �}tj| j_| �t| jj|� | �ttj|� d� d S )Nc @ s e Zd ZejZdS )z8RowFactoryTests.CheckFakeCursorClass.<locals>.FakeCursorN)r r
r r r � __class__r
r
r
r �
FakeCursor� s rP r
)�strr r9 r r r& r' r )r rP r
r
r �CheckFakeCursorClass� s
z$RowFactoryTests.CheckFakeCursorClassc C s | j �� d S )N)r r )r r
r
r r � s zRowFactoryTests.tearDownN)r r
r r r0 r< r? rB rC rG rH rL rN rR r r
r
r
r r. Z s "
r. c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�TextFactoryTestsc C s t �d�| _d S )Nz:memory:)r r r )r r
r
r r � s zTextFactoryTests.setUpc C s2 d}| j �d|f��� }| �t|d �td� d S )Nu Österreichzselect ?r z type of row[0] must be unicode)r r* r+ r: �typerQ )r �austriar r
r
r � CheckUnicode� s z TextFactoryTests.CheckUnicodec C sR t | j_d}| j�d|f��� }| �t|d �t d� | �|d |�d�d� d S )Nu Österreichzselect ?r z type of row[0] must be byteszutf-8z(column must equal original data in UTF-8)�bytesr � text_factoryr* r+ r: rT �encode)r rU r r
r
r � CheckString� s
z TextFactoryTests.CheckStringc C sT dd� | j _d}| j �d|f��� }| �t|d �td� | �|d �d�d� d S ) Nc S s t | dd�S )Nzutf-8�ignore)rQ )�xr
r
r r# � r$ z.TextFactoryTests.CheckCustom.<locals>.<lambda>u Österreichzselect ?r z type of row[0] must be unicodeZreichz!column must contain original data) r rX r* r+ r: rT rQ rI �endswith)r rU r r
r
r � CheckCustom� s
z TextFactoryTests.CheckCustomc C sj t j| j_d}d}| j�d|f��� }| j�d|f��� }| �t|d �td� | �t|d �td� d S )Nu ÖsterreichZ
Deutchlandzselect ?r z!type of non-ASCII row must be strz"type of ASCII-only row must be str) r ZOptimizedUnicoder rX r* r+ r: rT rQ )r rU ZgermanyZa_rowZd_rowr
r
r �CheckOptimizedUnicode s
z&TextFactoryTests.CheckOptimizedUnicodec C s | j �� d S )N)r r )r r
r
r r s zTextFactoryTests.tearDownN) r r
r r rV rZ r^ r_ r r
r
r
r rS � s rS c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�%TextFactoryTestsWithEmbeddedZeroBytesc C s* t �d�| _| j�d� | j�dd� d S )Nz:memory:z create table test (value text)z#insert into test (value) values (?))za b)r r r r* )r r
r
r r s z+TextFactoryTestsWithEmbeddedZeroBytes.setUpc C s8 | j �d��� }| �t|d �t� | �|d d� d S )Nzselect value from testr za b)r r* r+ �assertIsrT rQ r: )r r r
r
r rZ s z1TextFactoryTestsWithEmbeddedZeroBytes.CheckStringc C s@ t | j_| j�d��� }| �t|d �t � | �|d d� d S )Nzselect value from testr s a b)rW r rX r* r+ ra rT r: )r r r
r
r �
CheckBytes s z0TextFactoryTestsWithEmbeddedZeroBytes.CheckBytesc C s@ t | j_| j�d��� }| �t|d �t � | �|d d� d S )Nzselect value from testr s a b)� bytearrayr rX r* r+ ra rT r: )r r r
r
r �CheckBytearray s z4TextFactoryTestsWithEmbeddedZeroBytes.CheckBytearrayc C sD dd� | j _| j �d��� }| �t|d �t� | �|d d� d S )Nc S s | S )Nr
)r\ r
r
r r# ( r$ zCTextFactoryTestsWithEmbeddedZeroBytes.CheckCustom.<locals>.<lambda>zselect value from testr s a b)r rX r* r+ ra rT rW r: )r r r
r
r r^ & s z1TextFactoryTestsWithEmbeddedZeroBytes.CheckCustomc C s | j �� d S )N)r r )r r
r
r r - s z.TextFactoryTestsWithEmbeddedZeroBytes.tearDownN) r r
r r rZ rb rd r^ r r
r
r
r r` s r` c C s^ t �td�} t �td�}t �td�}t �td�}t �td�}t �td�}t �| |||||f�S )NZCheck) �unittestZ makeSuiter r" r) r. rS r` Z TestSuite)Zconnection_suiteZ cursor_suiteZrow_suite_compatZ row_suiteZ
text_suiteZtext_zero_bytes_suiter
r
r �suite0 s rf c C s t �� } | �t� � d S )N)re ZTextTestRunner�runrf )Zrunnerr
r
r �test9 s rh �__main__)re Zsqlite3r Zcollections.abcr r r r r r ZTestCaser r" r) r. rS r` rf rh r r
r
r
r �<module> s
%"