EVOLUTION-NINJA
Edit File: factory.cpython-38.pyc
U �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� 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 �3/opt/python38/lib/python3.8/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| f|�|� 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 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 5 Q R X | �t �� |d W 5 Q R X | �t �� |d W 5 Q R X | �t �� |d W 5 Q R X | �t �� |d W 5 Q R X | �t �� |dd W 5 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 r5 r6 r( � assertEqualr1 � IndexErrorr; r r r �CheckSqliteRowIndexc s. z#RowFactoryTests.CheckSqliteRowIndexc C sj t j| j_| j�d��� }| �|d d� | �t�� |d W 5 Q R X | �t�� |d W 5 Q R X 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 5 Q R X | � t �� ||k W 5 Q R X | � t �� ||k W 5 Q R X | � t �� ||k W 5 Q R X | � 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 %"