Home | History | Annotate | Download | only in test 
      1 """
      2 TestCases for checking set_get_returns_none.
      3 """
      4 
      5 import os, string
      6 import unittest
      7 
      8 from test_all import db, verbose, get_new_database_path
      9 
     10 
     11 #----------------------------------------------------------------------
     12 
     13 class GetReturnsNoneTestCase(unittest.TestCase):
     14     def setUp(self):
     15         self.filename = get_new_database_path()
     16 
     17     def tearDown(self):
     18         try:
     19             os.remove(self.filename)
     20         except os.error:
     21             pass
     22 
     23 
     24     def test01_get_returns_none(self):
     25         d = db.DB()
     26         d.open(self.filename, db.DB_BTREE, db.DB_CREATE)
     27         d.set_get_returns_none(1)
     28 
     29         for x in string.letters:
     30             d.put(x, x * 40)
     31 
     32         data = d.get('bad key')
     33         self.assertEqual(data, None)
     34 
     35         data = d.get(string.letters[0])
     36         self.assertEqual(data, string.letters[0]*40)
     37 
     38         count = 0
     39         c = d.cursor()
     40         rec = c.first()
     41         while rec:
     42             count = count + 1
     43             rec = c.next()
     44 
     45         self.assertEqual(rec, None)
     46         self.assertEqual(count, len(string.letters))
     47 
     48         c.close()
     49         d.close()
     50 
     51 
     52     def test02_get_raises_exception(self):
     53         d = db.DB()
     54         d.open(self.filename, db.DB_BTREE, db.DB_CREATE)
     55         d.set_get_returns_none(0)
     56 
     57         for x in string.letters:
     58             d.put(x, x * 40)
     59 
     60         self.assertRaises(db.DBNotFoundError, d.get, 'bad key')
     61         self.assertRaises(KeyError, d.get, 'bad key')
     62 
     63         data = d.get(string.letters[0])
     64         self.assertEqual(data, string.letters[0]*40)
     65 
     66         count = 0
     67         exceptionHappened = 0
     68         c = d.cursor()
     69         rec = c.first()
     70         while rec:
     71             count = count + 1
     72             try:
     73                 rec = c.next()
     74             except db.DBNotFoundError:  # end of the records
     75                 exceptionHappened = 1
     76                 break
     77 
     78         self.assertNotEqual(rec, None)
     79         self.assertTrue(exceptionHappened)
     80         self.assertEqual(count, len(string.letters))
     81 
     82         c.close()
     83         d.close()
     84 
     85 #----------------------------------------------------------------------
     86 
     87 def test_suite():
     88     return unittest.makeSuite(GetReturnsNoneTestCase)
     89 
     90 
     91 if __name__ == '__main__':
     92     unittest.main(defaultTest='test_suite')
     93