TestCases for checking set_get_returns_none.
from pprint
import pprint
# For Pythons w/distutils pybsddb
from test_all
import verbose
#----------------------------------------------------------------------
class GetReturnsNoneTestCase(unittest
.TestCase
):
self
.filename
= tempfile
.mktemp()
def test01_get_returns_none(self
):
d
.open(self
.filename
, db
.DB_BTREE
, db
.DB_CREATE
)
d
.set_get_returns_none(1)
def test02_get_raises_exception(self
):
d
.open(self
.filename
, db
.DB_BTREE
, db
.DB_CREATE
)
d
.set_get_returns_none(0)
self
.assertRaises(db
.DBNotFoundError
, d
.get
, 'bad key')
self
.assertRaises(KeyError, d
.get
, 'bad key')
except db
.DBNotFoundError
: # end of the records
#----------------------------------------------------------------------
return unittest
.makeSuite(GetReturnsNoneTestCase
)
if __name__
== '__main__':
unittest
.main(defaultTest
='test_suite')