mirror of
https://github.com/python/cpython.git
synced 2025-10-31 10:26:02 +00:00
6ba3329c27
Python 2.6 one, since the intention is to keep an unified 2.x/3.x codebase. The Python code is automatically translated using "2to3". Please, do not update this code in Python 3.0 by hand. Update the 2.6 one and then do "2to3".
92 lines
2.2 KiB
Python
92 lines
2.2 KiB
Python
"""
|
|
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 = next(c)
|
|
|
|
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 = next(c)
|
|
except db.DBNotFoundError: # end of the records
|
|
exceptionHappened = 1
|
|
break
|
|
|
|
self.assertNotEqual(rec, None)
|
|
self.assert_(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')
|