Lib/bsddb/test/test_dbobj.py

   1
   2 import os, string
   3 import unittest
   4
   5 from test_all import db, dbobj, test_support, get_new_environment_path, \
   6         get_new_database_path
   7
   8 #----------------------------------------------------------------------
   9
  10 class dbobjTestCase(unittest.TestCase):
  11     """Verify that dbobj.DB and dbobj.DBEnv work properly"""
  12     db_name = 'test-dbobj.db'
  13
  14     def setUp(self):
  15         self.homeDir = get_new_environment_path()
  16
  17     def tearDown(self):
  18         if hasattr(self, 'db'):
  19             del self.db
  20         if hasattr(self, 'env'):
  21             del self.env
  22         test_support.rmtree(self.homeDir)
  23
  24     def test01_both(self):
  25         class TestDBEnv(dbobj.DBEnv): pass
  26         class TestDB(dbobj.DB):
  27             def put(self, key, *args, **kwargs):
  28                 key = key.upper()
  29                 # call our parent classes put method with an upper case key
  30                 return dbobj.DB.put(self, key, *args, **kwargs)
  31         self.env = TestDBEnv()
  32         self.env.open(self.homeDir, db.DB_CREATE | db.DB_INIT_MPOOL)
  33         self.db = TestDB(self.env)
  34         self.db.open(self.db_name, db.DB_HASH, db.DB_CREATE)
  35         self.db.put('spam', 'eggs')
  36         self.assertEqual(self.db.get('spam'), None,
  37                "overridden dbobj.DB.put() method failed [1]")
  38         self.assertEqual(self.db.get('SPAM'), 'eggs',
  39                "overridden dbobj.DB.put() method failed [2]")
  40         self.db.close()
  41         self.env.close()
  42
  43     def test02_dbobj_dict_interface(self):
  44         self.env = dbobj.DBEnv()
  45         self.env.open(self.homeDir, db.DB_CREATE | db.DB_INIT_MPOOL)
  46         self.db = dbobj.DB(self.env)
  47         self.db.open(self.db_name+'02', db.DB_HASH, db.DB_CREATE)
  48         # __setitem__
  49         self.db['spam'] = 'eggs'
  50         # __len__
  51         self.assertEqual(len(self.db), 1)
  52         # __getitem__
  53         self.assertEqual(self.db['spam'], 'eggs')
  54         # __del__
  55         del self.db['spam']
  56         self.assertEqual(self.db.get('spam'), None, "dbobj __del__ failed")
  57         self.db.close()
  58         self.env.close()
  59
  60     def test03_dbobj_type_before_open(self):
  61         # Ensure this doesn't cause a segfault.
  62         self.assertRaises(db.DBInvalidArgError, db.DB().type)
  63
  64 #----------------------------------------------------------------------
  65
  66 def test_suite():
  67     return unittest.makeSuite(dbobjTestCase)
  68
  69 if __name__ == '__main__':
  70     unittest.main(defaultTest='test_suite')