File: //usr/local/lib/python3.7/sqlite3/test/__pycache__/factory.cpython-37.opt-1.pyc
B
��g0 � @ 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
�0/usr/local/lib/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 )� enumerate� 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! r0 )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 ÿ� ÿr2 u Ÿ� ß) r r: 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
r2 )r2 � )r4 rA )r4 rA � )r2 r4 rA r7 r6 )rA )rA rB rB r4 )r2 rA )rB r4 )r r: 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 r: 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�ar3 N)r r: 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 brE r3 N)r r: 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 r: 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 r: r r r+ r, rF r; r0 �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 rQ r
)�strr r: r r r' r( r )r rQ 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 r1 r= r@ rC rD rH rI rM rO rS 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; �typerR )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; rU �encode)r rV 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)rR )�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 rY r+ r, r; rU rR rJ �endswith)r rV 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 rY r+ r, r; rU rR )r rV 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 rW r[ r_ r` r r
r
r
r rT � s rT 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, �assertIsrU rR r; )r r r
r
r r[ s z1TextFactoryTestsWithEmbeddedZeroBytes.CheckStringc C s@ t | j_| j�d��� }| �t|d �t � | �|d d� d S )Nzselect value from testr s a b)rX r rY r+ r, rb rU 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 rY r+ r, rb rU 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 rY r+ r, rb rU rX 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 r[ rc re r_ r r
r
r
r ra s ra c C s^ t �td�} t �td�}t �td�}t �td�}t �td�}t �td�}t �| |||||f�S )NZCheck) �unittestZ makeSuiter r# r* r/ rT ra Z TestSuite)Zconnection_suiteZ cursor_suiteZrow_suite_compatZ row_suiteZ
text_suiteZtext_zero_bytes_suiter
r
r �suite0 s rg c C s t �� } | �t� � d S )N)rf ZTextTestRunner�runrg )Zrunnerr
r
r �test9 s ri �__main__)rf Zsqlite3r Zcollections.abcr r r r r r ZTestCaser r# r* r/ rT ra rg ri r r
r
r
r �<module> s
%"