3 (hz @s`dZdZddd ZdZy ddlZWnek r@ddlZYnXddlZGdd d ejZ dS) a Python DB API 2.0 driver compliance unit test suite. This software is Public Domain and may be used without restrictions. "Now we have booze and barflies entering the discussion, plus rumours of DBAs on drugs... and I won't tell you what flashes through my mind each time I read the subject line with 'Anal Compliance' in it. All around this is turning out to be a thoroughly unwholesome unit test." -- Ian Bicking z$Id$z $Revision$ z-Stuart Bishop Nc@seZdZdZdZfZiZdZdeZdeZ deZ deZ dZ d d Z d d Zd dZddZddZddZddZddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*ZdZd+d,Zd-d.Z d/d0Z!d1d2Z"d3d4Z#d5d6Z$d7d8d9d:d;dZ&d?d@Z'dAdBZ(dCdDZ)dEdFZ*dGdHZ+dIdJZ,dKdJZ,dLdMZ-dNdOZ.dPdQZ/dRdSZ0dTdUZ1dVdWZ2dXdYZ3dZd[Z4d\d]Z5d^d_Z6d`daZ7dbdcZ8dddeZ9dfdgZ:dS)hDatabaseAPI20TestaB 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): [...] Don't 'import DatabaseAPI20Test from dbapi20', or you will confuse the unit tester - just 'import dbapi20'. NZ dbapi20test_z'create table %sbooze (name varchar(20))z)create table %sbarflys (name varchar(20))zdrop table %sboozezdrop table %sbarflyslowercCs|j|jdS)N)executeddl1)selfcursorr R/tmp/pip-install-q3hcpn_q/PyMySQL/pymysql/tests/thirdparty/test_MySQLdb/dbapi20.py executeDDL1mszDatabaseAPI20Test.executeDDL1cCs|j|jdS)N)rddl2)rr r r r executeDDL2pszDatabaseAPI20Test.executeDDL2cCsdS)z self.drivers should override this method to perform required setup if any is necessary, such as creating the database. Nr )rr r r setUpsszDatabaseAPI20Test.setUpcCsj|j}zR|j}xD|j|jfD]4}y|j||jWq |jjk rRYq Xq WWd|jXdS)z self.drivers should override this method to perform required cleanup if any is necessary, such as deleting the test database. The default drops the tables that may be created. N) _connectr xddl1xddl2rcommitdriverErrorclose)rconcurZddlr r r tearDownys  zDatabaseAPI20Test.tearDownc Cs6y|jj|j|jStk r0|jdYnXdS)Nz-No connect method found in self.driver module)rconnect connect_argsconnect_kw_argsAttributeErrorfail)rr r r rs  zDatabaseAPI20Test._connectcCs|j}|jdS)N)rr)rrr r r test_connectszDatabaseAPI20Test.test_connectc Cs<y|jj}|j|dWntk r6|jdYnXdS)Nz2.0zDriver doesn't define apilevel)rapilevel assertEqualrr)rr r r r test_apilevels zDatabaseAPI20Test.test_apilevelc Cs>y|jj}|j|dkWntk r8|jdYnXdS)Nrrz"Driver doesn't define threadsafety)rr#rr$)r threadsafety assertTruerr)rr%r r r test_threadsafetys z#DatabaseAPI20Test.test_threadsafetyc Cs>y|jj}|j|dkWntk r8|jdYnXdS)Nqmarknumericnamedformatpyformatz Driver doesn't define paramstyle)r(r)r*r+r,)r paramstyler&rr)rr-r r r test_paramstyles  z!DatabaseAPI20Test.test_paramstylecCs|jt|jjt|jt|jjt|jt|jj|jj|jt|jj|jj|jt|jj|jj|jt|jj |jj|jt|jj |jj|jt|jj |jj|jt|jj |jjdS)N) r& issubclassrWarning ExceptionrInterfaceError DatabaseErrorOperationalErrorIntegrityError InternalErrorProgrammingErrorNotSupportedError)rr r r test_Exceptionss z!DatabaseAPI20Test.test_ExceptionscCs|j}|j}|j|j|jk|j|j|jk|j|j|jk|j|j|jk|j|j|jk|j|j|jk|j|j |j k|j|j |j k|j|j |j kdS)N) rrr&r0rr2r3r4r5r6r7r8)rrdrvr r r %test_ExceptionsAsConnectionAttributessz7DatabaseAPI20Test.test_ExceptionsAsConnectionAttributesc Cs$|j}z |jWd|jXdS)N)rrr)rrr r r test_commits zDatabaseAPI20Test.test_commitc Cs<|j}t|dr8y |jWn|jjk r6YnXdS)Nrollback)rhasattrr=rr8)rrr r r test_rollbacks   zDatabaseAPI20Test.test_rollbackc Cs$|j}z |j}Wd|jXdS)N)rr r)rrrr r r test_cursors zDatabaseAPI20Test.test_cursorc Cs|j}z~|j}|j}|j||jd|j|jd|j|j}|jt|d|jt|dd|j|dddWd|jXdS)Nz.insert into %sbooze values ('Victoria Bitter')zselect name from %sboozer#rzVictoria Bitter) rr r r table_prefixfetchallr!lenr)rrZcur1Zcur2Zboozer r r test_cursor_isolations  z'DatabaseAPI20Test.test_cursor_isolationc Cs|j}z|j}|j||j|jdd|jd|j|jt|jdd|jt|jddd|j|jddjdd |j|jdd|j j d |jdd|j ||j|jdd Wd|j XdS) Nzccursor.description should be none after executing a statement that can return no rows (such as DDL)zselect name from %sboozer#z-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)) rr r r! descriptionrrArCrrSTRINGrr)rrrr r r test_descriptions(     z"DatabaseAPI20Test.test_descriptionc Cs|j}z|j}|j||j|jdd|jd|j|j|jd kd|jd|j|j|jd kd|j||j|jd dWd|j XdS)Nr#zAcursor.rowcount should be -1 after executing no-result statementsz.insert into %sbooze values ('Victoria Bitter')zccursor.rowcount should == number or rows inserted, or set to -1 after executing an insert statementzselect name from %sboozezbcursor.rowcount should == number of rows returned, or set to -1 after executing a select statementzJcursor.rowcount not being reset to -1 after executing no-result statementsrJ)rJr#rJ)rJr#rJ) rr r r!rowcountrrAr&rr)rrrr r r test_rowcount1s"        zDatabaseAPI20Test.test_rowcountc Cs|j}z|j}|jrt|dr|j|jd }|jt|d|j|dd|j}|jt|dd|jt|ddd|j|ddddWd|jXdS) NcallprocFOOr#rzcallproc produced no result setz$callproc produced invalid result setZfooz!callproc produced invalid results)rN) rr lower_funcr>rMr!rCrBr)rrrrr r r test_callprocOs zDatabaseAPI20Test.test_callprocc Cs\|j}z |j}Wd|jX|j|jj|j||j|jj|j|j|jj|jdS)N)rr r assertRaisesrrr r)rrrr r r test_closebs  zDatabaseAPI20Test.test_closec Cs.|j}z|j}|j|Wd|jXdS)N)rr _paraminsertr)rrrr r r test_executets zDatabaseAPI20Test.test_executecCs^|j||jd|j|j|jdk|jjdkrJ|jd|jdn|jjdkrj|jd|jdnr|jjdkr|jd |jd dinN|jjd kr|jd |jdn.|jjd kr|jd|jd din |jd|j|jdk|jd|j|j}|j t |dd|dd|ddg}|j |j |ddd|j |ddddS)Nz.insert into %sbooze values ('Victoria Bitter')r#r(zinsert into %sbooze values (?)Cooper'sr)zinsert into %sbooze values (:1)r*z"insert into %sbooze values (:beer)beerr+z insert into %sbooze values (%%s)r,z&insert into %sbooze values (%%(beer)s)zInvalid paramstylezselect name from %sboozerz%cursor.fetchall returned too few rowsrzFcursor.fetchall retrieved incorrect data, or data inserted incorrectlyzVictoria BitterrJ)rJr#)rV)rV)rVrJ)rJr#) r rrAr&rKrr-rrBr!rCsort)rrresbeersr r r rT|sF            zDatabaseAPI20Test._paraminsertc Csx|j}z^|j}|j|ddg}ddiddig}|jjdkrV|jd|j|n|jjdkrv|jd|j|nj|jjdkr|jd |j|nJ|jjd kr|jd |j|n*|jjd kr|jd |j|n |jd|j|j dkd|j |j d|j|j }|j t |dd|dd|ddg}|j|j |ddd|j |dddWd|jXdS)NCooper'sBoag'srWr(zinsert into %sbooze values (?)r)zinsert into %sbooze values (:1)r*z"insert into %sbooze values (:beer)r+z insert into %sbooze values (%%s)r,z&insert into %sbooze values (%%(beer)s)zUnknown paramstyler#rzIinsert using cursor.executemany set cursor.rowcount to incorrect value %rzselect name from %sboozez2cursor.fetchall retrieved incorrect number of rowsrzincorrect data retrieved)r[)r\rJ)rJr)rr r rr-Z executemanyrArr&rKrrBr!rCrXr)rrrlargsZmargsrYrZr r r test_executemanysP          z"DatabaseAPI20Test.test_executemanyc Cs|j}z|j}|j|jj|j|j||j|jj|j|jd|j|j |jdd|j |j d k|jd|j|j|jj|j|jd|j|j}|j t |dd|j |ddd|j |jdd |j |j d kWd|j XdS)Nzselect name from %sboozez?cursor.fetchone should return None if a query retrieves no rowsr#rz.insert into %sbooze values ('Victoria Bitter')z2cursor.fetchone should have retrieved a single rowzVictoria Bitterz(cursor.fetchone retrieved incorrect datazsz/DatabaseAPI20Test._populate..)samples)rZpopulater )rr _populates  zDatabaseAPI20Test._populatec Csd|j}zJ|j}|j|jj|jd|j|x|jD]}|j|q.z,incorrect data retrieved by cursor.fetchmanyzfcursor.fetchmany should return an empty sequence if called after the whole result set has been fetchedzselect name from %sbarflyszKcursor.fetchmany should return an empty sequence if query retrieved no rowsrJ)rJrhrJ)rJrhrJ)rJrhrJ)rJrhrJ)rJr)rr rRrr fetchmanyr rerrAr!rC arraysizer&rKrXrangerdrr)rrrsqlrProwsir r r test_fetchmany sl             z DatabaseAPI20Test.test_fetchmanyc Csz|j}z`|j}|j|jj|j|j|x|jD]}|j|q:W|j|jj|j|jd|j |j}|j |j d t |j fk|jt |t |j ddd|D}|jx0tdt |j D]}|j|||j |dqW|j}|jt |dd|j |j d t |j fk|j||jd |j |j}|j |j dk|jt |dd Wd|jXdS)Nzselect name from %sboozer#z)cursor.fetchall did not retrieve all rowscSsg|] }|dqS)rr )rarPr r r rcsz3DatabaseAPI20Test.test_fetchall..rz(cursor.fetchall retrieved incorrect rowszacursor.fetchall should return an empty list if called after the whole result set has been fetchedzselect name from %sbarflyszMcursor.fetchall should return an empty list if a select query returns no rowsrJrJrJ)rJr)rr rRrrrBr rerrAr&rKrCrdr!rXrkrr)rrrrlrmrnr r r test_fetchallrs<     zDatabaseAPI20Test.test_fetchallc CsB|j}z(|j}|j|x|jD]}|j|q(W|jd|j|j}|jd}|j}|j}|j |j d k|j t |dd|j t |dd|dg}|j |dd|ddg|j|d|j |dd|ddg|jx2tdt |jD]} |j || |j| dqWWd|jXdS) Nzselect name from %sboozerr#rhz+fetchmany returned incorrect number of rowsz*fetchall returned incorrect number of rowsrz$incorrect data retrieved or insertedrJ)rJrh)rr r rerrAr_rirBr&rKr!rCextendappendrXrkrdr) rrrrlZrows1Zrows23Zrows4Zrows56rmrnr r r test_mixedfetchs2     z!DatabaseAPI20Test.test_mixedfetchcCs tddS)z Should create a procedure called deleteme that returns two result sets, first the number of rows in booze then "name from booze" zHelper not implementedN)NotImplementedError)rrr r r help_nextset_setUpsz$DatabaseAPI20Test.help_nextset_setUpcCs tddS)z*If cleaning up is needed after nextSetTestzHelper not implementedN)rt)rrr r r help_nextset_tearDownsz'DatabaseAPI20Test.help_nextset_tearDowncCs|j}z|j}t|ds dSz|j||j}x|jD]}|j|q>W|j||jd|j}|dt |j kst |j st |j }t |t |j kst |j }|dkst dWd|j|XWd|jXdS)NnextsetZdeletemerz'No more return sets, should return None)rr r>r rerrurMr_rCrdAssertionErrorrwrBrvr)rrrrlZ numberofrowsnamesrbr r r test_nextsets*     zDatabaseAPI20Test.test_nextsetcCs tddS)Nz"Drivers need to override this test)rt)rr r r rzsc Cs6|j}z|j}|jt|ddWd|jXdS)Nrjz cursor.arraysize must be defined)rr r&r>r)rrrr r r test_arraysizes   z DatabaseAPI20Test.test_arraysizec Cs8|j}z |j}|jd|j|Wd|jXdS)N)r|)rr Z setinputsizesrTr)rrrr r r test_setinputsizess  z$DatabaseAPI20Test.test_setinputsizesc CsD|j}z,|j}|jd|jdd|j|Wd|jXdS)Niir)rr Z setoutputsizerTr)rrrr r r test_setoutputsize_basic s  z*DatabaseAPI20Test.test_setoutputsize_basiccCs tddS)Nz!Driver need to override this test)rt)rr r r test_setoutputsizesz$DatabaseAPI20Test.test_setoutputsizec Cs|j}zx|j}|j||jd|j|jd|j|j}|jt|d|jt|dd|j|ddddWd|jXdS)Nz!insert into %sbooze values (NULL)zselect name from %sboozer#rzNULL value not returned as None) rr r rrArBr!rCr)rrrrPr r r test_Nones zDatabaseAPI20Test.test_Nonec Cs&|jjddd}|jjtjd}dS)N r|r) rrr|rrrrrr)rDateZ DateFromTickstimemktime)rZd1Zd2r r r test_Date)szDatabaseAPI20Test.test_Datec Cs&|jjddd}|jjtjd}dS)N -r#r) rr#r#rrrrrr)rZTimeZ TimeFromTicksrr)rt1t2r r r test_Time/szDatabaseAPI20Test.test_Timec Cs,|jjdddddd}|jjtjd}dS) Nrr|rrrr) rrr|rrrrrr)rZ TimestampZTimestampFromTicksrr)rrrr r r test_Timestamp5sz DatabaseAPI20Test.test_TimestampcCs|jjd}|jjd}dS)Ns Something)rBinary)rbr r r test_Binary=s zDatabaseAPI20Test.test_BinarycCs|jt|jdddS)NrHzmodule.STRING must be defined)r&r>r)rr r r test_STRINGAszDatabaseAPI20Test.test_STRINGcCs|jt|jdddS)NBINARYzmodule.BINARY must be defined.)r&r>r)rr r r test_BINARYFszDatabaseAPI20Test.test_BINARYcCs|jt|jdddS)NNUMBERzmodule.NUMBER must be defined.)r&r>r)rr r r test_NUMBERKszDatabaseAPI20Test.test_NUMBERcCs|jt|jdddS)NZDATETIMEz module.DATETIME must be defined.)r&r>r)rr r r test_DATETIMEPszDatabaseAPI20Test.test_DATETIMEcCs|jt|jdddS)NZROWIDzmodule.ROWID must be defined.)r&r>r)rr r r test_ROWIDUszDatabaseAPI20Test.test_ROWID);__name__ __module__ __qualname____doc__rrrrArr rrZ lowerfuncr rrrrrr"r'r.r9r;r<r?r@rDrIrLrOrQrSrUrTr^r`rdrerorprsrurvrzr{r}r~rrrrrrrrrrrr r r r rGs|    !24, R1!   r) rZ __rcs_id__ __version__ __author__Z unittest2Zunittest ImportErrorrZTestCaserr r r r  s  0