g[evddlZddlZddlmZGddejZedkrejdSdS)N)pg_tmpceZdZdZddlmcmZdZedzZdezZ dezZ dezZ d ezZ d Z d Zd Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZd ZdZdZ dZ!dZ"dZ#d Z$d!Z%d"Z&d#Z'gd$Z(d%Z)d&Z*d'Z+d(Z,d)Z-d*Z.d+Z/d,Z0d-Z1d.Z2d/Z3d0Z4d1Z5d2Z6d3Z7d4Z8d5Z9d6Z:d7Z;d8Zd;Z?dS)< test_dbapi20a Test a database self.driver for DB API 2.0 compatibility. This implementation tests Gadfly, but the TestCase is structured so that other self.drivers can subclass this test case to ensure compiliance with the DB-API. It is expected that this TestCase may be expanded in the future if ambiguities or edge conditions are discovered. The 'Optional Extensions' are not yet being tested. self.drivers should subclass this test, overriding setUp, tearDown, self.driver, connect_args and connect_kw_args. Class specification should be as follows: import dbapi20 class mytest(dbapi20.DatabaseAPI20Test): [...] __rcs_id__ = 'Id: dbapi20.py,v 1.10 2003/10/09 03:14:14 zenzen Exp' __version__ = 'Revision: 1.10' __author__ = 'Stuart Bishop ' rN dbapi20test_boozez'create temp table %s (name varchar(20))z.create temp table %sbarflys (name varchar(20))zdrop table %sboozezdrop table %sbarflyslowerc:||jdSN)executeddl1selfcursors R/granary/edev/quantum/lib/python3.11/site-packages/postgresql/test/test_dbapi20.py executeDDL1ztest_dbapi20.executeDDL1Z..c:||jdSr )r ddl2rs r executeDDL2ztest_dbapi20.executeDDL2]rrcT|} |}|j|jfD]A} |||-#|jj$rY>wxYw |dS#|wxYwr ) _connectrxddl1xddl2r commitdriverErrorclose)rconcurddls rtearDownztest_dbapi20.tearDown`s #  3z4:&  s [[ZZ\\\\ ;     T  99;;;;;399;;;;s.$B)A%$B%A74B6A77BB'ctjtj\}}|jd||S)Ntest)userhostport)rinitclusteraddressrconnect)rr'r(s rrztest_dbapi20._connectosJ+---~%%''*$    d   rcV|}|dSr )rrrr s r test_connectztest_dbapi20.test_connectvs  #))+++++rc |jj}||ddS#t$r|dYdSwxYw)Nz2.0zDriver doesn't define apilevel)rapilevel assertEqualAttributeErrorfail)rr1s r test_apilevelztest_dbapi20.test_apilevelzsd/k"8HU##### ///99 -....../s"&A A c |jj}||dvdS#t$r|dYdSwxYw)N)rrz"Driver doesn't define threadsafety)r threadsafety assertTruer3r4)rr9s rtest_threadsafetyztest_dbapi20.test_threadsafetyse3+*<??<9,----- 33399 12222223#'A  A c |jj}||dvdS#t$r|dYdSwxYw)N)qmarknumericnamedformatpyformatz Driver doesn't define paramstyle)r paramstyler:r3r4)rrCs rtest_paramstyleztest_dbapi20.test_paramstylesv1 &:??:" 11199 /0000001r<c|t|jj|jj|t|jj|jj|t|jj|jj|t|jj|jj|t|jj|jj|t|jj |jj|t|jj |jjdSr ) r: issubclassrInterfaceErrorr DatabaseErrorOperationalErrorIntegrityError InternalErrorProgrammingErrorNotSupportedErrorrs rtest_Exceptionsztest_dbapi20.test_Exceptionss//*T[7 8IJJKKK//*T[6t{7HIIJJJ//*T[9$+:KLLMMM//*T[7 8IJJKKK//*T[6t{7HIIJJJ//*T[9$+:KLLMMM//*T[:4;;LMMNNNNNrc|} |j}||j|ju||j|ju||j|ju||j|ju||j|ju||j|ju||j |j u||j |j u||j |j u| dS#| wxYwr ) rrr:WarningrrGrHrIrJrKrLrMr)rr drvs r%test_ExceptionsAsConnectionAttributesz2test_dbapi20.test_ExceptionsAsConnectionAttributessA #  3??3;#+-...??39 )***??3%);;<<<??3$(99:::??3'3+??@@@??3%);;<<<??3$(99:::??3'3+??@@@??3(C,AABBB99;;;;;399;;;;s D0EE2c|} ||dS#|wxYwr )rrrr.s r test_commitztest_dbapi20.test_commitsD #::<<<99;;;;;399;;;; AAc|} t|dr+ |n#|jj$rYnwxYw|dS#|wxYw)Nrollback)rhasattrrXrrMrr.s r test_rollbackztest_dbapi20.test_rollbacks # c*  \\^^^^ ; (    T 99;;;;;399;;;;s+A'<A'A A' AA''A=c|} |}|dS#|wxYwr )rrrrr r!s r test_cursorztest_dbapi20.test_cursorsD # 399;;;;;399;;;;rVc||} |}|}|||d|jz|d|jz|}|t|d|t|dd||ddd|dS#|wxYw)N.insert into %sbooze values ('Victoria Bitter')select name from %sboozer7rVictoria Bitter rrrr table_prefixfetchallr2lenr)rr cur1cur2rs rtest_cursor_isolationz"test_dbapi20.test_cursor_isolations # **,,4 **,,4D<<@<<*T->>??? ==??5CJJq!!!CaMM!$$$E!HQK 122299;;;;;399;;;;s C9D%%D;c`|} |}||||jdd|d|jz|t|jdd|t|jddd||jdddd ||jdd|j j d |jddz| |||jdd | dS#| wxYw) Nzccursor.description should be none after executing a statement that can return no rows (such as DDL)r`r7z-cursor.description describes too many columnsrz1cursor.description[x] tuples must have 7 elementsnamez0cursor.description[x][0] must return column namez8cursor.description[x][1] must return column type. Got %rzVcursor.description not being set to None when executing no-result statements (eg. DDL)) rrrr2 descriptionr rcrer rSTRINGrrr\s rtest_descriptionztest_dbapi20.test_descriptions # 3CCOD6;;)D,==>>>C((3C*++A7COA&q)//11&6COA&q)$+*<> q! CCOD% 99;;;;;399;;;;s E+FF-c:|} |}||||jdd|d|jz||jdd|d|jz||jd|d|jz||jd |d |jz||jdd ||||jdd |dS#|wxYw) NzAcursor.rowcount should be -1 after executing no-result statementsr_r7zccursor.rowcount should == number or rows inserted, or set to -1 after executing an insert statementz8insert into %sbooze select 'Victoria Bitter' WHERE FALSErz>>CL"3CCL" 99;;;;;399;;;;s EFFc|} |}|jrt|dr||jd}|t |d||dd|}|t |dd|t |ddd||dddd |dS#|wxYw) Ncallproc)FOOr7rruzcallproc produced no result setz$callproc produced invalid result setfooz!callproc produced invalid results) rr lower_funcrYrtr2rerdrrr r!rs r test_callprocztest_dbapi20.test_callproc/s* # 3 o '#j11  T_X..ASVVAQqT%    ASVVA?@@@S1YYq+ QqT!WU(99;;;;;399;;;;s D D66E c|} |}|n#|wxYw||jj|j|||jj|j||jj|jdSr )rrr assertRaisesrrrrr\s r test_closeztest_dbapi20.test_closeBs # 399;;;;399;;;;DK%d&6s;;;DK%cj111DK%ci00000s ?Ac|} |}|||jj|j|||jj|jdg||jj|jdddg||jj|j|dS#|wxYw)Nrvgenerate_seriesr7 ) rrrr|rrr executemanyrtr\s rtest_cursor_closeztest_dbapi20.test_cursor_closeTs # 399;;;T[&t'7<<<T[&subAAAT[&s|4E2wOOOT[&sy11199;;;;;399;;;;s CC//Dc|} |}|||dS#|wxYwr )rr _paraminsertrr\s r test_executeztest_dbapi20.test_executegsX # 3S99;;;;;399;;;;s )AA+cd|j_ |d|j_dS#d|j_wxYwNrArB)rrCrrNs rtest_format_executez test_dbapi20.test_format_executeosI#$+'&4;J4;&&&&0>c`|||d|jz||jdv|jjdkr|d|jzdn|jjdkr|d|jzdn|jjdkr!|d |jzd d inu|jjd kr|d |jzdnF|jjdkr!|d|jzd d in|d||jdv|d|jz|}| t|dd|dd|ddg}| | |dd d| |ddddS)Nr_rpr7r>insert into %sbooze values (?)Cooper'sr?insert into %sbooze values (:1)r@"insert into %sbooze values (:beer)beerrrA insert into %sbooze values (%%s)rB&insert into %sbooze values (%%(beer)s)zInvalid paramstyler`rz%cursor.fetchall returned too few rowsrr7zFcursor.fetchall retrieved incorrect data, or data inserted incorrectlyra) rr rcr:rqrrCr4rdr2resort)rr!resbeerss rrztest_dbapi20._paraminsertvsw3++>//#,&())) [w&&;;$t'88 {**;;%(99 {((;;(4+<< J {));;&):: {++;;,t/@@ J 99 !"""//#,&()))++(4+<<=== #3s88AEFFF q6!9SVAY %**,,,58J58-rc|} |}||ddg}ddiddig}|jjdkr|d|jz|n|jjdkr|d |jz|n|jjd kr|d |jz|ns|jjd kr|d |jz|nD|jjdkr|d|jz|n|d||j dvd|j z| d|jz| }| t|dd|dd|ddg}|| |ddd| |ddd|dS#|wxYw)Nr)Boag'srrrr>rr?rr@rrArrBrzUnknown paramstyle)rprzIinsert using cursor.executemany set cursor.rowcount to incorrect value %rr`rz2cursor.fetchall retrieved incorrect number of rowsrr7zincorrect data retrieved)rrrrrCrrcr4r:rqr rdr2rerr)rr r!largsmargsrrs rtest_executemanyztest_dbapi20.test_executemanys #0 3C [ *5j !FH#5 75 k''OO%(99  )++OO&)::  '))OO)D,==  (**OO'$*;;  *,,OO-   II"###??3<6)<(;;)D,==>>> 3CHHQ8 F1Ic!fQi 5::<<<E!HX&@AAAE!HZ(BCCC99;;;;;399;;;;s HIIcd|j_ |d|j_dS#d|j_wxYwr)rrCrrNs rtest_format_executemanyz$test_dbapi20.test_format_executemanysI#$+'&4;J4;&&&&rc&|} |}||jj|j||||jj|j|d|jz| |dd| |j dv|d|jz||jj|j|d|jz|}| t|dd| |ddd | |dd | |j d v| dS#| wxYw) Nr`z?cursor.fetchone should return None if a query retrieves no rowsrprr_r7z2cursor.fetchone should have retrieved a single rowrraz(cursor.fetchone retrieved incorrect dataz>>CLLNN4??3<6)***;;?T[&s|444;;)D,==>>> ||~~1CFF18AaD*.CLLNN4B??3<6)***99;;;;;399;;;;s GG::H)z Carlton Coldz Carlton Draftz Mountain GoatRedbackraXXXXc.fdjD}|S)zH Return a list of sql commands to setup the DB for the fetch tests. c,g|]}djd|dS)z insert into zbooze values ('z'))rc).0srs r z*test_dbapi20._populate..s= +/*;*;*;AAA>r)samples)rpopulates` r _populateztest_dbapi20._populates4 \( /rc0 |} |}||jj|jd|||D]}|||d|j z|}| t|ddd|_ |d}| t|dd|d}| t|dd|d}| t|d d | |jd vd|_ |d|j z|}| t|dd |}| t|d|}| t|d | |jd vd |_ |d|j z|}| |jd v| t|d | t|d d|D}|t!d d D]*}| |||j|d+|}| t|d d| |jd v|||d|j z|}| t|d d| |jdv|dS#|wxYw)Nr`r7zQcursor.fetchmany retrieved incorrect number of rows, default of arraysize is one.rr8z3cursor.fetchmany retrieved incorrect number of rowsrrzLcursor.fetchmany should return an empty sequence after results are exhaustedrpz0cursor.arraysize not being honoured by fetchmanyrcg|] }|d Srrrys rrz/test_dbapi20.test_fetchmany..W   A1Q4   rz,incorrect data retrieved by cursor.fetchmanyzfcursor.fetchmany should return an empty sequence if called after the whole result set has been fetchedselect name from %sbarflyszKcursor.fetchmany should return an empty sequence if query retrieved no rowsr)rrr|rr fetchmanyrrr rcr2re arraysizer:rqrrangerrr)rr r!sqlryrowsis rtest_fetchmanyztest_dbapi20.test_fetchmany!s> #N 3T[&s}Q777CnnsKK;;)D,==>>> }}1CFF1#3= }}Q1CFF19 }}Q1CFF19 }}Q1CFF1??3<6)***3=;;)D,==>>> }}1CFF16 }}1CFF1 }}1CFF1??3<6)***3=;;)D,==>>> --//4??3<6)***CIIa   CIIa       499;;; !AJJqT!WT\!_3 --//4CIIa9??3<6)***C;;+d.??@@@ }}1CFF1??3<6)***99;;;;;399;;;;s QQ??Rc|} |}||jj|j|||D]}||||jj|j|d|j z|}| |j dt|j fv|t|t|j dd|D}|t!dt|j D]*}||||j |d+|}|t|dd| |j dt|j fv|||d|j z|}| |j d v|t|dd |dS#|wxYw) Nr`rpz)cursor.fetchall did not retrieve all rowscg|] }|d Srrrs rrz.test_dbapi20.test_fetchall..rrrz(cursor.fetchall retrieved incorrect rowszacursor.fetchall should return an empty list if called after the whole result set has been fetchedrrzMcursor.fetchall should return an empty list if a select query returns no rows)rrr|rrrdrrr rcr:rqrerr2rrrr)rr r!rrrs r test_fetchallztest_dbapi20.test_fetchallss #, 3T[& 555CnnsKKT[&s|444;;)D,==>>> ,,..4??3>> LLNN5 MM!  6 LLNN5 LLNN6??3<6)***CKK1CKK0 8*4;;q ! VAYq\*+++;;uQx;;q ! VAYq\*+++99;;; !C %% & &qT!WT\!_+ 99;;;;;399;;;;s H H77I cz|d|jz|d|jzdS)z Should create a procedure called deleteme that returns two result sets, first the number of rows in booze then "name from booze" zselect name from zselect count(*) from N)r booze_namerr!s rhelp_nextset_setUpztest_dbapi20.help_nextset_setUps> ++!DO3444++%788888rcdS)z*If cleaning up is needed after nextSetTestNrrs rhelp_nextset_tearDownz"test_dbapi20.help_nextset_tearDown$rc|} |}t|ds |dS |||}|D]}|||||}|dt|j ksJ| sJ| }t|t|j ksJ| }|dks Jd | |n#| |wxYw |dS#|wxYw)Nnextsetrz'No more return sets, should return None)rrrYrrrr rrrerrrdr)rr r!r numberofrowsnamesrs r test_nextsetztest_dbapi20.test_nextsets # 3 #i   (99;;;;;%$S C~~[[C   L ?S.. . . . . ;;== ,,..E u::T\** * * * * kkmmA 999>9999s####Ds#####99;;;;;399;;;;s$$F-DE<&F-<FF--Gc|} |}|t|dd|dS#|wxYw)Nrz cursor.arraysize must be defined)rrr:rYrr\s rtest_arraysizeztest_dbapi20.test_arraysizesk # 3??73{++&99;;;;;399;;;;s 8A$$A:c|} |}|d|||dS#|wxYw)N))rr setinputsizesrrr\s rtest_setinputsizesztest_dbapi20.test_setinputsizessl # 3eS99;;;;;399;;;;s >A**Bc2|} |}|d|dd|||dS#|wxYw)Niir)rr setoutputsizerrr\s rtest_setoutputsize_basicz%test_dbapi20.test_setoutputsize_basics # 3TT!S99;;;;;399;;;;s ABBcdSr rrNs rtest_setoutputsizeztest_dbapi20.test_setoutputsize rrcX|}|} d|_||j|j||j|j|}|d|d|||dS#||wxYw)NTz(create table some_committed_table(i int)zdrop table some_committed_table) r autocommitr|rGrrXrr r)rr con2cs rtest_autocommitztest_dbapi20.test_autocommits # $3>JL zz||199 7888;;==:;;;;;===99;;;::<<<<<99;;;::<<<>> ||~~1CFF1C!IIa   AaDGD!BCCC99;;;;;399;;;;s C&DD(c|jddd}|jtjd}|t |t |dS)N r) rrrrrrrrr)rDate DateFromTickstimemktimer2str)rd1d2s r test_Dateztest_dbapi20.test_Date4sc {R##" {  -E!F!FGG"3r773r77#####rc|jddd}|jtjd}|t |t |dS)N -) ir7r7rrrrrr)rTime TimeFromTicksrrr2rrt1t2s r test_Timeztest_dbapi20.test_Time:sc {2b!!" {  -F!G!GHH"3r773r77#####rc|jdddddd}|jtjd}dS)Nrrrrrr) rrrrrrrrr)r TimestampTimestampFromTicksrrrs rtest_Timestampztest_dbapi20.test_Timestamp@sJ {T"R2b11" {%%;*++"""rcn|jd}|jd}dS)Ns Somethingr)rBinary)rbs r test_Binaryztest_dbapi20.test_BinaryHs0 k&&! k!!!rcX|t|jdddS)Nrmzmodule.STRING must be definedr:rYrrNs r test_STRINGztest_dbapi20.test_STRINGLs3//'$+h//"rcX|t|jdddS)NBINARYzmodule.BINARY must be defined.rrNs r test_BINARYztest_dbapi20.test_BINARYQ3//'$+h//#rcX|t|jdddS)NNUMBERzmodule.NUMBER must be defined.rrNs r test_NUMBERztest_dbapi20.test_NUMBERVr rcX|t|jdddS)NDATETIMEz module.DATETIME must be defined.rrNs r test_DATETIMEztest_dbapi20.test_DATETIME[s3//'$+j11%rcX|t|jdddS)NROWIDzmodule.ROWID must be defined.rrNs r test_ROWIDztest_dbapi20.test_ROWID`s3//'$+g.."rc|} |}|dd|d|d|dddi|d|d|dS#|wxYw)NzSELECT 100 %% %s)cr7rzSELECT 100 %% %(foo)srvr)rrr r2rr)rr rs rtest_placeholder_escapez$test_dbapi20.test_placeholder_escapees # zz||199 '''Aqzz||A'''99 $ubk222Aqzz||A'''99;;;;;399;;;;s BC C )@__name__ __module__ __qualname____doc__postgresql.driver.dbapi20rdbapi20rcrr rrr lowerfuncrrr#rr/r5r;rDrOrSrUrZr]rhrnrrrwrzr}rrrrrrrrrrrrrrrrrrrrrrrrrrrr rrrrrrrr5sm.,++++++++W$1J>8<G  , , .    ///333 1 1 1 O O O,   &B$$$L&111$&'''000d222h''')))V   PPPd...`B9996      .   $$$ $$$          rr__main__)unittestrtemporalrTestCaserrmainrrrr"s ^y y y y y 8$y y y v zr