Merge "Add more test coverage for shell formatter"

1 files changed, 18 insertions, 0 deletions

diff --git a/cliff/tests/test_formatters_shell.py b/cliff/tests/test_formatters_shell.py
index e6813e5..956e16b 100644
--- a/cliff/tests/test_formatters_shell.py
+++ b/cliff/tests/test_formatters_shell.py
@@ -1,4 +1,7 @@
 #!/usr/bin/env python
+
+import argparse
+
 from six import StringIO, text_type
 
 from cliff.formatters import shell
@@ -20,6 +23,21 @@ def test_shell_formatter():
     assert expected == actual
 
 
+def test_shell_formatter_args():
+    sf = shell.ShellFormatter()
+    c = ('a', 'b', 'c', 'd')
+    d = ('A', 'B', 'C', '"escape me"')
+    expected = 'Xd="\\"escape me\\""\n'
+    output = StringIO()
+    # Parse arguments as if passed on the command-line
+    parser = argparse.ArgumentParser(description='Testing...')
+    sf.add_argument_group(parser)
+    parsed_args = parser.parse_args(['--variable', 'd', '--prefix', 'X'])
+    sf.emit_one(c, d, output, parsed_args)
+    actual = output.getvalue()
+    assert expected == actual
+
+
 def test_shell_formatter_with_non_string_values():
     sf = shell.ShellFormatter()
     c = ('a', 'b', 'c', 'd', 'e')