Commit | Line | Data |
---|---|---|
86530b38 AT |
1 | """Miscellaneous bsddb module test cases |
2 | """ | |
3 | ||
4 | import os | |
5 | import sys | |
6 | import unittest | |
7 | ||
8 | try: | |
9 | # For Pythons w/distutils pybsddb | |
10 | from bsddb3 import db, dbshelve | |
11 | except ImportError: | |
12 | # For Python 2.3 | |
13 | from bsddb import db, dbshelve | |
14 | ||
15 | #---------------------------------------------------------------------- | |
16 | ||
17 | class MiscTestCase(unittest.TestCase): | |
18 | def setUp(self): | |
19 | self.filename = self.__class__.__name__ + '.db' | |
20 | homeDir = os.path.join(os.path.dirname(sys.argv[0]), 'db_home') | |
21 | self.homeDir = homeDir | |
22 | try: | |
23 | os.mkdir(homeDir) | |
24 | except OSError: | |
25 | pass | |
26 | ||
27 | def tearDown(self): | |
28 | try: | |
29 | os.remove(self.filename) | |
30 | except OSError: | |
31 | pass | |
32 | import glob | |
33 | files = glob.glob(os.path.join(self.homeDir, '*')) | |
34 | for file in files: | |
35 | os.remove(file) | |
36 | ||
37 | def test01_badpointer(self): | |
38 | dbs = dbshelve.open(self.filename) | |
39 | dbs.close() | |
40 | self.assertRaises(db.DBError, dbs.get, "foo") | |
41 | ||
42 | def test02_db_home(self): | |
43 | env = db.DBEnv() | |
44 | # check for crash fixed when db_home is used before open() | |
45 | assert env.db_home is None | |
46 | env.open(self.homeDir, db.DB_CREATE) | |
47 | assert self.homeDir == env.db_home | |
48 | ||
49 | ||
50 | #---------------------------------------------------------------------- | |
51 | ||
52 | ||
53 | def test_suite(): | |
54 | return unittest.makeSuite(MiscTestCase) | |
55 | ||
56 | ||
57 | if __name__ == '__main__': | |
58 | unittest.main(defaultTest='test_suite') |