From 4b0ab0ff728f8bd70581c84bf7c8d68c52448f80 Mon Sep 17 00:00:00 2001 From: Kelsey Hawley Date: Thu, 2 Jan 2014 16:29:15 -0800 Subject: [PATCH] updated file to PEP 8, as editor was complaining, and tabs are generally bad --- warcprox/tests/test_dump-anydbm.py | 209 +++++++++++++++-------------- 1 file changed, 109 insertions(+), 100 deletions(-) diff --git a/warcprox/tests/test_dump-anydbm.py b/warcprox/tests/test_dump-anydbm.py index b23fc44..fbe9a96 100644 --- a/warcprox/tests/test_dump-anydbm.py +++ b/warcprox/tests/test_dump-anydbm.py @@ -5,28 +5,28 @@ import os import subprocess # to access the script from shell # will try as python 3 then default to python 2 modules -try: - import dbm - from dbm import ndbm - from dbm import gnu as gdbm - from dbm import dumb +try: + import dbm + from dbm import ndbm + from dbm import gnu as gdbm + from dbm import dumb - whichdb = dbm.whichdb + whichdb = dbm.whichdb - ndbm_type = "dbm.ndbm" - gdbm_type = "dbm.gnu" - dumb_type = "dbm.dumb" + ndbm_type = "dbm.ndbm" + gdbm_type = "dbm.gnu" + dumb_type = "dbm.dumb" except: - import dbm as ndbm - import gdbm - import dumbdbm as dumb + import dbm as ndbm + import gdbm + import dumbdbm as dumb - from whichdb import whichdb + from whichdb import whichdb - ndbm_type = "dbm" - gdbm_type = "gdbm" - dumb_type = "dumbdbm" + ndbm_type = "dbm" + gdbm_type = "gdbm" + dumb_type = "dumbdbm" #global settings key1 = 'very first key' @@ -35,111 +35,120 @@ val1 = 'very first value' val2 = 'second value' dump_anydbm = "dump-anydbm" -@pytest.fixture(scope = "function") + +@pytest.fixture(scope="function") def make_gdbm_test_db(request): - db_name ="test_gdbm" - print("creating", db_name) - test_db = gdbm.open(db_name, "n") - test_db[key1] = val1 - test_db[key2] = val2 - test_db.close() - def delete_test_dumbdbm(): - print("deleting", db_name) - os.remove(db_name) + db_name = "test_gdbm" + print("creating", db_name) + test_db = gdbm.open(db_name, "n") + test_db[key1] = val1 + test_db[key2] = val2 + test_db.close() - request.addfinalizer(delete_test_dumbdbm) - return db_name + def delete_test_dumbdbm(): + print("deleting", db_name) + os.remove(db_name) -@pytest.fixture(scope = "function") + request.addfinalizer(delete_test_dumbdbm) + return db_name + + +@pytest.fixture(scope="function") def make_ndbm_test_db(request): - db_name = "test_ndbm" - print("creating", db_name) - test_db = ndbm.open(db_name, "n") - test_db[key1] = val1 - test_db[key2] = val2 - test_db.close() - def delete_test_ndbm(): - print("deleting", db_name) - os.remove(db_name+".db") + db_name = "test_ndbm" + print("creating", db_name) + test_db = ndbm.open(db_name, "n") + test_db[key1] = val1 + test_db[key2] = val2 + test_db.close() - request.addfinalizer(delete_test_ndbm) - return db_name + def delete_test_ndbm(): + print("deleting", db_name) + os.remove(db_name + ".db") -@pytest.fixture(scope = "function") + request.addfinalizer(delete_test_ndbm) + 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[key1] = val1 - test_db[key2] = val2 - 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") + db_name = "test_dumbdbm" + print("creating", db_name) + test_db = dumb.open(db_name, "n") + test_db[key1] = val1 + test_db[key2] = val2 + 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 - request.addfinalizer(delete_test_dumbdbm) - return db_name def test_dumpanydbm_identify_gdbm(make_gdbm_test_db): - print("running test_dumpanydbm_identify_gdbm") - output = subprocess.check_output([dump_anydbm, make_gdbm_test_db]) - output = output.decode(encoding = 'UTF-8').strip().split("\n") - assert len(output) == 3 # 2 keys plus whichdb line + print("running test_dumpanydbm_identify_gdbm") + output = subprocess.check_output([dump_anydbm, make_gdbm_test_db]) + output = output.decode(encoding='UTF-8').strip().split("\n") + assert len(output) == 3 # 2 keys plus whichdb line - # split on space, then grab 4th word, which is db type - which = output[0].split(' ')[3] - print(which) - assert which == gdbm_type + # split on space, then grab 4th word, which is db type + which = output[0].split(' ')[3] + print(which) + assert which == gdbm_type - #split remaining lines on ':' that separates key & value - db_dump_first_pair = output[1].split(':') - assert db_dump_first_pair[0] == key1 - assert db_dump_first_pair[1] == val1 + #split remaining lines on ':' that separates key & value + db_dump_first_pair = output[1].split(':') + assert db_dump_first_pair[0] == key1 + assert db_dump_first_pair[1] == val1 + + db_dump_second_pair = output[2].split(':') + assert db_dump_second_pair[0] == key2 + assert db_dump_second_pair[1] == val2 - db_dump_second_pair = output[2].split(':') - assert db_dump_second_pair[0] == key2 - assert db_dump_second_pair[1] == val2 def test_dumpanydbm_identify_ndbm(make_ndbm_test_db): - print("running test_dumpanydbm_identify_ndbm") - output = subprocess.check_output([dump_anydbm, make_ndbm_test_db]) - output = output.decode(encoding = 'UTF-8').strip().split("\n") + print("running test_dumpanydbm_identify_ndbm") + output = subprocess.check_output([dump_anydbm, make_ndbm_test_db]) + 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 - which = output[0].split(' ')[3] - print(which) - assert which == ndbm_type + # split on space, then grab 4th word, which is db type + which = output[0].split(' ')[3] + print(which) + assert which == ndbm_type - #split remaining lines on ':' that separates key & value - db_dump_first_pair = output[1].split(':') - assert db_dump_first_pair[0] == key1 - assert db_dump_first_pair[1] == val1 + #split remaining lines on ':' that separates key & value + db_dump_first_pair = output[1].split(':') + assert db_dump_first_pair[0] == key1 + assert db_dump_first_pair[1] == val1 + + db_dump_second_pair = output[2].split(':') + assert db_dump_second_pair[0] == key2 + assert db_dump_second_pair[1] == val2 - db_dump_second_pair = output[2].split(':') - assert db_dump_second_pair[0] == key2 - assert db_dump_second_pair[1] == val2 def test_dumpanydbm_identify_dumbdbm(make_dumbdbm_test_db): - print("running test_dumpanydbm_identify_dumbdbm") - output = subprocess.check_output([dump_anydbm, make_dumbdbm_test_db]) - output = output.decode(encoding = 'UTF-8').strip().split("\n") - assert len(output) == 3 # 2 keys plus whichdb line - print(output) + print("running test_dumpanydbm_identify_dumbdbm") + output = subprocess.check_output([dump_anydbm, make_dumbdbm_test_db]) + output = output.decode(encoding='UTF-8').strip().split("\n") + assert len(output) == 3 # 2 keys plus whichdb line + print(output) - # split on space, then grab 4th word, which is db type - which = output[0].split(' ')[3] - print(which) - assert which == dumb_type + # split on space, then grab 4th word, which is db type + which = output[0].split(' ')[3] + print(which) + assert which == dumb_type - #split remaining lines on ':' that separates key & value - db_dump_first_pair = output[1].split(':') - assert db_dump_first_pair[0] == key1 - assert db_dump_first_pair[1] == val1 + #split remaining lines on ':' that separates key & value + db_dump_first_pair = output[1].split(':') + assert db_dump_first_pair[0] == key1 + assert db_dump_first_pair[1] == val1 - db_dump_second_pair = output[2].split(':') - assert db_dump_second_pair[0] == key2 - assert db_dump_second_pair[1] == val2 + db_dump_second_pair = output[2].split(':') + assert db_dump_second_pair[0] == key2 + assert db_dump_second_pair[1] == val2