fix tabulate CLI test on Windows

astanin · astanin · commit e9490f1ffbc1 · 2015-01-27T14:20:37.000+01:00
diff --git a/test/test_cli.py b/test/test_cli.py
@@ -5,6 +5,7 @@
 
 from __future__ import print_function
 from __future__ import unicode_literals
+import os
 
 
 import subprocess
@@ -33,19 +34,31 @@ def get_stdout_and_stderr(cmd):
     return out
 
 
+class TemporaryTextFile(object):
+    def __init__(self):
+        self.tmpfile = None
+    def __enter__(self):
+        self.tmpfile = tempfile.NamedTemporaryFile("w+", prefix="tabulate-test-tmp-", delete=False)
+        return self.tmpfile
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        if self.tmpfile:
+            self.tmpfile.close()
+            os.unlink(self.tmpfile.name)
+
+
 def test_script_from_file_to_stdout():
     """CLI: read from file, print to stdout"""
-    with tempfile.NamedTemporaryFile("w+") as tmpfile:
+    with TemporaryTextFile() as tmpfile:
         write_sample_input(tmpfile)
         cmd = ["python", "tabulate.py", tmpfile.name]
         out = get_stdout_and_stderr(cmd)
         expected = "\n".join([
-            '-----  ------  -------------',
-            'Sun    696000     1.9891e+09',
-            'Earth    6371  5973.6',
-            'Moon     1737    73.5',
-            'Mars     3390   641.85',
-            '-----  ------  -------------'])
+                '-----  ------  -------------',
+                'Sun    696000     1.9891e+09',
+                'Earth    6371  5973.6',
+                'Moon     1737    73.5',
+                'Mars     3390   641.85',
+                '-----  ------  -------------'])
         print("got:     ",repr(out))
         print("expected:",repr(expected))
-        assert_equal(out.rstrip(), expected.rstrip())
+        assert_equal(out.rstrip().splitlines(), expected.rstrip().splitlines())
