f
��_0 c @ 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 @ � e Zd Zdd� ZdS )� MyConnectionc O s t jj| g|�R i |�� d S �N)�sqlite�
Connection�__init__��selfZargsZkwargs� r
�-/usr/lib64/python3.10/sqlite3/test/factory.pyr � zMyConnection.__init__N��__name__�
__module__� __qualname__r r
r
r
r r � r c C s, i }t | j�D ]\}}|| ||d < q|S )Nr )Z enumerateZ description)�cursor�row�dZidx�colr
r
r � dict_factory s r c @ r )�MyCursorc O s$ t jj| g|�R i |�� t| _d S r )r �Cursorr r � row_factoryr r
r
r r ' s zMyCursor.__init__Nr
r
r
r
r r & r r c @ �$ e Zd Zdd� Zdd� Zdd� ZdS )�ConnectionFactoryTestsc C s t jdtd�| _d S )N�:memory:�Zfactory)r �connectr �con�r r
r
r �setUp, r z ConnectionFactoryTests.setUpc C � | j �� d S r �r �closer r
r
r �tearDown/ r z ConnectionFactoryTests.tearDownc C s | � | jt� d S r )�assertIsInstancer r r r
r
r �CheckIsInstance2 r 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 � t �d�| _d S �Nr �r r r r r
r
r r! 6 r zCursorFactoryTests.setUpc C r" r r# r r
r
r r% 9 r zCursorFactoryTests.tearDownc C sR | j �� }| �|tj� | j �t�}| �|t� | j jdd� d�}| �|t� d S )Nc S s t | �S r )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 � d S r r
r
r
r
r r- H r. z8CursorFactoryTests.CheckInvalidFactory.<locals>.<lambda>c S r0 r r
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' r3 r
r
r
r r( 5 s r( c @ r )� RowFactoryTestsBackwardsCompatc C r) r* r+ r r
r
r r! M r z$RowFactoryTestsBackwardsCompat.setUpc C s8 | j jtd�}|�d� |�� }| �|t� |�� d S )Nr zselect 4+5 as foo)r r r �execute�fetchoner& �dictr$ )r r/ r r
r
r �CheckIsProducedByFactoryP s
z7RowFactoryTestsBackwardsCompat.CheckIsProducedByFactoryc C r" r r# r r
r
r r% W r z'RowFactoryTestsBackwardsCompat.tearDownN)r r r r! r8 r% r
r
r
r r4 L s r4 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 r) r* r+ r r
r
r r! [ r zRowFactoryTests.setUpc C s, dd� | j _| j �d��� }| �|t� d S )Nc S s t |�S r )�list)r/ r r
r
r r- _ r. z4RowFactoryTests.CheckCustomFactory.<locals>.<lambda>z select 1, 2)r r r5 r6 r& r: �r r r
r
r �CheckCustomFactory^ s z"RowFactoryTests.CheckCustomFactoryc C s� 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 � n1 s�0 Y | �t �� |d W d � n1 �s0 Y | �t �� |d W d � n1 �sH0 Y | �t �� |d W d � n1 �s|0 Y | �t �� |d W d � n1 �s�0 Y | �t �� |dd W d � n1 �s�0 Y 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'ZBz$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 -2Zcza_za1i����i� )
r �Rowr r r5 r6 r& � assertEqualr1 �
IndexErrorr; r
r
r �CheckSqliteRowIndexc s.
& ( ( ( ( z#RowFactoryTests.CheckSqliteRowIndexc C s� t j| j_| j�d��� }| �|d d� | �t�� |d W d � n1 sR0 Y | �t�� |d W d � n1 s�0 Y d S )Nu select 1 as ÿu ÿr= u Ÿu ß) r rB r r r5 r6 rC r1 rD 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
r= )r= � )r? rG )r? rG � )r= r? rG rA r@ )rG )rG rH rH r? )r= rG )rH r? )r rB r r r5 r6 rC r; r
r
r �CheckSqliteRowSlice� s
z#RowFactoryTests.CheckSqliteRowSlicec C s( t j| j_| j�d��� }|D ]}q dS )z$Checks if the row object is iterable�select 1 as a, 2 as bN)r rB r r r5 r6 )r r r r
r
r �CheckSqliteRowIter� �
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 tuplerJ �ar> N)r rB r r r5 r6 �tuplerC )r r Ztr
r
r �CheckSqliteRowAsTuple� rL 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 dictionaryrJ rM r> N)r rB r r r5 r6 r7 rC )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 d � n1 �s00 Y | � t
�� ||k W d � n1 �sd0 Y | � t
�� ||k W d � n1 �s�0 Y | � t
�� ||k W d � n1 �s�0 Y | � t |�t |�� dS )z6Checks if the row object compares and hashes correctlyrJ zselect 1 as a, 3 as bzselect 1 as b, 2 as azselect 2 as b, 1 as aN)
r rB r r r5 r6 �
assertTrueZ assertFalseZobjectr1 r2 rC Zhash)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 rJ N) r rB r r r5 r6 rN rC r: Zreversedr& 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 Z __class__r
r
r
r �
FakeCursor� r rT r
)�strr rB r r r1 r2 r )r rT r
r
r �CheckFakeCursorClass� s
z$RowFactoryTests.CheckFakeCursorClassc C r" r r# r r
r
r r% � r zRowFactoryTests.tearDownN)r r r r! r< rE rF rI rK rO rP rR rS rV r% r
r
r
r r9 Z s "
r9 c @ �< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�TextFactoryTestsc C r) r* r+ r r
r
r r! � r zTextFactoryTests.setUpc C s2 d}| j �d|f��� }| �t|d �td� d S )N� Österreich�select ?r � type of row[0] must be unicode)r r5 r6 rC �typerU �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 )NrY rZ r z type of row[0] must be bytes�utf-8z(column must equal original data in UTF-8)�bytesr � text_factoryr5 r6 rC r\ Zencoder] 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 )Nr` Zignore)rU �Zxr
r
r r- � r. z.TextFactoryTests.CheckCustom.<locals>.<lambda>rY rZ r r[ Zreichz!column must contain original data) r rb r5 r6 rC r\ rU rQ Zendswithr] 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 )NrY Z
DeutchlandrZ r z!type of non-ASCII row must be strz"type of ASCII-only row must be str) r ZOptimizedUnicoder rb r5 r6 rC r\ rU )r r^ ZgermanyZa_rowZd_rowr
r
r �CheckOptimizedUnicode s
z&TextFactoryTests.CheckOptimizedUnicodec C r" r r# r r
r
r r% r zTextFactoryTests.tearDownN) r r r r! r_ rc re rf r% r
r
r
r rX � s rX c @ rW )�%TextFactoryTestsWithEmbeddedZeroBytesc C s* t �d�| _| j�d� | j�dd� d S )Nr z create table test (value text)z#insert into test (value) values (?))�a b)r r r r5 r r
r
r r! s z+TextFactoryTestsWithEmbeddedZeroBytes.setUpc C s8 | j �d��� }| �t|d �t� | �|d d� d S )N�select value from testr rh )r r5 r6 �assertIsr\ rU rC r; r
r
r rc s z1TextFactoryTestsWithEmbeddedZeroBytes.CheckStringc C �@ t | j_| j�d��� }| �t|d �t � | �|d d� d S �Nri r � a b)ra r rb r5 r6 rj r\ rC r; r
r
r �
CheckBytes � z0TextFactoryTestsWithEmbeddedZeroBytes.CheckBytesc C rk rl )Z bytearrayr rb r5 r6 rj r\ rC r; r
r
r �CheckBytearray ro z4TextFactoryTestsWithEmbeddedZeroBytes.CheckBytearrayc C sD dd� | j _| j �d��� }| �t|d �t� | �|d d� d S )Nc S s | S r r
rd r
r
r r- ( r. zCTextFactoryTestsWithEmbeddedZeroBytes.CheckCustom.<locals>.<lambda>ri r rm )r rb r5 r6 rj r\ ra rC r; r
r
r re & s z1TextFactoryTestsWithEmbeddedZeroBytes.CheckCustomc C r" r r# r r
r
r r% - r z.TextFactoryTestsWithEmbeddedZeroBytes.tearDownN) r r r r! rc rn rp re r% r
r
r
r rg s rg c C s^ t �td�} t �td�}t �td�}t �td�}t �td�}t �td�}t �| |||||f�S )NZCheck) �unittestZ makeSuiter r( r4 r9 rX rg Z TestSuite)Zconnection_suiteZ cursor_suiteZrow_suite_compatZ row_suiteZ
text_suiteZtext_zero_bytes_suiter
r
r �suite0 s rr c C s t �� } | �t� � d S r )rq ZTextTestRunnerZrunrr )Zrunnerr
r
r �test9 s rs Z__main__)rq Zsqlite3r Zcollections.abcr r r r r r ZTestCaser r( r4 r9 rX rg rr rs r r
r
r
r Z<module> s
%"