Don't leave *c files around after tests

This commit is contained in:
Cole Robinson 2013-04-23 17:17:15 -04:00
parent b2041f7399
commit 8d1ad359ee
2 changed files with 21 additions and 10 deletions

5
.gitignore vendored
View File

@ -6,11 +6,6 @@
/dist
po/virt-manager.pot
virt-clonec
virt-convertc
virt-imagec
virt-installc
/.coverage
/MANIFEST

View File

@ -30,11 +30,6 @@ import virtinst.cli
from tests import utils
virtinstall = imp.load_source("virtinstall", "virt-install")
virtimage = imp.load_source("virtimage", "virt-image")
virtclone = imp.load_source("virtclone", "virt-clone")
virtconvert = imp.load_source("virtconvert", "virt-convert")
os.environ["VIRTCONV_TEST_NO_DISK_CONVERSION"] = "1"
os.environ["LANG"] = "en_US.UTF-8"
@ -134,6 +129,27 @@ test_files = {
}
_cleanup_imports = []
def _import(name, path):
_cleanup_imports.append(path + "c")
return imp.load_source(name, path)
def _cleanup_imports_cb():
for f in _cleanup_imports:
if os.path.exists(f):
os.unlink(f)
atexit.register(_cleanup_imports_cb)
virtinstall = _import("virtinstall", "virt-install")
virtimage = _import("virtimage", "virt-image")
virtclone = _import("virtclone", "virt-clone")
virtconvert = _import("virtconvert", "virt-convert")
######################
# Test class helpers #
######################