#!/usr/bin/env python''' 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'''__rcs_id__  = '$Id: dbapi20.py 396 2006-02-25 03:44:32Z adustman $'__version__ = '$Revision: 396 $'[11:-2]__author__ = 'Stuart Bishop <zen@shangri-la.dropbear.id.au>'import unittestimport time# $Log$# Revision 1.1.2.1  2006/02/25 03:44:32  adustman# Generic DB-API unit test module## Revision 1.10  2003/10/09 03:14:14  zenzen# Add test for DB API 2.0 optional extension, where database exceptions# are exposed as attributes on the Connection object.## Revision 1.9  2003/08/13 01:16:36  zenzen# Minor tweak from Stefan Fleiter## Revision 1.8  2003/04/10 00:13:25  zenzen# Changes, as per suggestions by M.-A. Lemburg# - Add a table prefix, to ensure namespace collisions can always be avoided## Revision 1.7  2003/02/26 23:33:37  zenzen# Break out DDL into helper functions, as per request by David Rushby## Revision 1.6  2003/02/21 03:04:33  zenzen# Stuff from Henrik Ekelund:#     added test_None#     added test_nextset & hooks## Revision 1.5  2003/02/17 22:08:43  zenzen# Implement suggestions and code from Henrik Eklund - test that cursor.arraysize# defaults to 1 & generic cursor.callproc test added## Revision 1.4  2003/02/15 00:16:33  zenzen# Changes, as per suggestions and bug reports by M.-A. Lemburg,# Matthew T. Kromer, Federico Di Gregorio and Daniel Dittmar# - Class renamed# - Now a subclass of TestCase, to avoid requiring the driver stub#   to use multiple inheritance# - Reversed the polarity of buggy test in test_description# - Test exception heirarchy correctly# - self.populate is now self._populate(), so if a driver stub#   overrides self.ddl1 this change propogates# - VARCHAR columns now have a width, which will hopefully make the#   DDL even more portible (this will be reversed if it causes more problems)# - cursor.rowcount being checked after various execute and fetchXXX methods# - Check for fetchall and fetchmany returning empty lists after results#   are exhausted (already checking for empty lists if select retrieved#   nothing# - Fix bugs in test_setoutputsize_basic and test_setinputsizes#class DatabaseAPI20Test(unittest.TestCase):    ''' 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'.    '''    # The self.driver module. This should be the module where the 'connect'    # method is to be found    driver = None    connect_args = () # List of arguments to pass to connect    connect_kw_args = {} # Keyword arguments for connect    table_prefix = 'dbapi20test_' # If you need to specify a prefix for tables    ddl1 = 'create table %sbooze (name varchar(20))' % table_prefix    ddl2 = 'create table %sbarflys (name varchar(20))' % table_prefix    xddl1 = 'drop table %sbooze' % table_prefix    xddl2 = 'drop table %sbarflys' % table_prefix    lowerfunc = 'lower' # Name of stored procedure to convert string->lowercase            # Some drivers may need to override these helpers, for example adding    # a 'commit' after the execute.    def executeDDL1(self,cursor):        cursor.execute(self.ddl1)    def executeDDL2(self,cursor):        cursor.execute(self.ddl2)    def setUp(self):        ''' self.drivers should override this method to perform required setup            if any is necessary, such as creating the database.        '''        pass    def tearDown(self):        ''' 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.        '''        con = self._connect()        try:            cur = con.cursor()            for ddl in (self.xddl1,self.xddl2):                try:                     cur.execute(ddl)                    con.commit()                except self.driver.Error:                     # Assume table didn't exist. Other tests will check if                    # execute is busted.                    pass        finally:            con.close()    def _connect(self):        try:            return self.driver.connect(                *self.connect_args,**self.connect_kw_args                )        except AttributeError:            self.fail("No connect method found in self.driver module")    def test_connect(self):        con = self._connect()        con.close()    def test_apilevel(self):        try:            # Must exist            apilevel = self.driver.apilevel            # Must equal 2.0            self.assertEqual(apilevel,'2.0')        except AttributeError:            self.fail("Driver doesn't define apilevel")    def test_threadsafety(self):        try:            # Must exist            threadsafety = self.driver.threadsafety            # Must be a valid value            self.failUnless(threadsafety in (0,1,2,3))        except AttributeError:            self.fail("Driver doesn't define threadsafety")    def test_paramstyle(self):        try:            # Must exist            paramstyle = self.driver.paramstyle            # Must be a valid value            self.failUnless(paramstyle in (                'qmark','numeric','named','format','pyformat'                ))        except AttributeError:            self.fail("Driver doesn't define paramstyle")    def test_Exceptions(self):        # Make sure required exceptions exist, and are in the        # defined heirarchy.        self.failUnless(issubclass(self.driver.Warning,StandardError))        self.failUnless(issubclass(self.driver.Error,StandardError))        self.failUnless(            issubclass(self.driver.InterfaceError,self.driver.Error)            )        self.failUnless(            issubclass(self.driver.DatabaseError,self.driver.Error)            )        self.failUnless(            issubclass(self.driver.OperationalError,self.driver.Error)            )        self.failUnless(            issubclass(self.driver.IntegrityError,self.driver.Error)            )        self.failUnless(            issubclass(self.driver.InternalError,self.driver.Error)            )        self.failUnless(            issubclass(self.driver.ProgrammingError,self.driver.Error)            )        self.failUnless(            issubclass(self.driver.NotSupportedError,self.driver.Error)            )    def test_ExceptionsAsConnectionAttributes(self):        # OPTIONAL EXTENSION        # Test for the optional DB API 2.0 extension, where the exceptions        # are exposed as attributes on the Connection object        # I figure this optional extension will be implemented by any        # driver author who is using this test suite, so it is enabled        # by default.        con = self._connect()        drv = self.driver        self.failUnless(con.Warning is drv.Warning)        self.failUnless(con.Error is drv.Error)        self.failUnless(con.InterfaceError is drv.InterfaceError)        self.failUnless(con.DatabaseError is drv.DatabaseError)        self.failUnless(con.OperationalError is drv.OperationalError)        self.failUnless(con.IntegrityError is drv.IntegrityError)        self.failUnless(con.InternalError is drv.InternalError)        self.failUnless(con.ProgrammingError is drv.ProgrammingError)        self.failUnless(con.NotSupportedError is drv.NotSupportedError)    def test_commit(self):        con = self._connect()        try:            # Commit must work, even if it doesn't do anything            con.commit()        finally:            con.close()    def test_rollback(self):        con = self._connect()        # If rollback is defined, it should either work or throw        # the documented exception        if hasattr(con,'rollback'):            try:                con.rollback()            except self.driver.NotSupportedError:                pass        def test_cursor(self):        con = self._connect()        try:            cur = con.cursor()        finally:            con.close()    def test_cursor_isolation(self):        con = self._connect()        try:            # Make sure cursors created from the same connection have            # the documented transaction isolation level            cur1 = con.cursor()            cur2 = con.cursor()            self.executeDDL1(cur1)            cur1.execute("insert into %sbooze values ('Victoria Bitter')" % (                self.table_prefix                ))            cur2.execute("select name from %sbooze" % self.table_prefix)            booze = cur2.fetchall()            self.assertEqual(len(booze),1)            self.assertEqual(len(booze[0]),1)            self.assertEqual(booze[0][0],'Victoria Bitter')        finally:            con.close()    def test_description(self):        con = self._connect()        try:            cur = con.cursor()            self.executeDDL1(cur)            self.assertEqual(cur.description,None,                'cursor.description should be none after executing a '                'statement that can return no rows (such as DDL)'                )            cur.execute('select name from %sbooze' % self.table_prefix)            self.assertEqual(len(cur.description),1,                'cursor.description describes too many columns'                )            self.assertEqual(len(cur.description[0]),7,                'cursor.description[x] tuples must have 7 elements'                )            self.assertEqual(cur.description[0][0].lower(),'name',                'cursor.description[x][0] must return column name'