Your IP : 216.73.216.85
o
0�i=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� Zedkrke� dS dS )� N)�Sequencec @ � e Zd Zdd� ZdS )� MyConnectionc O s t jj| g|�R i |�� d S �N)�sqlite�
Connection�__init__��self�args�kwargs� r
�;/opt/alt/python310/lib64/python3.10/sqlite3/test/factory.pyr s zMyConnection.__init__N��__name__�
__module__� __qualname__r r
r
r
r r � 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 @ 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:��factory)r �connectr �con�r
r
r
r �setUp+ s z ConnectionFactoryTests.setUpc C � | j �� d S r �r% �closer& r
r
r �tearDown. � z ConnectionFactoryTests.tearDownc C s | � | jt� d S r )�assertIsInstancer% r r& r
r
r �test_is_instance1 s z'ConnectionFactoryTests.test_is_instanceN)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' 5 � zCursorFactoryTests.setUpc C r( r r) r& r
r
r r+ 8 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>@ � z5CursorFactoryTests.test_is_instance.<locals>.<lambda>r" )r% r r- r r r )r
�curr
r
r r. ; s
z#CursorFactoryTests.test_is_instancec C sB | � t| jjd � | � t| jjdd� � | � t| jjdd� � d S )Nc S � d S r r
r
r
r
r r5 G � z9CursorFactoryTests.test_invalid_factory.<locals>.<lambda>c S r8 r r
r4 r
r
r r5 I r9 )� assertRaises� TypeErrorr% r r& r
r
r �test_invalid_factoryC s z'CursorFactoryTests.test_invalid_factoryN)r r r r' r+ r. r< r
r
r
r r/ 4 s
r/ c @ r )� RowFactoryTestsBackwardsCompatc C r0 r1 r2 r& r
r
r r' L r3 z$RowFactoryTestsBackwardsCompat.setUpc C s8 | j jtd�}|�d� |�� }| �|t� |�� d S )Nr" zselect 4+5 as foo)r% r r �execute�fetchoner- �dictr* )r
r7 r r
r
r �test_is_produced_by_factoryO s
z:RowFactoryTestsBackwardsCompat.test_is_produced_by_factoryc C r( r r) r& r
r
r r+ V r, z'RowFactoryTestsBackwardsCompat.tearDownN)r r r r' rA r+ r
r
r
r r= K 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 r0 r1 r2 r& r
r
r r' Z r3 zRowFactoryTests.setUpc C s, dd� | j _| j �d��� }| �|t� d S )Nc S s t |�S r )�list)r7 r r
r
r r5 ^ r6 z5RowFactoryTests.test_custom_factory.<locals>.<lambda>z select 1, 2)r% r r> r? r- rC �r
r r
r
r �test_custom_factory] s z#RowFactoryTests.test_custom_factoryc 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 spw Y | �t �� |d W d � n1 s�w Y | �t �� |d W d � n1 s�w Y | �t �� |d W d � n1 s�w Y | �t �� |d W d � n1 s�w Y | �t �� |dd W d � d S 1 s�w 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 r> r? r- � assertEqualr: �
IndexErrorrD r
r
r �test_sqlite_row_indexb s:
�
�
�
�
� "�z%RowFactoryTests.test_sqlite_row_indexc C s� t j| j_| j�d��� }| �|d d� | �t�� |d W d � n1 s)w Y | �t��
|d W d � d S 1 sCw Y d S )Nu select 1 as ÿ� ÿrF u Ÿ� ß) r rN r% r r> r? rO r: rP rD r
r
r � test_sqlite_row_index_unicode s
�
"�z-RowFactoryTests.test_sqlite_row_index_unicodec 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
rF )rF � )rH rU )rH rU � )rF rH rU rK rJ )rU )rU rV rV rH )rF rU )rV rH �r rN r% r r> r? rO rD r
r
r �test_sqlite_row_slice� s
z%RowFactoryTests.test_sqlite_row_slicec C s\ t j| j_| j�d��� }dd� |D �}| �|ddg� dd� |D �}| �|ddg� d S )N�select 1 as a, 2 as bc S � g | ]}|�qS r
r
��.0r r
r
r �
<listcomp>� � z8RowFactoryTests.test_sqlite_row_iter.<locals>.<listcomp>rF rH c S rZ r
r
r[ r
r
r r] � r^ rW )r
r �itemsr
r
r �test_sqlite_row_iter� s
z$RowFactoryTests.test_sqlite_row_iterc C s@ t j| j_| j�d��� }t|�}| �||d |d f� d S �NrY �arG )r rN r% r r> r? �tuplerO )r
r �tr
r
r �test_sqlite_row_as_tuple� s
z(RowFactoryTests.test_sqlite_row_as_tuplec C sP t j| j_| j�d��� }t|�}| �|d |d � | �|d |d � d S ra )r rN r% r r> r? r@ rO )r
r r r
r
r �test_sqlite_row_as_dict� s
z'RowFactoryTests.test_sqlite_row_as_dictc 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 s�w Y | � t
�� ||k W d � n1 s�w Y | � t
�� ||k W d � n1 s�w Y | � t
�� ||k W d � n1 s�w Y | � t |�t |�� d S )NrY zselect 1 as a, 3 as bzselect 1 as b, 2 as azselect 2 as b, 1 as a)
r rN r% r r> r? �
assertTrueZ assertFalse�objectr: r; rO �hash)r
Zrow_1Zrow_2Zrow_3Zrow_4Zrow_5r
r
r �test_sqlite_row_hash_cmp� s@
�
�
�
�z(RowFactoryTests.test_sqlite_row_hash_cmpc C sP t j| j_| j�d��� }t|�}| �tt |��tt |��� | �
|t � d S )NrY ) r rN r% r r> r? rc rO rC �reversedr- r )r
r Zas_tupler
r
r �test_sqlite_row_as_sequence� s
z+RowFactoryTests.test_sqlite_row_as_sequencec C sD G dd� dt �}tj| j_| �t| jj|� | �ttj|� d� d S )Nc @ s e Zd ZejZdS )z:RowFactoryTests.test_fake_cursor_class.<locals>.FakeCursorN)r r r r r � __class__r
r
r
r �
FakeCursor� s
rn r
)�strr rN r% r r: r; r )r
rn r
r
r �test_fake_cursor_class� s
z&RowFactoryTests.test_fake_cursor_classc C r( r r) r& r
r
r r+ � r, zRowFactoryTests.tearDownN)r r r r' rE rQ rT rX r` re rf rj rl rp r+ r
r
r
r rB Y s
"
rB c @ �< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�TextFactoryTestsc C r0 r1 r2 r& r
r
r r' � r3 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% r> r? rO �typero �r
�austriar r
r
r � test_unicode� s z TextFactoryTests.test_unicodec C sR t | j_d}| j�d|f��� }| �t|d �t d� | �|d |�d�d� d S )Nrs rt r z type of row[0] must be bytes�utf-8z(column must equal original data in UTF-8)�bytesr% � text_factoryr> r? rO rv �encoderw r
r
r � test_string� s
z TextFactoryTests.test_stringc C sT dd� | j _d}| j �d|f��� }| �t|d �td� | �|d �d�d� d S ) Nc S s t | dd�S )Nrz �ignore)ro ��xr
r
r r5 � s z.TextFactoryTests.test_custom.<locals>.<lambda>rs rt r ru Zreichz!column must contain original data) r% r| r> r? rO rv ro rg �endswithrw r
r
r � test_custom� s
z TextFactoryTests.test_customc C s� | � t��
}tj| j_W d � n1 sw Y | �d|j� d}d}| j�d|f�� � }| j�d|f�� � }| �
t |d �t d� | �
t |d �t d� d S )Nz
factory.pyrs Z
Deutchlandrt r z!type of non-ASCII row must be strz"type of ASCII-only row must be str)
Z assertWarns�DeprecationWarningr ZOptimizedUnicoder% r| ZassertIn�filenamer> r? rO rv ro )r
�cmrx ZgermanyZa_rowZd_rowr
r
r �test_optimized_unicode s �z'TextFactoryTests.test_optimized_unicodec C r( r r) r& r
r
r r+ r, zTextFactoryTests.tearDownN) r r r r' ry r~ r� r� r+ r
r
r
r rr � s rr c @ rq )�%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% r> 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 r� )r% r> r? �assertIsrv ro rO rD r
r
r r~ s z1TextFactoryTestsWithEmbeddedZeroBytes.test_stringc C �@ t | j_| j�d��� }| �t|d �t � | �|d d� d S �Nr� r � a b)r{ r% r| r> r? r� rv rO rD r
r
r �
test_bytes � z0TextFactoryTestsWithEmbeddedZeroBytes.test_bytesc C r� r� )� bytearrayr% r| r> r? r� rv rO rD r
r
r �test_bytearray& r� z4TextFactoryTestsWithEmbeddedZeroBytes.test_bytearrayc C sD dd� | j _| j �d��� }| �t|d �t� | �|d d� d S )Nc S s | S r r
r� r
r
r r5 . r9 zCTextFactoryTestsWithEmbeddedZeroBytes.test_custom.<locals>.<lambda>r� r r� )r% r| r> r? r� rv r{ rO rD r
r
r r� , s z1TextFactoryTestsWithEmbeddedZeroBytes.test_customc C r( r r) r& r
r
r r+ 3 r, z.TextFactoryTestsWithEmbeddedZeroBytes.tearDownN) r r r r' r~ r� r� r� r+ r
r
r
r r� s r� c C s$ t tttttg} t�dd� | D ��S )Nc S s g | ] }t �� �|��qS r
)�unittestZ
TestLoaderZloadTestsFromTestCase)r\ rd r
r
r r] @ s zsuite.<locals>.<listcomp>)r r/ rB r= rr r� r� Z TestSuite)Ztestsr
r
r �suite6 s � �r� c C s t �� } | �t� � d S r )r� ZTextTestRunner�runr� )Zrunnerr
r
r �testC s r� �__main__)r� Zsqlite3r Zcollections.abcr r r r r r ZTestCaser r/ r= rB rr r� r� r� r r
r
r
r �<module> s$
&"
�