simpler temp dir cleanup

This commit is contained in:
Skip Montanaro 2003-05-17 02:54:11 +00:00
parent de9a0d3158
commit ab0053aa13

View file

@ -11,6 +11,7 @@
import whichdb
import os
import tempfile
import shutil
class Bsddb185Tests(unittest.TestCase):
@ -28,15 +29,12 @@ def test_anydbm_create(self):
# 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)