Your IP : 216.73.216.213
3
� fV, � @ 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/python36/lib64/python3.6/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 j� d S )N)r �close)r r
r
r �tearDown/ s z ConnectionFactoryTests.tearDownc C s | j | 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 jd�| _d S )Nz:memory:)r r r )r r
r
r r 6 s zCursorFactoryTests.setUpc C s | j j� d S )N)r r )r r
r
r r 9 s zCursorFactoryTests.tearDownc C sR | j j� }| j|tj� | j jt�}| j|t� | j jdd� d�}| j|t� d S )Nc S s t | �S )N)r )r r
r
r �<lambda>A s z4CursorFactoryTests.CheckIsInstance.<locals>.<lambda>)r )r r r r r r )r �curr
r
r r! < s
z"CursorFactoryTests.CheckIsInstancec C sB | j t| jjd � | j t| jjdd� � | j t| jjdd� � d S )Nc S s d S )Nr
r
r
r
r r# H s z8CursorFactoryTests.CheckInvalidFactory.<locals>.<lambda>c S s d S )Nr
)r r
r
r r# J s )� 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 jd�| _d S )Nz:memory:)r r r )r r
r
r r M s z$RowFactoryTestsBackwardsCompat.setUpc C s8 | j jtd�}|jd� |j� }| j|t� |j� 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 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 @ sd e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dS )�RowFactoryTestsc C s t jd�| _d S )Nz:memory:)r r r )r r
r
r r [ s zRowFactoryTests.setUpc C s, dd� | j _| j jd�j� }| j|t� d S )Nc S s t |�S )N)�list)r$ r r
r
r r# _ s 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 sL t j| j_| jjd�j� }| j|t j� |d |d }}| j|dd� | j|dd� |d |d }}| j|dd � | j|dd
� | j|d dd � | j|d dd
� | j|d dd� | j|d dd� | jt �� |d W d Q R X | jt �� |d W d Q R X | jt �� |d W d Q R X | jt �� |dd W d Q R X d S )Nzselect 1 as a, 2 as b�a�b� z$by name: wrong result for column 'a'� �A�Bz$by name: wrong result for column 'A'z$by name: wrong result for column 'B'r z#by index: wrong result for column 0z#by index: wrong result for column 1z$by index: wrong result for column -1z$by index: wrong result for column -2�c� i� ���������)
r �Rowr r r) r* r � assertEqualr% �
IndexError)r r Zcol1Zcol2r
r
r �CheckSqliteRowIndexc s*
z#RowFactoryTests.CheckSqliteRowIndexc C s� t j| j_| jjd�j� }| j|dd� f � | j|dd� d� | j|dd� d� | j|dd� f � | j|dd � d � | j|d d� d
� | j|d d � d
� | j|dd � d� | j|ddd� d� | j|ddd� d� d S )Nzselect 1, 2, 3, 4r r2 r7 r3 � )r2 )r3 r7 )r3 r7 r? )r2 r3 r7 r9 r8 )r7 r9 )r7 r? )r2 r7 r9 )r? r3 )r r; r r r) r* r< )r r r
r
r �CheckSqliteRowSlice~ s
z#RowFactoryTests.CheckSqliteRowSlicec C s, t j| j_| jjd�j� }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_| jjd�j� }t|�}| j||d |d f� dS )z4Checks if the row object can be converted to a tuplezselect 1 as a, 2 as br0 r1 N)r r; r r r) r* �tupler< )r r �tr
r
r �CheckSqliteRowAsTuple� s
z%RowFactoryTests.CheckSqliteRowAsTuplec C sN t j| j_| jjd�j� }t|�}| j|d |d � | j|d |d � dS )zCChecks if the row object can be correctly converted to a dictionaryzselect 1 as a, 2 as br0 r1 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_| jjd�j� }| jjd�j� }| jjd�j� }| j||� | j||� | j||k� | j||k� | j||k� | j||k� | j||� | jt |�t |�� | j
||� | j
t |�t |�� dS )z6Checks if the row object compares and hashes correctlyzselect 1 as a, 2 as bzselect 1 as a, 3 as bN) r r; r r r) r* r< �
assertTrueZ assertFalse�hashZassertNotEqual)r Zrow_1Zrow_2Zrow_3r
r
r �CheckSqliteRowHashCmp� s
z%RowFactoryTests.CheckSqliteRowHashCmpc C sN t j| j_| jjd�j� }t|�}| jtt |��tt |��� | j
|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* rB r< r. �reversedr r )r r Zas_tupler
r
r �CheckSqliteRowAsSequence� s
z(RowFactoryTests.CheckSqliteRowAsSequencec C sD G dd� dt �}tj| j_| jt| jj|� | jttj|� f � d S )Nc @ s e Zd ZejZdS )z8RowFactoryTests.CheckFakeCursorClass.<locals>.FakeCursorN)r r
r r r � __class__r
r
r
r �
FakeCursor� s rL )�strr r; r r r% r& r )r rL r
r
r �CheckFakeCursorClass� s
z$RowFactoryTests.CheckFakeCursorClassc C s | j j� d S )N)r r )r r
r
r r � s zRowFactoryTests.tearDownN)r r
r r r/ r> r@ rA rD rE rH rJ rN 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 jd�| _d S )Nz:memory:)r r r )r r
r
r r � s zTextFactoryTests.setUpc C s2 d}| j jd|f�j� }| jt|d �td� d S )Nu Österreichzselect ?r z type of row[0] must be unicode)r r) r* r< �typerM )r �austriar r
r
r � CheckUnicode� s z TextFactoryTests.CheckUnicodec C sR t | j_d}| jjd|f�j� }| jt|d �t d� | j|d |jd�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< rP �encode)r rQ r r
r
r � CheckString� s
z TextFactoryTests.CheckStringc C sT dd� | j _d}| j jd|f�j� }| jt|d �td� | j|d jd�d� d S ) Nc S s t | dd�S )Nzutf-8�ignore)rM )�xr
r
r r# � s z.TextFactoryTests.CheckCustom.<locals>.<lambda>u Österreichzselect ?r z type of row[0] must be unicodeZreichz!column must contain original data) r rT r) r* r< rP rM rF �endswith)r rQ r r
r
r � CheckCustom� s
z TextFactoryTests.CheckCustomc C sj t j| j_d}d}| jjd|f�j� }| jjd|f�j� }| jt|d �td� | jt|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 rT r) r* r< rP rM )r rQ ZgermanyZa_rowZd_rowr
r
r �CheckOptimizedUnicode� s
z&TextFactoryTests.CheckOptimizedUnicodec C s | j j� d S )N)r r )r r
r
r r � s zTextFactoryTests.tearDownN) r r
r r rR rV rZ r[ r r
r
r
r rO � s rO c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�%TextFactoryTestsWithEmbeddedZeroBytesc C s* t jd�| _| jjd� | jjdd� d S )Nz:memory:z create table test (value text)z#insert into test (value) values (?)�a b)r] )r r r r) )r r
r
r r � s z+TextFactoryTestsWithEmbeddedZeroBytes.setUpc C s8 | j jd�j� }| jt|d �t� | j|d d� d S )Nzselect value from testr za b)r r) r* �assertIsrP rM r< )r r r
r
r rV � s z1TextFactoryTestsWithEmbeddedZeroBytes.CheckStringc C s@ t | j_| jjd�j� }| jt|d �t � | j|d d� d S )Nzselect value from testr s a b)rS r rT r) r* r^ rP r< )r r r
r
r �
CheckBytes s z0TextFactoryTestsWithEmbeddedZeroBytes.CheckBytesc C s@ t | j_| jjd�j� }| jt|d �t � | j|d d� d S )Nzselect value from testr s a b)� bytearrayr rT r) r* r^ rP r< )r r r
r
r �CheckBytearray s z4TextFactoryTestsWithEmbeddedZeroBytes.CheckBytearrayc C sD dd� | j _| j jd�j� }| jt|d �t� | j|d d� d S )Nc S s | S )Nr
)rX r
r
r r# s zCTextFactoryTestsWithEmbeddedZeroBytes.CheckCustom.<locals>.<lambda>zselect value from testr s a b)r rT r) r* r^ rP rS r< )r r r
r
r rZ
s z1TextFactoryTestsWithEmbeddedZeroBytes.CheckCustomc C s | j j� d S )N)r r )r r
r
r r s z.TextFactoryTestsWithEmbeddedZeroBytes.tearDownN) r r
r r rV r_ ra rZ r r
r
r
r r\ � s r\ c C s^ t jtd�} t jtd�}t jtd�}t jtd�}t jtd�}t jtd�}t j| |||||f�S )NZCheck) �unittestZ makeSuiter r" r( r- rO r\ Z TestSuite)Zconnection_suiteZ cursor_suiteZrow_suite_compatZ row_suiteZ
text_suiteZtext_zero_bytes_suiter
r
r �suite s rc c C s t j� } | jt� � d S )N)rb ZTextTestRunnerZrunrc )Zrunnerr
r
r �test s rd �__main__)rb Zsqlite3r �collections.abcr r r r r r ZTestCaser r" r( r- rO r\ rc rd r r
r
r
r �<module> s
v%"