Changeset 760 for trunk


Ignore:
Timestamp:
Jan 16, 2016, 7:09:51 AM (4 years ago)
Author:
cito
Message:

Test reset() and reopen() methods of DB

File:
1 edited

Legend:

Unmodified
Added
Removed
  • trunk/tests/test_classic_dbwrapper.py

    r759 r760  
    232232        self.assertRaises(pg.InternalError, self.db.close)
    233233        self.assertRaises(pg.InternalError, self.db.query, 'select 1')
     234
     235    def testMethodReset(self):
     236        con = self.db.db
     237        self.db.reset()
     238        self.assertIs(self.db.db, con)
     239        self.db.query("select 1+1")
     240        self.db.close()
     241
     242    def testMethodReopen(self):
     243        con = self.db.db
     244        self.db.reopen()
     245        self.assertIsNot(self.db.db, con)
     246        self.db.query("select 1+1")
     247        self.db.close()
    234248
    235249    def testExistingConnection(self):
     
    556570        self.db.end()
    557571        self.assertEqual(g('standard_conforming_strings'), 'off')
     572
     573    def testReset(self):
     574        db = DB()
     575        default_datestyle = db.get_parameter('datestyle')
     576        changed_datestyle = 'ISO, DMY'
     577        if changed_datestyle == default_datestyle:
     578            changed_datestyle == 'ISO, YMD'
     579        self.db.set_parameter('datestyle', changed_datestyle)
     580        r = self.db.get_parameter('datestyle')
     581        self.assertEqual(r, changed_datestyle)
     582        con = self.db.db
     583        q = con.query("show datestyle")
     584        self.db.reset()
     585        r = q.getresult()[0][0]
     586        self.assertEqual(r, changed_datestyle)
     587        q = con.query("show datestyle")
     588        r = q.getresult()[0][0]
     589        self.assertEqual(r, default_datestyle)
     590        r = self.db.get_parameter('datestyle')
     591        self.assertEqual(r, default_datestyle)
     592
     593    def testReopen(self):
     594        db = DB()
     595        default_datestyle = db.get_parameter('datestyle')
     596        changed_datestyle = 'ISO, DMY'
     597        if changed_datestyle == default_datestyle:
     598            changed_datestyle == 'ISO, YMD'
     599        self.db.set_parameter('datestyle', changed_datestyle)
     600        r = self.db.get_parameter('datestyle')
     601        self.assertEqual(r, changed_datestyle)
     602        con = self.db.db
     603        q = con.query("show datestyle")
     604        self.db.reopen()
     605        r = q.getresult()[0][0]
     606        self.assertEqual(r, changed_datestyle)
     607        self.assertRaises(TypeError, getattr, con, 'query')
     608        r = self.db.get_parameter('datestyle')
     609        self.assertEqual(r, default_datestyle)
    558610
    559611    def testQuery(self):
Note: See TracChangeset for help on using the changeset viewer.