summaryrefslogtreecommitdiffstats
path: root/test/integration/targets/expect/files/test_command.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/integration/targets/expect/files/test_command.py')
-rw-r--r--test/integration/targets/expect/files/test_command.py25
1 files changed, 25 insertions, 0 deletions
diff --git a/test/integration/targets/expect/files/test_command.py b/test/integration/targets/expect/files/test_command.py
new file mode 100644
index 0000000..0e0e264
--- /dev/null
+++ b/test/integration/targets/expect/files/test_command.py
@@ -0,0 +1,25 @@
+from __future__ import (absolute_import, division, print_function)
+__metaclass__ = type
+
+import sys
+
+try:
+ input_function = raw_input
+except NameError:
+ input_function = input
+
+prompts = sys.argv[1:] or ['foo']
+
+# latin1 encoded bytes
+# to ensure pexpect doesn't have any encoding errors
+data = b'premi\xe8re is first\npremie?re is slightly different\n????????? is Cyrillic\n? am Deseret\n'
+
+try:
+ sys.stdout.buffer.write(data)
+except AttributeError:
+ sys.stdout.write(data)
+print()
+
+for prompt in prompts:
+ user_input = input_function(prompt)
+ print(user_input)