Your IP : 216.73.216.213
���]V, � @ 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
�,/opt/alt/python35/lib64/python3.5/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 )� enumerate� 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 r3 r8 r4 � )r3 )r4 r8 )r4 r8 r@ )r3 r4 r8 r: r9 )r8 r: )r8 r@ )r3 r8 r: )r@ r4 )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)Nzselect 1 as a, 2 as b)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)Nzselect 1 as a, 2 as br1 r2 )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)Nzselect 1 as a, 2 as br1 r2 )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)Nzselect 1 as a, 2 as bzselect 1 as a, 3 as b) 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)Nzselect 1 as a, 2 as b) r r< r r r* r+ rC 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 rM )�strr r< r r r& r' r )r rM 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 r0 r? rA rB rE rF rI rK rO 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= �typerN )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= rQ �encode)r rR 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)rN )�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 rU r* r+ r= rQ rN rG �endswith)r rR 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 rU r* r+ r= rQ rN )r rR 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 rS rW r[ r\ r r
r
r
r rP � s rP 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+ �assertIsrQ rN r= )r r r
r
r rW � 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)rT r rU r* r+ r_ rQ 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 rU r* r+ r_ rQ 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
)rY r
r
r r$ s zCTextFactoryTestsWithEmbeddedZeroBytes.CheckCustom.<locals>.<lambda>zselect value from testr s a b)r rU r* r+ r_ rQ rT r= )r r r
r
r r[
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 rW r` rb r[ 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. rP r] Z TestSuite)Zconnection_suiteZ cursor_suiteZrow_suite_compatZ row_suiteZ
text_suiteZtext_zero_bytes_suiter
r
r �suite s rd c C s t j � } | j t � � d S)N)rc ZTextTestRunner�runrd )Zrunnerr
r
r �test s rf �__main__)rc Zsqlite3r �collections.abcr r r r r r ZTestCaser r# r) r. rP r] rd rf r r
r
r
r �<module> s
v%"