summaryrefslogtreecommitdiffstats
path: root/test/integration/targets/expect/files/test_command.py
blob: 0e0e2646669e70a51507d2fd18c8c6ee280c7b9b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
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)