1 files changed, 92 insertions, 0 deletions
diff --git a/lib/python2.7/bsddb/test/test_get_none.py b/lib/python2.7/bsddb/test/test_get_none.py
new file mode 100644
index 0000000..8763b54
--- /dev/null
+++ b/lib/python2.7/bsddb/test/test_get_none.py
@@ -0,0 +1,92 @@
+"""
+TestCases for checking set_get_returns_none.
+"""
+
+import os, string
+import unittest
+
+from test_all import db, verbose, get_new_database_path
+
+
+#----------------------------------------------------------------------
+
+class GetReturnsNoneTestCase(unittest.TestCase):
+    def setUp(self):
+        self.filename = get_new_database_path()
+
+    def tearDown(self):
+        try:
+            os.remove(self.filename)
+        except os.error:
+            pass
+
+
+    def test01_get_returns_none(self):
+        d = db.DB()
+        d.open(self.filename, db.DB_BTREE, db.DB_CREATE)
+        d.set_get_returns_none(1)
+
+        for x in string.letters:
+            d.put(x, x * 40)
+
+        data = d.get('bad key')
+        self.assertEqual(data, None)
+
+        data = d.get(string.letters[0])
+        self.assertEqual(data, string.letters[0]*40)
+
+        count = 0
+        c = d.cursor()
+        rec = c.first()
+        while rec:
+            count = count + 1
+            rec = c.next()
+
+        self.assertEqual(rec, None)
+        self.assertEqual(count, len(string.letters))
+
+        c.close()
+        d.close()
+
+
+    def test02_get_raises_exception(self):
+        d = db.DB()
+        d.open(self.filename, db.DB_BTREE, db.DB_CREATE)
+        d.set_get_returns_none(0)
+
+        for x in string.letters:
+            d.put(x, x * 40)
+
+        self.assertRaises(db.DBNotFoundError, d.get, 'bad key')
+        self.assertRaises(KeyError, d.get, 'bad key')
+
+        data = d.get(string.letters[0])
+        self.assertEqual(data, string.letters[0]*40)
+
+        count = 0
+        exceptionHappened = 0
+        c = d.cursor()
+        rec = c.first()
+        while rec:
+            count = count + 1
+            try:
+                rec = c.next()
+            except db.DBNotFoundError:  # end of the records
+                exceptionHappened = 1
+                break
+
+        self.assertNotEqual(rec, None)
+        self.assertTrue(exceptionHappened)
+        self.assertEqual(count, len(string.letters))
+
+        c.close()
+        d.close()
+
+#----------------------------------------------------------------------
+
+def test_suite():
+    return unittest.makeSuite(GetReturnsNoneTestCase)
+
+
+if __name__ == '__main__':
+    unittest.main(defaultTest='test_suite')