Brett Cannon | 065f7b8 | 2003-05-13 06:42:59 +0000 | [diff] [blame^] | 1 | """Tests for the bsddb185 module. |
Skip Montanaro | 46f4e79 | 2003-05-06 20:36:57 +0000 | [diff] [blame] | 2 | |
Brett Cannon | 065f7b8 | 2003-05-13 06:42:59 +0000 | [diff] [blame^] | 3 | The file 185test.db found in Lib/test/ is for testing purposes with this |
| 4 | testing suite. |
| 5 | |
| 6 | """ |
Skip Montanaro | 46f4e79 | 2003-05-06 20:36:57 +0000 | [diff] [blame] | 7 | from test.test_support import verbose, run_unittest, findfile |
| 8 | import unittest |
| 9 | import bsddb185 |
| 10 | import anydbm |
| 11 | import whichdb |
| 12 | import os |
| 13 | import tempfile |
| 14 | |
| 15 | class Bsddb185Tests(unittest.TestCase): |
Brett Cannon | 065f7b8 | 2003-05-13 06:42:59 +0000 | [diff] [blame^] | 16 | |
Skip Montanaro | 46f4e79 | 2003-05-06 20:36:57 +0000 | [diff] [blame] | 17 | def test_open_existing_hash(self): |
Brett Cannon | 065f7b8 | 2003-05-13 06:42:59 +0000 | [diff] [blame^] | 18 | # Verify we can open a file known to be a hash v2 file |
Skip Montanaro | 46f4e79 | 2003-05-06 20:36:57 +0000 | [diff] [blame] | 19 | db = bsddb185.hashopen(findfile("185test.db")) |
| 20 | self.assertEqual(db["1"], "1") |
| 21 | db.close() |
| 22 | |
| 23 | def test_whichdb(self): |
Brett Cannon | 065f7b8 | 2003-05-13 06:42:59 +0000 | [diff] [blame^] | 24 | # Verify that whichdb correctly sniffs the known hash v2 file |
Skip Montanaro | 46f4e79 | 2003-05-06 20:36:57 +0000 | [diff] [blame] | 25 | self.assertEqual(whichdb.whichdb(findfile("185test.db")), "bsddb185") |
| 26 | |
| 27 | def test_anydbm_create(self): |
Brett Cannon | 065f7b8 | 2003-05-13 06:42:59 +0000 | [diff] [blame^] | 28 | # Verify that anydbm.open does *not* create a bsddb185 file |
Skip Montanaro | 46f4e79 | 2003-05-06 20:36:57 +0000 | [diff] [blame] | 29 | tmpdir = tempfile.mkdtemp() |
| 30 | try: |
| 31 | try: |
| 32 | dbfile = os.path.join(tmpdir, "foo.db") |
Brett Cannon | 065f7b8 | 2003-05-13 06:42:59 +0000 | [diff] [blame^] | 33 | anydbm.open(os.path.splitext(dbfile)[0], "c").close() |
| 34 | ftype = whichdb.whichdb(dbfile) |
Skip Montanaro | 46f4e79 | 2003-05-06 20:36:57 +0000 | [diff] [blame] | 35 | self.assertNotEqual(ftype, "bsddb185") |
| 36 | finally: |
| 37 | os.unlink(dbfile) |
| 38 | finally: |
| 39 | os.rmdir(tmpdir) |
| 40 | |
| 41 | def test_main(): |
| 42 | run_unittest(Bsddb185Tests) |
| 43 | |
| 44 | if __name__ == "__main__": |
| 45 | test_main() |