1 files changed, 12 insertions, 2 deletions
diff --git a/echo.py b/echo.py
index 9ec8437..abc0e52 100644
--- a/echo.py
+++ b/echo.py
@@ -13,15 +13,22 @@ def assertEqual(a, b):
         raise Exception(
             'expected {!r} == {!r}, but was disappointed'.format(a, b))
 
-def run_echo_without_args():
+def _run_echo(args):
     cmd = '/bin/echo'
     p = subprocess.Popen(
-        [cmd], stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
+        [cmd] + args, stdout=subprocess.PIPE, stderr=subprocess.PIPE,
+        text=True)
     out, err = p.communicate()
     context["stdout"] = out
     context["stderr"] = err
     context["exit_code"] = p.returncode
 
+def run_echo_without_args():
+    _run_echo([])
+
+def run_echo_with_args(args=None):
+    _run_echo(args.split())
+
 def exit_code_is_zero(exit_code=None):
     exit_code = int(exit_code)
     assertEqual(_get("exit_code"), exit_code)
@@ -29,5 +36,8 @@ def exit_code_is_zero(exit_code=None):
 def stdout_is_a_newline():
     assertEqual(_get('stdout'), '\n')
 
+def stdout_is_text(text=None):
+    assertEqual(_get('stdout'), text + '\n')
+
 def stderr_is_empty():
     assertEqual(_get('stderr'), '')