Your IP : 216.73.216.213
��YfV, � @ 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 � 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/python35/lib64/python3.5/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 @ s@ e Z d Z d 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 ss | j j � } | j | t j � | j j t � } | j | t � | j j d d d � � } | j | t � d S)Nr c S s
t | � S)N)r )r r
r
r �<lambda>A s z4CursorFactoryTests.CheckIsInstance.<locals>.<lambda>)r r r r r r )r �curr
r
r r! < s z"CursorFactoryTests.CheckIsInstancec C s[ | j t | j j d � | j t | j j d d � � | j t | j j d d � � d S)Nc S s d S)Nr
r
r
r
r r# H s z8CursorFactoryTests.CheckInvalidFactory.<locals>.<lambda>c S s d S)Nr
)r r
r
r r# J s )� assertRaises� TypeErrorr r )r r
r
r �CheckInvalidFactoryD s z&CursorFactoryTests.CheckInvalidFactoryN)r 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 M 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 �CheckIsProducedByFactoryP s
z7RowFactoryTestsBackwardsCompat.CheckIsProducedByFactoryc C s | j j � d S)N)r r )r r
r
r r W s z'RowFactoryTestsBackwardsCompat.tearDownN)r r
r r r, r r
r
r
r r( L 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 d � Z
d S)�RowFactoryTestsc C s t j d � | _ d S)Nz:memory:)r r r )r r
r
r r [ 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 r# _ s z4RowFactoryTests.CheckCustomFactory.<locals>.<lambda>z select 1, 2)r r r) r* r r. )r r r
r
r �CheckCustomFactory^ 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 QRX| j t � �
| d Wd QRX| j t � �
| d Wd QRX| j t � �
| d Wd QRXd 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 � assertEqualr% �
IndexError)r r Zcol1Zcol2r
r
r �CheckSqliteRowIndexc s* z#RowFactoryTests.CheckSqliteRowIndexc C s5 t j | j _ | j j d � j � } | j | d d � f � | j | d d � d � | j | d d � d � | j | d d � f � | j | d d � d � | j | d d � d
� | j | d d � d
� | j | d d � d � | j | d d d � d � | j | d d d � d � d S)Nzselect 1, 2, 3, 4r r2 r7 r3 � )r2 )r3 r7 )r3 r7 r? )r2 r3 r7 r9 r8 )r7 r9 )r7 r? )r2 r7 r9 )r? r3 )r r; r r r) r* r< )r r r
r
r �CheckSqliteRowSlice~ s z#RowFactoryTests.CheckSqliteRowSlicec 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 r; r r r) r* )r r r r
r
r �CheckSqliteRowIter� 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 br0 r1 N)r r; r r r) r* �tupler< )r r �tr
r
r �CheckSqliteRowAsTuple� 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 br0 r1 N)r r; r r r) r* r+ r< )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 r; r r r) r* r< �
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 r; r r r) r* rB r< r. �reversedr r )r r Zas_tupler
r
r �CheckSqliteRowAsSequence� s
(z(RowFactoryTests.CheckSqliteRowAsSequencec C s^ Gd d � d t � } t j | j _ | j t | j j | � | 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 rL )�strr r; r r r% r& r )r rL 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@ rA rD rE rH rJ rN r r
r
r
r r- Z 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* r< �typerM )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* r< rP �encode)r rQ 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)rM )�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 rT r) r* r< rP rM rF �endswith)r rQ 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 rT r) r* r< rP rM )r rQ 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 rR rV rZ r[ r r
r
r
r rO � s rO 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)r] )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 za b)r r) r* �assertIsrP rM r< )r r r
r
r rV � 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)rS r rT r) r* r^ rP r< )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 rT r) r* r^ rP r< )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
)rX r
r
r r# s zCTextFactoryTestsWithEmbeddedZeroBytes.CheckCustom.<locals>.<lambda>zselect value from testr s a b)r rT r) r* r^ rP rS r< )r r r
r
r rZ
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 rV r_ ra rZ r r
r
r
r r\ � s r\ 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- rO r\ Z TestSuite)Zconnection_suiteZ cursor_suiteZrow_suite_compatZ row_suiteZ
text_suiteZtext_zero_bytes_suiter
r
r �suite s rc c C s t j � } | j t � � d S)N)rb ZTextTestRunnerZrunrc )Zrunnerr
r
r �test s rd �__main__)rb Zsqlite3r �collections.abcr r r r r r ZTestCaser r" r( r- rO r\ rc rd r r
r
r
r �<module> s
v%"