[PATCH p2v-transfer.git] Clean up temp files generated by unit tests

[Ben Lipton]

Signed-off-by: Ben Lipton <benlip...@google.com>
---
 instance-p2v-target/test/fix_fstab_test.py |   10 +++++++---
 1 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/instance-p2v-target/test/fix_fstab_test.py 
b/instance-p2v-target/test/fix_fstab_test.py
index 36db907..e9431ef 100755
--- a/instance-p2v-target/test/fix_fstab_test.py
+++ b/instance-p2v-target/test/fix_fstab_test.py
@@ -51,6 +51,7 @@ class FixFstabTest(unittest.TestCase):
     desired_data = desired.read()
     out.close()
     desired.close()
+    os.remove(fname_out)
 
     self.assertEqual(desired_data, out_data)
 
@@ -74,10 +75,13 @@ class FixFstabTest(unittest.TestCase):
     handle_out, fname_out = tempfile.mkstemp()
     os.close(handle_out)
 
-    self.assertRaises(fix_fstab.fixlib.FixError, fix_fstab.FixFstab,
-                      fname_in, fname_out)
+    try:
+      self.assertRaises(fix_fstab.fixlib.FixError, fix_fstab.FixFstab,
+                        fname_in, fname_out)
 
-    self.mox.VerifyAll()
+      self.mox.VerifyAll()
+    finally:
+      os.remove(fname_out)
 
 
 if __name__ == '__main__':
-- 
1.7.3.1