Importing test suite from bsddb3 3.4.0 (with modifications).
diff --git a/Lib/bsddb/test/test_dbobj.py b/Lib/bsddb/test/test_dbobj.py
new file mode 100644
index 0000000..ee700e3
--- /dev/null
+++ b/Lib/bsddb/test/test_dbobj.py
@@ -0,0 +1,72 @@
+
+import sys, os, string
+import unittest
+import glob
+
+from bsddb import db, dbobj
+
+
+#----------------------------------------------------------------------
+
+class dbobjTestCase(unittest.TestCase):
+ """Verify that dbobj.DB and dbobj.DBEnv work properly"""
+ db_home = 'db_home'
+ db_name = 'test-dbobj.db'
+
+ def setUp(self):
+ homeDir = os.path.join(os.path.dirname(sys.argv[0]), 'db_home')
+ self.homeDir = homeDir
+ try: os.mkdir(homeDir)
+ except os.error: pass
+
+ def tearDown(self):
+ if hasattr(self, 'db'):
+ del self.db
+ if hasattr(self, 'env'):
+ del self.env
+ files = glob.glob(os.path.join(self.homeDir, '*'))
+ for file in files:
+ os.remove(file)
+
+ def test01_both(self):
+ class TestDBEnv(dbobj.DBEnv): pass
+ class TestDB(dbobj.DB):
+ def put(self, key, *args, **kwargs):
+ key = string.upper(key)
+ # call our parent classes put method with an upper case key
+ return apply(dbobj.DB.put, (self, key) + args, kwargs)
+ self.env = TestDBEnv()
+ self.env.open(self.db_home, db.DB_CREATE | db.DB_INIT_MPOOL)
+ self.db = TestDB(self.env)
+ self.db.open(self.db_name, db.DB_HASH, db.DB_CREATE)
+ self.db.put('spam', 'eggs')
+ assert self.db.get('spam') == None, "overridden dbobj.DB.put() method failed [1]"
+ assert self.db.get('SPAM') == 'eggs', "overridden dbobj.DB.put() method failed [2]"
+ self.db.close()
+ self.env.close()
+
+ def test02_dbobj_dict_interface(self):
+ self.env = dbobj.DBEnv()
+ self.env.open(self.db_home, db.DB_CREATE | db.DB_INIT_MPOOL)
+ self.db = dbobj.DB(self.env)
+ self.db.open(self.db_name+'02', db.DB_HASH, db.DB_CREATE)
+ # __setitem__
+ self.db['spam'] = 'eggs'
+ # __len__
+ assert len(self.db) == 1
+ # __getitem__
+ assert self.db['spam'] == 'eggs'
+ # __del__
+ del self.db['spam']
+ assert self.db.get('spam') == None, "dbobj __del__ failed"
+ self.db.close()
+ self.env.close()
+
+#----------------------------------------------------------------------
+
+def suite():
+ return unittest.makeSuite(dbobjTestCase)
+
+if __name__ == '__main__':
+ unittest.main( defaultTest='suite' )
+