diff --git a/tests/test_dump-anydbm.py b/tests/test_dump-anydbm.py index 5ea245a..b1923d5 100644 --- a/tests/test_dump-anydbm.py +++ b/tests/test_dump-anydbm.py @@ -34,6 +34,23 @@ def make_gdbm_test_db(request): request.addfinalizer(delete_test_dumbdbm) return db_name +@pytest.fixture(scope = "function") +def make_dumbdbm_test_db(request): + db_name ="test_dumbdbm" + print "creating", db_name + test_db = dumb.open(db_name, "n") + test_db['very first key'] = 'very first value' + test_db['second key'] = 'second value' + test_db.close() + def delete_test_dumbdbm(): + print "deleting", db_name + os.remove(db_name+".dir") + os.remove(db_name+".bak") + os.remove(db_name+".dat") + + request.addfinalizer(delete_test_dumbdbm) + return db_name+".dir" + def test_assert_gdbm_db_is_created_and_correctly_identified(make_gdbm_test_db): print "runing assert_gdbm_db_is_created_and_correctly_identified with gdbm test file" assert whichdb(make_gdbm_test_db) == "dbm.gdbm" or "gdbm" @@ -43,4 +60,8 @@ def test_assert_reading_gdbm_correctly(make_gdbm_test_db): db = gdbm.open(make_gdbm_test_db, "r") assert len(db.keys()) == 2 assert db.has_key('very first key') - assert db['very first key'] == 'very first value' \ No newline at end of file + assert db['very first key'] == 'very first value' + +def test_assert_dumbdbm_db_is_created_and_correctly_identified(make_dumbdbm_test_db): + print "runing assert_dumbdbm_db_is_created_and_correctly_identified with gdbm test file" + assert whichdb(make_dumbdbm_test_db) == "dbm.dumb" or "dumbdbm" \ No newline at end of file