mirror of
https://github.com/internetarchive/warcprox.git
synced 2025-01-18 13:22:09 +01:00
add dubmdbm fixture and recognition test
This commit is contained in:
parent
41836c86f8
commit
1ea6eb5b9a
@ -34,6 +34,23 @@ def make_gdbm_test_db(request):
|
|||||||
request.addfinalizer(delete_test_dumbdbm)
|
request.addfinalizer(delete_test_dumbdbm)
|
||||||
return db_name
|
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):
|
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"
|
print "runing assert_gdbm_db_is_created_and_correctly_identified with gdbm test file"
|
||||||
assert whichdb(make_gdbm_test_db) == "dbm.gdbm" or "gdbm"
|
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")
|
db = gdbm.open(make_gdbm_test_db, "r")
|
||||||
assert len(db.keys()) == 2
|
assert len(db.keys()) == 2
|
||||||
assert db.has_key('very first key')
|
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"
|
Loading…
x
Reference in New Issue
Block a user