# HG changeset patch
# User Idan Kamara <idankk86@gmail.com>
# Date 2011-10-13 15:54:35
# Node ID 3cd1605e9d8e9c148c5fd079b1c8941233f477b4
# Parent  7b15dd9125b301a1b9e1d642268df0f2fd04e4d4

tests: remove temp doctest file when finished running it

diff --git a/tests/heredoctest.py b/tests/heredoctest.py
--- a/tests/heredoctest.py
+++ b/tests/heredoctest.py
@@ -2,8 +2,12 @@ import doctest, tempfile, os, sys
 
 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)