backup/warcprox
1
0
Fork 0
mirror of https://github.com/internetarchive/warcprox.git synced 2025-01-18 13:22:09 +01:00
Code Issues Projects Releases Activity

add dubmdbm fixture and recognition test

Browse Source
This commit is contained in:
Kelsey Hawley 2013-12-17 14:33:31 -08:00
parent 41836c86f8
commit 1ea6eb5b9a
1 changed files with 22 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
tests/test_dump-anydbm.py
View File

@ -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'
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"