Update and move test_dbm

1 files changed, 90 insertions, 0 deletions

diff --git a/extra_tests/test_dbm.py b/extra_tests/test_dbm.py
new file mode 100644
index 0000000000..6873c7c197
--- /dev/null
+++ b/extra_tests/test_dbm.py
@@ -0,0 +1,90 @@
+import pytest
+
+import os
+
+dbm = pytest.importorskip('dbm')
+
+
+def test_get(tmpdir):
+    path = str(tmpdir.join('test_dbm_extra.test_get'))
+    d = dbm.open(path, 'c')
+    x = d.get("42")
+    assert x is None
+    d.close()
+
+def test_delitem(tmpdir):
+    path = str(tmpdir.join('test_dbm_extra.test_delitem'))
+    d = dbm.open(path, 'c')
+    with pytest.raises(KeyError):
+        del d['xyz']
+
+def test_nonstring(tmpdir):
+    path = str(tmpdir.join('test_dbm_extra.test_nonstring'))
+    d = dbm.open(path, 'c')
+    with pytest.raises(TypeError):
+        d[123] = 'xyz'
+    with pytest.raises(TypeError):
+        d['xyz'] = 123
+    with pytest.raises(TypeError):
+        d['xyz'] = None
+    with pytest.raises(TypeError):
+        del d[123]
+    with pytest.raises(TypeError):
+        d[123]
+    with pytest.raises(TypeError):
+        123 in d
+    with pytest.raises(TypeError):
+        d.has_key(123)
+    with pytest.raises(TypeError):
+        d.setdefault(123, 'xyz')
+    with pytest.raises(TypeError):
+        d.setdefault('xyz', 123)
+    with pytest.raises(TypeError):
+        d.get(123)
+    assert dict(d) == {}
+    d.setdefault('xyz', '123')
+    assert dict(d) == {'xyz': '123'}
+    d.close()
+
+def test_multiple_sets(tmpdir):
+    path = str(tmpdir.join('test_dbm_extra.test_multiple_sets'))
+    d = dbm.open(path, 'c')
+    d['xyz'] = '12'
+    d['xyz'] = '3'
+    d['xyz'] = '546'
+    assert dict(d) == {'xyz': '546'}
+    assert d['xyz'] == '546'
+
+@pytest.mark.skipif("'__pypy__' not in sys.modules")
+def test_extra():
+    with pytest.raises(TypeError):
+        dbm.datum(123)
+    with pytest.raises(TypeError):
+        dbm.datum(False)
+
+def test_null():
+    db = dbm.open('test', 'c')
+    db['1'] = 'a\x00b'
+    db.close()
+
+    db = dbm.open('test', 'r')
+    assert db['1'] == 'a\x00b'
+    db.close()
+
+def test_key_with_empty_value(tmpdir):
+    # this test fails on CPython too (at least on tannit), and the
+    # case shows up when gdbm is not installed and test_anydbm.py
+    # falls back dbm.
+    pytest.skip("test may fail on CPython too")
+    path = str(tmpdir.join('test_dbm_extra.test_key_with_empty_value'))
+    d = dbm.open(path, 'c')
+    assert 'key_with_empty_value' not in d
+    d['key_with_empty_value'] = ''
+    assert 'key_with_empty_value' in d
+    assert d['key_with_empty_value'] == ''
+    d.close()
+
+def test_unicode_filename(tmpdir):
+    path = str(tmpdir) + os.sep + u'test_dbm_extra.test_unicode_filename'
+    d = dbm.open(path, 'c')
+    d.close()