tests: Fix coverage after recent import tweaks

crobinso · crobinso · commit 3939b452ce23 · 2013-04-23T16:41:29.000-04:00
We need to cleanup the c file at exit.
diff --git a/.gitignore b/.gitignore
@@ -4,4 +4,3 @@ MANIFEST
 dist
 build
 .coverage
-bin/bugzillac
diff --git a/tests/__init__.py b/tests/__init__.py
@@ -1,12 +1,33 @@
 
+import atexit
 import commands
 import difflib
+import imp
 import os
 import shlex
 import sys
 import StringIO
 
 
+_cleanup = []
+
+
+def _import(name, path):
+    _cleanup.append(path + "c")
+    return imp.load_source(name, path)
+
+
+def _cleanup_cb():
+    for f in _cleanup:
+        if os.path.exists(f):
+            os.unlink(f)
+
+
+atexit.register(_cleanup_cb)
+bugzillascript = _import("bugzillascript", "bin/bugzilla")
+
+
+
 def diff(orig, new):
     """
     Return a unified diff string between the passed strings
@@ -31,8 +52,6 @@ def clicomm(argv, bzinstance, returnmain=False, printcliout=False,
     """
     Run bin/bugzilla.main() directly with passed argv
     """
-    import imp
-    bugzillascript = imp.load_source("bugzillascript", "bin/bugzilla")
 
     argv = shlex.split(argv)
 
