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

take advantage of teardown of fixture to delete temp db file

Browse Source
This commit is contained in:
Kelsey Hawley 2013-12-17 14:27:14 -08:00
parent acc55ce266
commit 7991640291
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/test_dump-anydbm.py
View File

@ -2,6 +2,7 @@
#from warcprox.bin import dump-anydbm #from warcprox.bin import dump-anydbm
import pytest import pytest
import os
# will try as python 3 then default to python 2 modules # will try as python 3 then default to python 2 modules
try: try:
@ -16,14 +17,21 @@ except:
import dumbdbm as dumb import dumbdbm as dumb
from whichdb import whichdb from whichdb import whichdb
@pytest.fixture @pytest.fixture(scope = "function")
def make_gdbm_test_db(): 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['very first key'] = 'very first value' test_db['very first key'] = 'very first value'
test_db['second key'] = 'second value' test_db['second key'] = 'second value'
test_db.close() 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 return db_name
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):