summaryrefslogtreecommitdiffstats
path: root/Lib/test/test_dbm.py
diff options
context:
space:
mode:
authorMiss Islington (bot) <31488909+miss-islington@users.noreply.github.com>2018-04-29 11:50:26 (GMT)
committerSerhiy Storchaka <storchaka@gmail.com>2018-04-29 11:50:26 (GMT)
commitee95feb69d937149bef3d245e87df1aef412b7fc (patch)
treee006b3c754dff27577a127ec85a5f6fda873f9db /Lib/test/test_dbm.py
parent1487c37e7e85e7221adc3030f0f7de023cc99ced (diff)
downloadcpython-ee95feb69d937149bef3d245e87df1aef412b7fc.zip
cpython-ee95feb69d937149bef3d245e87df1aef412b7fc.tar.gz
cpython-ee95feb69d937149bef3d245e87df1aef412b7fc.tar.bz2
[3.7] bpo-33383: Fix crash in get() of the dbm.ndbm database object. (GH-6630) (GH-6631)
(cherry picked from commit 2e38cc39330bd7f3003652869b644110a97a78d8) Co-authored-by: Serhiy Storchaka <storchaka@gmail.com>
Diffstat (limited to 'Lib/test/test_dbm.py')
-rw-r--r--Lib/test/test_dbm.py9
1 files changed, 9 insertions, 0 deletions
diff --git a/Lib/test/test_dbm.py b/Lib/test/test_dbm.py
index f0a428d..fb89807 100644
--- a/Lib/test/test_dbm.py
+++ b/Lib/test/test_dbm.py
@@ -86,12 +86,21 @@ class AnyDBMTestCase:
f = dbm.open(_fname, 'c')
self._dict['g'] = f[b'g'] = b"indented"
self.read_helper(f)
+ # setdefault() works as in the dict interface
+ self.assertEqual(f.setdefault(b'xxx', b'foo'), b'foo')
+ self.assertEqual(f[b'xxx'], b'foo')
f.close()
def test_anydbm_read(self):
self.init_db()
f = dbm.open(_fname, 'r')
self.read_helper(f)
+ # get() works as in the dict interface
+ self.assertEqual(f.get(b'a'), self._dict['a'])
+ self.assertEqual(f.get(b'xxx', b'foo'), b'foo')
+ self.assertIsNone(f.get(b'xxx'))
+ with self.assertRaises(KeyError):
+ f[b'xxx']
f.close()
def test_anydbm_keys(self):