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

corrected the import of python 3 modules. Updated print statements to function form to work correctly with python 2 & 3. Also decoded the subprocess output into unicode string for splicing and processing of keys

Browse Source
This commit is contained in:
Kelsey Hawley 2013-12-20 13:51:34 -08:00
parent 48a2db509f
commit 6a129fed34
1 changed files with 24 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
tests/test_dump-anydbm.py
View File

@ -7,14 +7,14 @@ import subprocess # to access the script from shell
# will try as python 3 then default to python 2 modules # will try as python 3 then default to python 2 modules
try: try:
import dbm import dbm
ndbm = dbm.ndbm from dbm import ndbm
gdbm = dbm.gdbm from dbm import gnu as gdbm
dumb = dbm.dumb from dbm import dumb
whichdb = dbm.whichdb whichdb = dbm.whichdb
ndbm_type = "dbm.ndbm" ndbm_type = "dbm.ndbm"
gdbm_type = "dbm.gdbm" gdbm_type = "dbm.gnu"
dumb_type = "dbm.dumb" dumb_type = "dbm.dumb"
except: except:
@ -33,17 +33,18 @@ key1 = 'very first key'
key2 = 'second key' key2 = 'second key'
val1 = 'very first value' val1 = 'very first value'
val2 = 'second value' val2 = 'second value'
dump_anydbm = "dump-anydbm"
@pytest.fixture(scope = "function") @pytest.fixture(scope = "function")
def make_gdbm_test_db(request): def make_gdbm_test_db(request):
db_name ="test_gdbm" db_name ="test_gdbm"
print "creating", db_name print("creating", db_name)
test_db = gdbm.open(db_name, "n") test_db = gdbm.open(db_name, "n")
test_db[key1] = val1 test_db[key1] = val1
test_db[key2] = val2 test_db[key2] = val2
test_db.close() test_db.close()
def delete_test_dumbdbm(): def delete_test_dumbdbm():
print "deleting", db_name print("deleting", db_name)
os.remove(db_name) os.remove(db_name)
request.addfinalizer(delete_test_dumbdbm) request.addfinalizer(delete_test_dumbdbm)
@ -52,13 +53,13 @@ def make_gdbm_test_db(request):
@pytest.fixture(scope = "function") @pytest.fixture(scope = "function")
def make_ndbm_test_db(request): def make_ndbm_test_db(request):
db_name = "test_ndbm" db_name = "test_ndbm"
print "creating", db_name print("creating", db_name)
test_db = ndbm.open(db_name, "n") test_db = ndbm.open(db_name, "n")
test_db[key1] = val1 test_db[key1] = val1
test_db[key2] = val2 test_db[key2] = val2
test_db.close() test_db.close()
def delete_test_ndbm(): def delete_test_ndbm():
print "deleting", db_name print("deleting", db_name)
os.remove(db_name+".db") os.remove(db_name+".db")
request.addfinalizer(delete_test_ndbm) request.addfinalizer(delete_test_ndbm)
@ -67,13 +68,13 @@ def make_ndbm_test_db(request):
@pytest.fixture(scope = "function") @pytest.fixture(scope = "function")
def make_dumbdbm_test_db(request): def make_dumbdbm_test_db(request):
db_name ="test_dumbdbm" db_name ="test_dumbdbm"
print "creating", db_name print("creating", db_name)
test_db = dumb.open(db_name, "n") test_db = dumb.open(db_name, "n")
test_db[key1] = val1 test_db[key1] = val1
test_db[key2] = val2 test_db[key2] = val2
test_db.close() test_db.close()
def delete_test_dumbdbm(): def delete_test_dumbdbm():
print "deleting", db_name print("deleting", db_name)
os.remove(db_name+".dir") os.remove(db_name+".dir")
os.remove(db_name+".bak") os.remove(db_name+".bak")
os.remove(db_name+".dat") os.remove(db_name+".dat")
@ -82,14 +83,14 @@ def make_dumbdbm_test_db(request):
return db_name return db_name
def test_dumpanydbm_identify_gdbm(make_gdbm_test_db): def test_dumpanydbm_identify_gdbm(make_gdbm_test_db):
print "running test_dumpanydbm_identify_gdbm" print("running test_dumpanydbm_identify_gdbm")
output = subprocess.check_output(["dump-anydbm", make_gdbm_test_db]) output = subprocess.check_output([dump_anydbm, make_gdbm_test_db])
output = output.strip().split("\n") output = output.decode(encoding = 'UTF-8').strip().split("\n")
assert len(output) == 3 # 2 keys plus whichdb line assert len(output) == 3 # 2 keys plus whichdb line
# split on space, then grab 4th word, which is db type # split on space, then grab 4th word, which is db type
which = output[0].split(' ')[3] which = output[0].split(' ')[3]
print which print(which)
assert which == gdbm_type assert which == gdbm_type
#split remaining lines on ':' that separates key & value #split remaining lines on ':' that separates key & value
@ -102,15 +103,15 @@ def test_dumpanydbm_identify_gdbm(make_gdbm_test_db):
assert db_dump_second_pair[1] == val2 assert db_dump_second_pair[1] == val2
def test_dumpanydbm_identify_ndbm(make_ndbm_test_db): def test_dumpanydbm_identify_ndbm(make_ndbm_test_db):
print "running test_dumpanydbm_identify_ndbm" print("running test_dumpanydbm_identify_ndbm")
output = subprocess.check_output(["dump-anydbm", make_ndbm_test_db]) output = subprocess.check_output([dump_anydbm, make_ndbm_test_db])
output = output.strip().split("\n") output = output.decode(encoding = 'UTF-8').strip().split("\n")
assert len(output) == 3 # 2 keys plus whichdb line assert len(output) == 3 # 2 keys plus whichdb line
# split on space, then grab 4th word, which is db type # split on space, then grab 4th word, which is db type
which = output[0].split(' ')[3] which = output[0].split(' ')[3]
print which print(which)
assert which == ndbm_type assert which == ndbm_type
#split remaining lines on ':' that separates key & value #split remaining lines on ':' that separates key & value
@ -123,14 +124,15 @@ def test_dumpanydbm_identify_ndbm(make_ndbm_test_db):
assert db_dump_second_pair[1] == val2 assert db_dump_second_pair[1] == val2
def test_dumpanydbm_identify_dumbdbm(make_dumbdbm_test_db): def test_dumpanydbm_identify_dumbdbm(make_dumbdbm_test_db):
print "running test_dumpanydbm_identify_dumbdbm" print("running test_dumpanydbm_identify_dumbdbm")
output = subprocess.check_output(["dump-anydbm", make_dumbdbm_test_db]) output = subprocess.check_output([dump_anydbm, make_dumbdbm_test_db])
output = output.strip().split("\n") output = output.decode(encoding = 'UTF-8').strip().split("\n")
assert len(output) == 3 # 2 keys plus whichdb line assert len(output) == 3 # 2 keys plus whichdb line
print(output)
# split on space, then grab 4th word, which is db type # split on space, then grab 4th word, which is db type
which = output[0].split(' ')[3] which = output[0].split(' ')[3]
print which print(which)
assert which == dumb_type assert which == dumb_type
#split remaining lines on ':' that separates key & value #split remaining lines on ':' that separates key & value
@ -141,5 +143,3 @@ def test_dumpanydbm_identify_dumbdbm(make_dumbdbm_test_db):
db_dump_second_pair = output[2].split(':') db_dump_second_pair = output[2].split(':')
assert db_dump_second_pair[0] == key2 assert db_dump_second_pair[0] == key2
assert db_dump_second_pair[1] == val2 assert db_dump_second_pair[1] == val2