mirror of
https://github.com/python/cpython.git
synced 2025-07-23 19:25:40 +00:00
41631e8f66
features in BerkeleyDB not exposed. notably: the DB_MPOOLFILE interface has not yet been wrapped in an object. Adds support for building and installing bsddb3 in python2.3 that has an older version of this module installed as bsddb without conflicts. The pybsddb.sf.net build/packaged version of the module uses a dynamicly loadable module called _pybsddb rather than _bsddb.
78 lines
2.4 KiB
Python
78 lines
2.4 KiB
Python
|
|
import sys, os, string
|
|
import unittest
|
|
import glob
|
|
|
|
try:
|
|
# For Pythons w/distutils pybsddb
|
|
from bsddb3 import db, dbobj
|
|
except ImportError:
|
|
# For Python 2.3
|
|
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.homeDir, 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.homeDir, 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 test_suite():
|
|
return unittest.makeSuite(dbobjTestCase)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main(defaultTest='test_suite')
|