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
|
#!/usr/bin/env python
import unittest
import pexpect
from re import search, escape
from os import kill
from signal import SIGINT, SIGUSR1
class PexpectTest(unittest.TestCase):
def setUp(self):
self.worker = pexpect.spawn('python ../worker/worker.py')
# worker should always be very fast to span
self.expect('#0#\r\n')
def sig(self, sig):
kill(self.worker.pid, sig)
def lose_expect(self, v, timeout = 3):
self.worker.expect(v, timeout)
def expect(self, v, timeout = 3):
self.lose_expect(v, timeout)
self.assertEqual(self.worker.before, '')
def send(self, text):
self.worker.send(text)
self.expect(escape(text))
def sendlines(self, lines):
for line in lines:
self.worker.send(line+'\n')
for line in lines:
self.expect(escape(line)+'\r\n')
def __finishes_ok__(self):
self.expect(pexpect.EOF)
self.worker.close()
self.assertEqual(self.worker.exitstatus, 0)
def test_exit(self):
self.sendlines(['exit'])
self.__finishes_ok__()
def test_ctrlc_on_cmd_prompt_quits(self):
self.sig(SIGINT)
self.expect(pexpect.EOF)
self.__finishes_ok__()
def test_ctrlc_when_entering_command_quits(self):
# "Mon clavier se blo" -- French reference
self.send('glo')
self.sig(SIGINT)
self.expect(pexpect.EOF)
def test_ctrlc_on_readparms_drops(self):
self.sendlines(['exec'])
self.sig(SIGINT)
self.expect('#0#\r\n')
self.sendlines(['glob'])
self.sig(SIGINT)
self.expect('#0#\r\n')
def test_basic_noop(self):
self.sendlines(['noop'])
self.expect('#1#\r\n')
self.test_exit()
def test_basic_ping(self):
self.sendlines(['ping'])
self.expect('ALIVE\r\n#1#\r\n')
self.test_exit()
def test_basic_glob(self):
self.sendlines(['glob', '/*bin'])
self.expect('2\r\n(/bin\r\n/sbin|/sbin\r\n/bin)\r\n#1#\r\n')
self.test_exit()
self.test_exit()
if __name__ == '__main__':
unittest.main()
|
