f
��_0 � @ 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| g|�R i |�� d S �N)�sqlite�
Connection�__init__��self�args�kwargs� r �-/usr/lib64/python3.10/sqlite3/test/factory.pyr s zMyConnection.__init__N��__name__�
__module__� __qualname__r r r r r
r s r c C s, i }t | j�D ]\}}|| ||d < q|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| 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 & s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�ConnectionFactoryTestsc C s t jdtd�| _d S )N�:memory:��factory)r �connectr �con�r r r r
�setUp, s z ConnectionFactoryTests.setUpc C s | j �� d S r �r" �closer# r r r
�tearDown/ s z ConnectionFactoryTests.tearDownc C s | � | jt� d S r )�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 �Nr �r r! r" r# r r r
r$ 6 s zCursorFactoryTests.setUpc C s | j �� d S 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 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 s d S r r r r r r
r. H r/ z8CursorFactoryTests.CheckInvalidFactory.<locals>.<lambda>c S s d S 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 @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� RowFactoryTestsBackwardsCompatc C s t �d�| _d S r+ r, r# r r r
r$ M s z$RowFactoryTestsBackwardsCompat.setUpc C s8 | j jtd�}|�d� |�� }| �|t� |�� d S )Nr zselect 4+5 as foo)r" r r �execute�fetchoner( �dictr& )r r0 r r r r
�CheckIsProducedByFactoryP s
z7RowFactoryTestsBackwardsCompat.CheckIsProducedByFactoryc C s | j �� d S r r% r# r r r
r' W s 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 s t �d�| _d S r+ r, r# r r r
r$ [ s zRowFactoryTests.setUpc C s, dd� | j _| j �d��� }| �|t� d S )Nc S s t |�S r )�list)r0 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'�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 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 ÿ� ÿr= u Ÿ� ß) r rE r" r r5 r6 rF r1 rG 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? rL )r? rL � )r= r? rL rB rA )rL )rL rM rM r? )r= rL )rM r? )r rE r" r r5 r6 rF 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 rE r" r r5 r6 )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 tuplerO �ar> N)r rE r" r r5 r6 �tuplerF )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 dictionaryrO rQ r> N)r rE r" r r5 r6 r7 rF )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 correctlyrO zselect 1 as a, 3 as bzselect 1 as b, 2 as azselect 2 as b, 1 as aN)
r rE r" r r5 r6 �
assertTrueZ assertFalse�objectr1 r2 rF �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 rO N) r rE r" r r5 r6 rR rF 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 r] r )�strr rE r" r r1 r2 r )r r] r r r
�CheckFakeCursorClass� s
z$RowFactoryTests.CheckFakeCursorClassc C s | j �� d S r r% r# r r r
r' � s zRowFactoryTests.tearDownN)r r r r$ r< rH rK rN rP rT rU rY r[ r_ r' r r r r
r9 Z s "
r9 c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�TextFactoryTestsc C s t �d�| _d S r+ r, r# r r r
r$ � s 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 rF �typer^ �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 )Nra rb r z type of row[0] must be bytes�utf-8z(column must equal original data in UTF-8)�bytesr" � text_factoryr5 r6 rF rd �encodere 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 )Nrh �ignore)r^ ��xr r r
r. � r/ z.TextFactoryTests.CheckCustom.<locals>.<lambda>ra rb r rc Zreichz!column must contain original data) r" rj r5 r6 rF rd r^ rV �endswithre 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 )Nra Z
Deutchlandrb r z!type of non-ASCII row must be strz"type of ASCII-only row must be str) r ZOptimizedUnicoder" rj r5 r6 rF rd r^ )r rf ZgermanyZa_rowZd_rowr r r
�CheckOptimizedUnicode s
z&TextFactoryTests.CheckOptimizedUnicodec C s | j �� d S r r% r# r r r
r' s zTextFactoryTests.tearDownN) r r r r$ rg rl rq rr r' r r r r
r` � s r` 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 )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 rt )r" r5 r6 �assertIsrd r^ rF r; r r r
rl s z1TextFactoryTestsWithEmbeddedZeroBytes.CheckStringc C s@ t | j_| j�d��� }| �t|d �t � | �|d d� d S �Nru r � a b)ri r" rj r5 r6 rv rd rF r; r r r
�
CheckBytes s z0TextFactoryTestsWithEmbeddedZeroBytes.CheckBytesc C s@ t | j_| j�d��� }| �t|d �t � | �|d d� d S rw )� bytearrayr" rj r5 r6 rv rd rF 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 r r rn r r r
r. ( r/ zCTextFactoryTestsWithEmbeddedZeroBytes.CheckCustom.<locals>.<lambda>ru r rx )r" rj r5 r6 rv rd ri rF r; r r r
rq & s z1TextFactoryTestsWithEmbeddedZeroBytes.CheckCustomc C s | j �� d S r r% r# r r r
r' - s z.TextFactoryTestsWithEmbeddedZeroBytes.tearDownN) r r r r$ rl ry r{ rq r' r r r r
rs s rs c C s^ t �td�} t �td�}t �td�}t �td�}t �td�}t �td�}t �| |||||f�S )NZCheck) �unittestZ makeSuiter r* r4 r9 r` rs Z TestSuite)Zconnection_suiteZ cursor_suiteZrow_suite_compatZ row_suiteZ
text_suiteZtext_zero_bytes_suiter r r
�suite0 s r} c C s t �� } | �t� � d S r )r| ZTextTestRunner�runr} )Zrunnerr r r
�test9 s r �__main__)r| Zsqlite3r �collections.abcr r r r r r ZTestCaser r* r4 r9 r` rs r} r r r r r r
�<module> s
%"