tests: remove temp doctest file when finished running it
authorIdan Kamara <idankk86@gmail.com>
Thu, 13 Oct 2011 17:54:35 +0200
changeset 15247 3cd1605e9d8e
parent 15246 7b15dd9125b3
child 15248 3efbb5e8bf5c
tests: remove temp doctest file when finished running it
tests/heredoctest.py
--- a/tests/heredoctest.py	Thu Oct 13 04:27:49 2011 +0200
+++ b/tests/heredoctest.py	Thu Oct 13 17:54:35 2011 +0200
@@ -2,8 +2,12 @@
 
 if __name__ == "__main__":
     fd, name = tempfile.mkstemp(suffix='hg-tst')
-    os.write(fd, sys.stdin.read())
-    os.close(fd)
-    failures, _ = doctest.testfile(name, module_relative=False)
-    if failures:
-        sys.exit(1)
+
+    try:
+        os.write(fd, sys.stdin.read())
+        os.close(fd)
+        failures, _ = doctest.testfile(name, module_relative=False)
+        if failures:
+            sys.exit(1)
+    finally:
+        os.remove(name)