test tabulate CLI: stdin-to-stdout, file-to-file

astanin · astanin · commit ace538a40dfa · 2015-01-27T16:08:55.000+01:00
diff --git a/test/test_cli.py b/test/test_cli.py
@@ -15,21 +15,30 @@
 from common import assert_equal
 
 
-def write_sample_input(tmpfile, sep=' '):
+SAMPLE_SIMPLE_FORMAT = "\n".join(
+    ['-----  ------  -------------',
+     'Sun    696000     1.9891e+09',
+     'Earth    6371  5973.6',
+     'Moon     1737    73.5',
+     'Mars     3390   641.85',
+     '-----  ------  -------------'])
+
+
+def sample_input(sep=' '):
     table = "\n".join([
         sep.join(['Sun', '696000', '1.9891e9']),
         sep.join(['Earth', '6371', '5973.6']),
         sep.join(['Moon', '1737', '73.5']),
         sep.join(['Mars', '3390', '641.85'])])
-    tmpfile.write(table)
-    tmpfile.file.seek(0)
+    return table
 
 
-def run_and_capture_stdout(cmd):
+def run_and_capture_stdout(cmd, input=None):
     x = subprocess.Popen(cmd,
+                         stdin=subprocess.PIPE,
                          stdout=subprocess.PIPE,
                          stderr=subprocess.PIPE)
-    out, err = x.communicate()
+    out, err = x.communicate(input=input)
     out = out.decode("utf-8")
     if x.returncode != 0:
         raise IOError(err)
@@ -48,37 +57,59 @@ def __exit__(self, exc_type, exc_val, exc_tb):
             os.unlink(self.tmpfile.name)
 
 
+def test_script_from_stdin_to_stdout():
+    """Command line utility: read from stdin, print to stdout"""
+    cmd = ["python", "tabulate.py"]
+    out = run_and_capture_stdout(cmd, input=sample_input())
+    expected = SAMPLE_SIMPLE_FORMAT
+    print("got:     ",repr(out))
+    print("expected:",repr(expected))
+    assert_equal(out.splitlines(), expected.splitlines())
+
+
 def test_script_from_file_to_stdout():
     """Command line utility: read from file, print to stdout"""
     with TemporaryTextFile() as tmpfile:
-        write_sample_input(tmpfile)
+        tmpfile.write(sample_input())
+        tmpfile.seek(0)
         cmd = ["python", "tabulate.py", tmpfile.name]
         out = run_and_capture_stdout(cmd)
-        expected = "\n".join([
-                '-----  ------  -------------',
-                'Sun    696000     1.9891e+09',
-                'Earth    6371  5973.6',
-                'Moon     1737    73.5',
-                'Mars     3390   641.85',
-                '-----  ------  -------------'])
+        expected = SAMPLE_SIMPLE_FORMAT
         print("got:     ",repr(out))
         print("expected:",repr(expected))
-        assert_equal(out.rstrip().splitlines(), expected.rstrip().splitlines())
+        assert_equal(out.splitlines(), expected.splitlines())
+
+
+def test_script_from_file_to_file():
+    """Command line utility: read from file, write to file"""
+    with TemporaryTextFile() as input_file:
+        with TemporaryTextFile() as output_file:
+            input_file.write(sample_input())
+            input_file.seek(0)
+            cmd = ["python", "tabulate.py", "-o", output_file.name, input_file.name]
+            out = run_and_capture_stdout(cmd)
+            # check that nothing is printed to stdout
+            expected = ""
+            print("got:     ", repr(out))
+            print("expected:", repr(expected))
+            assert_equal(out.splitlines(), expected.splitlines())
+            # check that the output was written to file
+            output_file.seek(0)
+            out = output_file.file.read()
+            expected = SAMPLE_SIMPLE_FORMAT
+            print("got:     ", repr(out))
+            print("expected:", repr(expected))
+            assert_equal(out.splitlines(), expected.splitlines())
 
 
 def test_script_sep_option():
     """Command line utility: --sep option"""
     with TemporaryTextFile() as tmpfile:
-        write_sample_input(tmpfile, sep=",")
+        tmpfile.write(sample_input(sep=","))
+        tmpfile.seek(0)
         cmd = ["python", "tabulate.py", "--sep", ",", tmpfile.name]
         out = run_and_capture_stdout(cmd)
-        expected = "\n".join([
-                '-----  ------  -------------',
-                'Sun    696000     1.9891e+09',
-                'Earth    6371  5973.6',
-                'Moon     1737    73.5',
-                'Mars     3390   641.85',
-                '-----  ------  -------------'])
+        expected = SAMPLE_SIMPLE_FORMAT
         print("got:     ",repr(out))
         print("expected:",repr(expected))
-        assert_equal(out.rstrip().splitlines(), expected.rstrip().splitlines())
+        assert_equal(out.splitlines(), expected.splitlines())
