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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
|
#!/usr/bin/env python
import pexpect
import unittest
import commands
import sys
import PexpectTestCase
#import pdb
# Many of these test cases blindly assume that sequential directory
# listings of the /bin directory will yield the same results.
# This may not be true, but seems adequate for testing now.
# I should fix this at some point.
FILTER=''.join([(len(repr(chr(x)))==3) and chr(x) or '.' for x in range(256)])
def hex_dump(src, length=16):
result=[]
for i in xrange(0, len(src), length):
s = src[i:i+length]
hexa = ' '.join(["%02X"%ord(x) for x in s])
printable = s.translate(FILTER)
result.append("%04X %-*s %s\n" % (i, length*3, hexa, printable))
return ''.join(result)
class ExpectTestCase (PexpectTestCase.PexpectTestCase):
def test_expect_basic (self):
p = pexpect.spawn('cat')
p.sendline ('Hello')
p.sendline ('there')
p.sendline ('Mr. Python')
p.expect ('Hello')
p.expect ('there')
p.expect ('Mr. Python')
p.sendeof ()
p.expect (pexpect.EOF)
def test_expect_ignore_case(self):
"""This test that the ignorecase flag will match patterns
even if case is different using the regex (?i) directive.
"""
p = pexpect.spawn('cat')
p.sendline ('HELLO')
p.sendline ('there')
p.expect ('(?i)hello')
p.expect ('(?i)THERE')
p.sendeof ()
p.expect (pexpect.EOF)
def test_expect_ignore_case_flag(self):
"""This test that the ignorecase flag will match patterns
even if case is different using the ignorecase flag.
"""
p = pexpect.spawn('cat')
p.ignorecase = True
p.sendline ('HELLO')
p.sendline ('there')
p.expect ('hello')
p.expect ('THERE')
p.sendeof ()
p.expect (pexpect.EOF)
def test_expect_order (self):
"""This tests that patterns are matched in the same order as given in the pattern_list.
"""
p = pexpect.spawn('cat')
p.sendline ('1234')
p.sendline ('abcd')
p.sendline ('wxyz')
p.sendline ('7890')
p.sendeof ()
index = p.expect (['1234','abcd','wxyz',pexpect.EOF,'7890'])
assert index == 0, "index="+str(index)
index = p.expect (['1234','abcd','wxyz',pexpect.EOF,'7890'])
assert index == 0, "index="+str(index)
index = p.expect ([pexpect.EOF,pexpect.TIMEOUT,'wxyz','abcd','1234'])
assert index == 3, "index="+str(index)
index = p.expect (['54321',pexpect.TIMEOUT,'1234','abcd','wxyz',pexpect.EOF], timeout=5)
assert index == 3, "index="+str(index)
index = p.expect (['54321',pexpect.TIMEOUT,'1234','abcd','wxyz',pexpect.EOF], timeout=5)
assert index == 4, "index="+str(index)
index = p.expect (['54321',pexpect.TIMEOUT,'1234','abcd','wxyz',pexpect.EOF], timeout=5)
assert index == 4, "index="+str(index)
index = p.expect ([pexpect.EOF,'abcd','wxyz','7890'])
assert index == 3, "index="+str(index)
index = p.expect ([pexpect.EOF,'abcd','wxyz','7890'])
assert index == 3, "index="+str(index)
def test_expect_echo (self):
"""This tests that echo can be turned on and off.
"""
p = pexpect.spawn('cat', timeout=10)
p.sendline ('1234') # Should see this twice (once from tty echo and again from cat).
index = p.expect (['1234','abcd','wxyz',pexpect.EOF,pexpect.TIMEOUT])
assert index == 0, "index="+str(index)+"\n"+p.before
index = p.expect (['1234','abcd','wxyz',pexpect.EOF])
assert index == 0, "index="+str(index)
p.setecho(0) # Turn off tty echo
p.sendline ('abcd') # Now, should only see this once.
p.sendline ('wxyz') # Should also be only once.
index = p.expect ([pexpect.EOF,pexpect.TIMEOUT,'abcd','wxyz','1234'])
assert index == 2, "index="+str(index)
index = p.expect ([pexpect.EOF,'abcd','wxyz','7890'])
assert index == 2, "index="+str(index)
p.setecho(1) # Turn on tty echo
p.sendline ('7890') # Should see this twice.
index = p.expect ([pexpect.EOF,'abcd','wxyz','7890'])
assert index == 3, "index="+str(index)
index = p.expect ([pexpect.EOF,'abcd','wxyz','7890'])
assert index == 3, "index="+str(index)
p.sendeof()
def test_expect_index (self):
"""This tests that mixed list of regex strings, TIMEOUT, and EOF all
return the correct index when matched.
"""
#pdb.set_trace()
p = pexpect.spawn('cat')
p.setecho(0)
p.sendline ('1234')
index = p.expect (['abcd','wxyz','1234',pexpect.EOF])
assert index == 2, "index="+str(index)
p.sendline ('abcd')
index = p.expect ([pexpect.TIMEOUT,'abcd','wxyz','1234',pexpect.EOF])
assert index == 1, "index="+str(index)
p.sendline ('wxyz')
index = p.expect (['54321',pexpect.TIMEOUT,'abcd','wxyz','1234',pexpect.EOF], timeout=5)
assert index == 3, "index="+str(index) # Expect 'wxyz'
p.sendline ('$*!@?')
index = p.expect (['54321',pexpect.TIMEOUT,'abcd','wxyz','1234',pexpect.EOF], timeout=5)
assert index == 1, "index="+str(index) # Expect TIMEOUT
p.sendeof ()
index = p.expect (['54321',pexpect.TIMEOUT,'abcd','wxyz','1234',pexpect.EOF], timeout=5)
assert index == 5, "index="+str(index) # Expect EOF
def test_expect (self):
the_old_way = commands.getoutput('ls -l /bin')
p = pexpect.spawn('ls -l /bin')
the_new_way = ''
while 1:
i = p.expect (['\n', pexpect.EOF])
the_new_way = the_new_way + p.before
if i == 1:
break
the_new_way = the_new_way[:-1]
the_new_way = the_new_way.replace('\r','\n')
# For some reason I get an extra newline under OS X evey once in a while.
# I found it by looking through the hex_dump().
assert the_old_way == the_new_way, hex_dump(the_new_way) + "\n" + hex_dump(the_old_way)
def test_expect_exact (self):
the_old_way = commands.getoutput('ls -l /bin')
p = pexpect.spawn('ls -l /bin')
the_new_way = ''
while 1:
i = p.expect_exact (['\n', pexpect.EOF])
the_new_way = the_new_way + p.before
if i == 1:
break
the_new_way = the_new_way[:-1]
the_new_way = the_new_way.replace('\r','\n')
assert the_old_way == the_new_way, repr(the_old_way) + '\n' + repr(the_new_way)
p = pexpect.spawn('echo hello.?world')
i = p.expect_exact('.?')
assert p.before == 'hello' and p.after == '.?'
def test_expect_eof (self):
the_old_way = commands.getoutput('/bin/ls -l /bin')
p = pexpect.spawn('/bin/ls -l /bin')
p.expect(pexpect.EOF) # This basically tells it to read everything. Same as pexpect.run() function.
the_new_way = p.before
the_new_way = the_new_way.replace('\r','') # Remember, pty line endings are '\r\n'.
the_new_way = the_new_way[:-1]
assert the_old_way == the_new_way
def test_expect_timeout (self):
p = pexpect.spawn('ed', timeout=10)
i = p.expect(pexpect.TIMEOUT) # This tells it to wait for timeout.
assert p.after == pexpect.TIMEOUT
def test_unexpected_eof (self):
p = pexpect.spawn('ls -l /bin')
try:
p.expect('_Z_XY_XZ') # Probably never see this in ls output.
except pexpect.EOF, e:
pass
else:
self.fail ('Expected an EOF exception.')
if __name__ == '__main__':
unittest.main()
suite = unittest.makeSuite(ExpectTestCase,'test')
#fout = open('delete_me_1','wb')
#fout.write(the_old_way)
#fout.close
#fout = open('delete_me_2', 'wb')
#fout.write(the_new_way)
#fout.close
|
