B 18™\V,ã@sÖddlZddlZddlmZGdd„dejƒZdd„ZGdd„dejƒZ Gd d „d ej ƒZ Gd d „d ej ƒZ Gd d„dej ƒZ Gdd„dej ƒZGdd„dej ƒZGdd„dej ƒZdd„Zdd„ZedkrÒeƒdS)éN)ÚSequencec@seZdZdd„ZdS)Ú MyConnectioncOstjj|f|ž|ŽdS)N)ÚsqliteÚ ConnectionÚ__init__)ÚselfÚargsÚkwargs©r úX/oak/stanford/groups/akundaje/marinovg/programs/Python-3.7.3/Lib/sqlite3/test/factory.pyrszMyConnection.__init__N)Ú__name__Ú __module__Ú __qualname__rr r r r rsrcCs0i}x&t|jƒD]\}}||||d<qW|S)Nr)Ú enumerateÚ description)ÚcursorÚrowÚdÚidxÚcolr r r Ú dict_factory src@seZdZdd„ZdS)ÚMyCursorcOstjj|f|ž|Žt|_dS)N)rÚCursorrrÚ row_factory)rrr r r r r'szMyCursor.__init__N)r r rrr r r r r&src@s$eZdZdd„Zdd„Zdd„ZdS)ÚConnectionFactoryTestscCstjdtd|_dS)Nz:memory:)Úfactory)rÚconnectrÚcon)rr r r ÚsetUp,szConnectionFactoryTests.setUpcCs|j ¡dS)N)rÚclose)rr r r ÚtearDown/szConnectionFactoryTests.tearDowncCs| |jt¡dS)N)ÚassertIsInstancerr)rr r r ÚCheckIsInstance2sz&ConnectionFactoryTests.CheckIsInstanceN)r r rrr r"r r r r r+src@s,eZdZdd„Zdd„Zdd„Zdd„Zd S) ÚCursorFactoryTestscCst d¡|_dS)Nz:memory:)rrr)rr r r r6szCursorFactoryTests.setUpcCs|j ¡dS)N)rr)rr r r r 9szCursorFactoryTests.tearDowncCsR|j ¡}| |tj¡|j t¡}| |t¡|jjdd„d}| |t¡dS)NcSst|ƒS)N)r)rr r r ÚAóz4CursorFactoryTests.CheckIsInstance..)r)rrr!rrr)rÚcurr r r r"<s    z"CursorFactoryTests.CheckIsInstancecCsB| t|jjd¡| t|jjdd„¡| t|jjdd„¡dS)NcSsdS)Nr r r r r r$Hr%z8CursorFactoryTests.CheckInvalidFactory..cSsdS)Nr )rr r r r$Jr%)Ú assertRaisesÚ TypeErrorrr)rr r r ÚCheckInvalidFactoryDsz&CursorFactoryTests.CheckInvalidFactoryN)r r rrr r"r)r r r r r#5sr#c@s$eZdZdd„Zdd„Zdd„ZdS)ÚRowFactoryTestsBackwardsCompatcCst d¡|_dS)Nz:memory:)rrr)rr r r rMsz$RowFactoryTestsBackwardsCompat.setUpcCs8|jjtd}| d¡| ¡}| |t¡| ¡dS)N)rzselect 4+5 as foo)rrrÚexecuteÚfetchoner!Údictr)rr&rr r r ÚCheckIsProducedByFactoryPs   z7RowFactoryTestsBackwardsCompat.CheckIsProducedByFactorycCs|j ¡dS)N)rr)rr r r r Wsz'RowFactoryTestsBackwardsCompat.tearDownN)r r rrr.r r r r r r*Lsr*c@sdeZdZdd„Zdd„Zdd„Zdd„Zd d „Zd d „Zd d„Z dd„Z dd„Z dd„Z dd„Z dS)ÚRowFactoryTestscCst d¡|_dS)Nz:memory:)rrr)rr r r r[szRowFactoryTests.setUpcCs,dd„|j_|j d¡ ¡}| |t¡dS)NcSst|ƒS)N)Úlist)r&rr r r r$_r%z4RowFactoryTests.CheckCustomFactory..z select 1, 2)rrr+r,r!r0)rrr r r ÚCheckCustomFactory^s z"RowFactoryTests.CheckCustomFactoryc CsLtj|j_|j d¡ ¡}| |tj¡|d|d}}| |dd¡| |dd¡|d|d}}| |dd ¡| |dd ¡| |d dd ¡| |ddd ¡| |ddd¡| |ddd¡| t ¡|dWdQRX| t ¡|dWdQRX| t ¡|dWdQRX| t ¡|ddWdQRXdS)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'rz#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Úcéýÿÿÿiè) rÚRowrrr+r,r!Ú assertEqualr'Ú IndexError)rrZcol1Zcol2r r r ÚCheckSqliteRowIndexcs*     z#RowFactoryTests.CheckSqliteRowIndexcCsêtj|j_|j d¡ ¡}| |dd…d¡| |dd…d¡| |dd…d¡| |dd…d¡| |dd…d¡| |dd…d ¡| |d d …d ¡| |d d…d ¡| |ddd…d¡| |ddd …d¡dS)Nzselect 1, 2, 3, 4rr r4)r4é)r5r@)r5r@é)r4r5r@r9r8)r@)r@rArAr5)r4r@)rAr5)rr<rrr+r,r=)rrr r r ÚCheckSqliteRowSlice~s z#RowFactoryTests.CheckSqliteRowSlicecCs,tj|j_|j d¡ ¡}x |D]}q WdS)z$Checks if the row object is iterablezselect 1 as a, 2 as bN)rr<rrr+r,)rrrr r r ÚCheckSqliteRowIters  z"RowFactoryTests.CheckSqliteRowItercCs>tj|j_|j d¡ ¡}t|ƒ}| ||d|df¡dS)z4Checks if the row object can be converted to a tuplezselect 1 as a, 2 as br2r3N)rr<rrr+r,Útupler=)rrÚtr r r ÚCheckSqliteRowAsTuple—s z%RowFactoryTests.CheckSqliteRowAsTuplecCsNtj|j_|j d¡ ¡}t|ƒ}| |d|d¡| |d|d¡dS)zCChecks if the row object can be correctly converted to a dictionaryzselect 1 as a, 2 as br2r3N)rr<rrr+r,r-r=)rrrr r r ÚCheckSqliteRowAsDictžs  z$RowFactoryTests.CheckSqliteRowAsDictcCsÎtj|j_|j d¡ ¡}|j d¡ ¡}|j d¡ ¡}| ||¡| ||¡| ||k¡| ||k¡| ||k¡| ||k¡| ||¡| t |ƒt |ƒ¡|  ||¡|  t |ƒt |ƒ¡dS)z6Checks if the row object compares and hashes correctlyzselect 1 as a, 2 as bzselect 1 as a, 3 as bN) rr<rrr+r,r=Ú assertTrueÚ assertFalseÚhashÚassertNotEqual)rZrow_1Zrow_2Zrow_3r r r ÚCheckSqliteRowHashCmp¦s     z%RowFactoryTests.CheckSqliteRowHashCmpcCsNtj|j_|j d¡ ¡}t|ƒ}| tt |ƒƒtt |ƒƒ¡|  |t ¡dS)z2 Checks if the row object can act like a sequence zselect 1 as a, 2 as bN) rr<rrr+r,rDr=r0Úreversedr!r)rrZas_tupler r r ÚCheckSqliteRowAsSequenceºs  z(RowFactoryTests.CheckSqliteRowAsSequencecCsDGdd„dtƒ}tj|j_| t|jj|¡| ttj|ƒd¡dS)Nc@seZdZejZdS)z8RowFactoryTests.CheckFakeCursorClass..FakeCursorN)r r rrrÚ __class__r r r r Ú FakeCursorÇsrPr )Ústrrr<rrr'r(r)rrPr r r ÚCheckFakeCursorClassÃs z$RowFactoryTests.CheckFakeCursorClasscCs|j ¡dS)N)rr)rr r r r ÍszRowFactoryTests.tearDownN)r r rrr1r?rBrCrFrGrLrNrRr r r r r r/Zs  r/c@s<eZdZdd„Zdd„Zdd„Zdd„Zd d „Zd d „Zd S)ÚTextFactoryTestscCst d¡|_dS)Nz:memory:)rrr)rr r r rÑszTextFactoryTests.setUpcCs2d}|j d|f¡ ¡}| t|dƒtd¡dS)Nu Österreichzselect ?rztype of row[0] must be unicode)rr+r,r=ÚtyperQ)rÚaustriarr r r Ú CheckUnicodeÔszTextFactoryTests.CheckUnicodecCsRt|j_d}|j d|f¡ ¡}| t|dƒtd¡| |d| d¡d¡dS)Nu Österreichzselect ?rztype of row[0] must be byteszutf-8z(column must equal original data in UTF-8)ÚbytesrÚ text_factoryr+r,r=rTÚencode)rrUrr r r Ú CheckStringÙs zTextFactoryTests.CheckStringcCsTdd„|j_d}|j d|f¡ ¡}| t|dƒtd¡| |d d¡d¡dS) NcSs t|ddƒS)Nzutf-8Úignore)rQ)Úxr r r r$ár%z.TextFactoryTests.CheckCustom..u Österreichzselect ?rztype of row[0] must be unicodeZreichz!column must contain original data) rrXr+r,r=rTrQrHÚendswith)rrUrr r r Ú CheckCustomàs  zTextFactoryTests.CheckCustomcCsjtj|j_d}d}|j d|f¡ ¡}|j d|f¡ ¡}| t|dƒtd¡| t|dƒtd¡dS)Nu ÖsterreichZ Deutchlandzselect ?rz!type of non-ASCII row must be strz"type of ASCII-only row must be str) rÚOptimizedUnicoderrXr+r,r=rTrQ)rrUZgermanyZa_rowZd_rowr r r ÚCheckOptimizedUnicodeçs z&TextFactoryTests.CheckOptimizedUnicodecCs|j ¡dS)N)rr)rr r r r òszTextFactoryTests.tearDownN) r r rrrVrZr^r`r r r r r rSÐs  rSc@s<eZdZdd„Zdd„Zdd„Zdd„Zd d „Zd d „Zd S)Ú%TextFactoryTestsWithEmbeddedZeroBytescCs*t d¡|_|j d¡|j dd¡dS)Nz:memory:zcreate table test (value text)z#insert into test (value) values (?))zab)rrrr+)rr r r rös  z+TextFactoryTestsWithEmbeddedZeroBytes.setUpcCs8|j d¡ ¡}| t|dƒt¡| |dd¡dS)Nzselect value from testrzab)rr+r,ÚassertIsrTrQr=)rrr r r rZûsz1TextFactoryTestsWithEmbeddedZeroBytes.CheckStringcCs@t|j_|j d¡ ¡}| t|dƒt¡| |dd¡dS)Nzselect value from testrsab)rWrrXr+r,rbrTr=)rrr r r Ú CheckBytessz0TextFactoryTestsWithEmbeddedZeroBytes.CheckBytescCs@t|j_|j d¡ ¡}| t|dƒt¡| |dd¡dS)Nzselect value from testrsab)Ú bytearrayrrXr+r,rbrTr=)rrr r r ÚCheckBytearraysz4TextFactoryTestsWithEmbeddedZeroBytes.CheckBytearraycCsDdd„|j_|j d¡ ¡}| t|dƒt¡| |dd¡dS)NcSs|S)Nr )r\r r r r$r%zCTextFactoryTestsWithEmbeddedZeroBytes.CheckCustom..zselect value from testrsab)rrXr+r,rbrTrWr=)rrr r r r^ s z1TextFactoryTestsWithEmbeddedZeroBytes.CheckCustomcCs|j ¡dS)N)rr)rr r r r sz.TextFactoryTestsWithEmbeddedZeroBytes.tearDownN) r r rrrZrcrer^r r r r r raõs racCs^t td¡}t td¡}t td¡}t td¡}t td¡}t td¡}t ||||||f¡S)NÚCheck) ÚunittestÚ makeSuiterr#r*r/rSraÚ TestSuite)Úconnection_suiteÚ cursor_suiteZrow_suite_compatZ row_suiteZ text_suiteZtext_zero_bytes_suiter r r Úsuites      rlcCst ¡}| tƒ¡dS)N)rgÚTextTestRunnerÚrunrl)Úrunnerr r r Útest srpÚ__main__)rgÚsqlite3rÚcollections.abcrrrrrrÚTestCaserr#r*r/rSrarlrpr r r r r Ús  v%"