simpler temp dir cleanup

This commit is contained in:
Skip Montanaro 2003-05-17 02:54:11 +00:00
parent de9a0d3158
commit ab0053aa13
1 changed files with 6 additions and 8 deletions

View File

@ -11,6 +11,7 @@ import anydbm
import whichdb
import os
import tempfile
import shutil
class Bsddb185Tests(unittest.TestCase):
@ -28,15 +29,12 @@ class Bsddb185Tests(unittest.TestCase):
# Verify that anydbm.open does *not* create a bsddb185 file
tmpdir = tempfile.mkdtemp()
try:
try:
dbfile = os.path.join(tmpdir, "foo.db")
anydbm.open(os.path.splitext(dbfile)[0], "c").close()
ftype = whichdb.whichdb(dbfile)
self.assertNotEqual(ftype, "bsddb185")
finally:
os.unlink(dbfile)
dbfile = os.path.join(tmpdir, "foo.db")
anydbm.open(dbfile, "c").close()
ftype = whichdb.whichdb(dbfile)
self.assertNotEqual(ftype, "bsddb185")
finally:
os.rmdir(tmpdir)
shutil.rmtree(tmpdir)
def test_main():
run_unittest(Bsddb185Tests)