Your IP : 216.73.216.213
�
e f�& � @ s. d d l Z d d l Z d d l m Z Gd d � d e j � Z d d � Z Gd d � d e j � Z Gd d
� d
e j
� Z Gd d � d e j
� Z Gd
d � d e j
� Z
Gd d � d e j
� Z Gd d � d e j
� Z Gd d � d e j
� Z d d � Z d d � Z e d k r*e � n d S)� N)�Sequencec @ s e Z d Z d d � Z d S)� MyConnectionc O s t j j | | | � d S)N)�sqlite�
Connection�__init__)�self�args�kwargs� r
�9/opt/alt/python34/lib64/python3.4/sqlite3/test/factory.pyr s zMyConnection.__init__N)�__name__�
__module__� __qualname__r r
r
r
r r s r c C s< i } x/ t | j � D] \ } } | | | | d <q W| S)Nr )� enumerateZ description)�cursor�row�d�idx�colr
r
r � dict_factory s r c @ s e Z d Z d d � Z d S)�MyCursorc O s# t j j | | | � 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 @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�ConnectionFactoryTestsc C s t j d d t �| _ 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 | j t � 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 @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�CursorFactoryTestsc C s t j d � | _ 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 s) | j j d t � } | j | t � d S)Nr )r r r r )r �curr
r
r r! < s z"CursorFactoryTests.CheckIsInstanceN)r r
r r r r! r
r
r
r r" 5 s r" c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)� RowFactoryTestsBackwardsCompatc C s t j d � | _ d S)Nz:memory:)r r r )r r
r
r r A s z$RowFactoryTestsBackwardsCompat.setUpc C sL | j j d t � } | j d � | j � } | j | t � | j � d S)Nr zselect 4+5 as foo)r r r �execute�fetchoner �dictr )r r# r r
r
r �CheckIsProducedByFactoryD s
z7RowFactoryTestsBackwardsCompat.CheckIsProducedByFactoryc C s | j j � d S)N)r r )r r
r
r r K s z'RowFactoryTestsBackwardsCompat.tearDownN)r r
r r r( r r
r
r
r r$ @ s r$ c @ s� e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d S)�RowFactoryTestsc C s t j d � | _ d S)Nz:memory:)r r r )r r
r
r r O s zRowFactoryTests.setUpc C s> d d � | j _ | j j d � j � } | j | t � d S)Nc S s
t | � S)N)�list)r# r r
r
r �<lambda>S s z4RowFactoryTests.CheckCustomFactory.<locals>.<lambda>z select 1, 2)r r r% r& r r* )r r r
r
r �CheckCustomFactoryR s z"RowFactoryTests.CheckCustomFactoryc C s� t j | j _ | j j d � j � } | j | t j � | d | d } } | j | d d � | j | d d � | d | d } } | j | d d � | j | d d
� | j | d d d � | j | d d d
� | j | d d d � | j | d d d � | j t � �
| d Wd QX| j t � �
| d Wd QX| j t � �
| d Wd QX| j t � �
| d Wd QXd 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� ���������lC )
r �Rowr r r% r& r � assertEqual� assertRaises�
IndexError)r r Zcol1Zcol2r
r
r �CheckSqliteRowIndexW s* z#RowFactoryTests.CheckSqliteRowIndexc C s< t j | j _ | j j d � j � } x | D] } q. Wd S)z$Checks if the row object is iterablezselect 1 as a, 2 as bN)r r8 r r r% r& )r r r r
r
r �CheckSqliteRowIterr s
z"RowFactoryTests.CheckSqliteRowIterc C sU t j | j _ | j j d � j � } t | � } | j | | d | d f � d S)z4Checks if the row object can be converted to a tuplezselect 1 as a, 2 as br- r. N)r r8 r r r% r& �tupler9 )r r �tr
r
r �CheckSqliteRowAsTupley s z%RowFactoryTests.CheckSqliteRowAsTuplec C sg t j | j _ | j j d � j � } t | � } | j | d | d � | j | d | d � d S)zCChecks if the row object can be correctly converted to a dictionaryzselect 1 as a, 2 as br- r. N)r r8 r r r% r& r' r9 )r r r r
r
r �CheckSqliteRowAsDict� s
z$RowFactoryTests.CheckSqliteRowAsDictc C s t j | j _ | j j d � j � } | j j d � j � } | j j d � j � } | j | | � | j | | � | j | | k � | j | | k � | j | | k � | j | | k � | j | | � | j t | � t | � � | j
| | � | j
t | � t | � � d S)z6Checks if the row object compares and hashes correctlyzselect 1 as a, 2 as bzselect 1 as a, 3 as bN) r r8 r r r% r& r9 �
assertTrueZ assertFalse�hashZassertNotEqual)r Zrow_1Zrow_2Zrow_3r
r
r �CheckSqliteRowHashCmp� s z%RowFactoryTests.CheckSqliteRowHashCmpc C so t j | j _ | j j d � j � } t | � } | j t t | � � t t | � � � | j
| t � d S)z2 Checks if the row object can act like a sequence zselect 1 as a, 2 as bN) r r8 r r r% r& r> r9 r* �reversedr r )r r Zas_tupler
r
r �CheckSqliteRowAsSequence� s
(z(RowFactoryTests.CheckSqliteRowAsSequencec C sH Gd d � d t � } | j j d | � } | j t t j | f � d S)Nc @ s e Z d Z e j Z d S)z8RowFactoryTests.CheckFakeCursorClass.<locals>.FakeCursorN)r r
r r r � __class__r
r
r
r �
FakeCursor� s rH r )�strr r r: � TypeErrorr r8 )r rH r# 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= r@ rA rD rF rK r r
r
r
r r) N s r) c @ sX e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
S)�TextFactoryTestsc C s t j d � | _ d S)Nz:memory:)r r r )r r
r
r r � s zTextFactoryTests.setUpc C sE d } | j j d | f � j � } | j t | d � t d � d S)Nu Österreichzselect ?r z type of row[0] must be unicode)r r% r& r9 �typerI )r �austriar r
r
r � CheckUnicode� s z TextFactoryTests.CheckUnicodec C sq t | j _ d } | j j d | f � j � } | j t | d � t d � | j | d | j d � 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& r9 rM �encode)r rN r r
r
r � CheckString� s
z TextFactoryTests.CheckStringc C st d d � | j _ d } | j j d | f � j � } | j t | d � t d � | j | d j d � d � d S) Nc S s t | d d � S)Nzutf-8�ignore)rI )�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 rQ r% r& r9 rM rI rB �endswith)r rN r r
r
r � CheckCustom� s
z TextFactoryTests.CheckCustomc C s� t j | j _ d } d } | j j d | f � j � } | j j d | f � j � } | j t | d � t d � | j t | d � t d � 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 rQ r% r& r9 rM rI )r rN 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 rO rS rW rX r r
r
r
r rL � s rL c @ sX e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
S)�%TextFactoryTestsWithEmbeddedZeroBytesc C s9 t j d � | _ | j j d � | j j d d � d S)Nz:memory:z create table test (value text)z#insert into test (value) values (?)�a b)rZ )r r r r% )r r
r
r r � s z+TextFactoryTestsWithEmbeddedZeroBytes.setUpc C sJ | j j d � j � } | j t | d � t � | j | d d � d S)Nzselect value from testr rZ )r r% r& �assertIsrM rI r9 )r r r
r
r rS � s z1TextFactoryTestsWithEmbeddedZeroBytes.CheckStringc C sV t | j _ | j j d � j � } | j t | d � t � | j | d d � d S)Nzselect value from testr s a b)rP r rQ r% r& r[ rM r9 )r r r
r
r �
CheckBytes� s z0TextFactoryTestsWithEmbeddedZeroBytes.CheckBytesc C sV t | j _ | j j d � j � } | j t | d � t � | j | d d � d S)Nzselect value from testr s a b)� bytearrayr rQ r% r& r[ rM r9 )r r r
r
r �CheckBytearray� s z4TextFactoryTestsWithEmbeddedZeroBytes.CheckBytearrayc C s\ d d � | j _ | j j d � j � } | j t | d � t � | j | d d � d S)Nc S s | S)Nr
)rU r
r
r r+ � s zCTextFactoryTestsWithEmbeddedZeroBytes.CheckCustom.<locals>.<lambda>zselect value from testr s a b)r rQ r% r& r[ rM rP r9 )r r r
r
r rW � 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 rS r\ r^ rW r r
r
r
r rY � s rY c C s� t j t d � } t j t d � } t j t d � } t j t d � } t j t d � } t j t d � } t j | | | | | | f � S)NZCheck) �unittestZ makeSuiter r" r$ r) rL rY Z TestSuite)Zconnection_suiteZ cursor_suiteZrow_suite_compatZ row_suiteZ
text_suiteZtext_zero_bytes_suiter
r
r �suite� s r` c C s t j � } | j t � � d S)N)r_ ZTextTestRunnerZrunr` )Zrunnerr
r
r �test s ra �__main__)r_ Zsqlite3r Zcollections.abcr r r r r r ZTestCaser r" r$ r) rL rY r` ra r r
r
r
r �<module> s
b%"