warcprox/tests/test_dump-anydbm.py

155 lines
5.3 KiB
Python
Raw Permalink Normal View History

#!/usr/bin/env python
2016-04-06 19:37:55 -07:00
#
# tests/test_dump-anydbm.py - tests for dump-anydbm
#
# Copyright (C) 2013-2016 Internet Archive
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
# USA.
#
import pytest
import os
import tempfile
import subprocess # to access the script from shell
import sys
import glob
import distutils
# 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
whichdb = dbm.whichdb
2014-01-06 17:22:49 -08:00
ndbm_type = b"dbm.ndbm"
gdbm_type = b"dbm.gnu"
dumb_type = b"dbm.dumb"
except:
import dbm as ndbm
import gdbm
import dumbdbm as dumb
from whichdb import whichdb
2014-01-06 17:22:49 -08:00
ndbm_type = b"dbm"
gdbm_type = b"gdbm"
dumb_type = b"dumbdbm"
#global settings
key1 = 'very first key'
key2 = 'second key'
val1 = 'very first value'
val2 = 'second value'
py = sys.executable
dump_anydbm_loc = distutils.spawn.find_executable("dump-anydbm")
@pytest.fixture(scope="function")
2014-01-06 17:22:49 -08:00
def gdbm_test_db(request):
temp_file = tempfile.NamedTemporaryFile(delete=False)
print("creating test gdbm file {}".format(temp_file.name))
test_db = gdbm.open(temp_file.name, "n")
test_db[key1] = val1
test_db[key2] = val2
test_db.close()
2014-01-06 17:22:49 -08:00
def delete_gdbm_test_db():
temp_file.close()
for f in glob.glob("{}*".format(temp_file.name)):
print("deleting test gdbm file {}".format(f))
os.remove(f)
2014-01-06 17:22:49 -08:00
request.addfinalizer(delete_gdbm_test_db)
return temp_file.name
@pytest.fixture(scope="function")
2014-01-06 17:22:49 -08:00
def ndbm_test_db(request):
temp_file = tempfile.NamedTemporaryFile(delete=False)
test_db = ndbm.open(temp_file.name, "n")
test_db[key1] = val1
test_db[key2] = val2
test_db.close()
def delete_test_ndbm():
temp_file.close()
for f in glob.glob("{}*".format(temp_file.name)):
print("deleting test ndbm file {}".format(f))
os.remove(f)
request.addfinalizer(delete_test_ndbm)
return temp_file.name
@pytest.fixture(scope="function")
2014-01-06 17:22:49 -08:00
def dumbdbm_test_db(request):
temp_file = tempfile.NamedTemporaryFile(delete=False)
print("creating test dumbdbm file {}".format(temp_file.name))
test_db = dumb.open(temp_file.name, "n")
test_db[key1] = val1
test_db[key2] = val2
test_db.close()
2014-01-06 17:22:49 -08:00
def delete_dumbdbm_test_db():
temp_file.close()
for f in glob.glob("{}*".format(temp_file.name)):
print("deleting test dumbdbm file {}".format(f))
os.remove(f)
2014-01-06 17:22:49 -08:00
request.addfinalizer(delete_dumbdbm_test_db)
return temp_file.name
2014-01-06 17:22:49 -08:00
def test_dumpanydbm_identify_gdbm(gdbm_test_db):
print("running test_dumpanydbm_identify_gdbm")
output = subprocess.check_output([py, dump_anydbm_loc, gdbm_test_db])
print("script printout: ")
print(output)
print("check_one: ")
print(gdbm_test_db.encode(encoding='UTF-8') + b' is a ' + gdbm_type + b' db\nvery first key:very first value\nsecond key:second value\n')
2014-01-06 17:22:49 -08:00
assert (output == gdbm_test_db.encode(encoding='UTF-8') + b' is a ' + gdbm_type + b' db\nvery first key:very first value\nsecond key:second value\n' or
output == gdbm_test_db.encode(encoding='UTF-8') + b' is a ' + gdbm_type + b' db\nsecond key:second value\nvery first key:very first value\n')
2014-01-06 17:22:49 -08:00
def test_dumpanydbm_identify_ndbm(ndbm_test_db):
print("running test_dumpanydbm_identify_ndbm")
output = subprocess.check_output([py, dump_anydbm_loc, ndbm_test_db])
print("script printout: ")
print(output)
print("check_one: ")
print(ndbm_test_db.encode(encoding='UTF-8') + b' is a ' + ndbm_type + b' db\nvery first key:very first value\nsecond key:second value\n')
2014-01-06 17:22:49 -08:00
assert (output == ndbm_test_db.encode(encoding='UTF-8') + b' is a ' + ndbm_type + b' db\nvery first key:very first value\nsecond key:second value\n' or
output == ndbm_test_db.encode(encoding='UTF-8') + b' is a ' + ndbm_type + b' db\nsecond key:second value\nvery first key:very first value\n')
2014-01-06 17:22:49 -08:00
def test_dumpanydbm_identify_dumbdbm(dumbdbm_test_db):
print("running test_dumpanydbm_identify_dumbdbm")
output = subprocess.check_output([py, dump_anydbm_loc, dumbdbm_test_db])
print("script printout: ")
print(output)
print("check_one: ")
print(dumbdbm_test_db.encode(encoding='UTF-8') + b' is a ' + dumb_type + b' db\nvery first key:very first value\nsecond key:second value\n')
2014-01-06 17:22:49 -08:00
assert (output == dumbdbm_test_db.encode(encoding='UTF-8') + b' is a ' + dumb_type + b' db\nvery first key:very first value\nsecond key:second value\n' or
output == dumbdbm_test_db.encode(encoding='UTF-8') + b' is a ' + dumb_type + b' db\nsecond key:second value\nvery first key:very first value\n')