diff options
Diffstat (limited to 'tests/worker.py')
-rw-r--r-- | tests/worker.py | 148 |
1 files changed, 0 insertions, 148 deletions
diff --git a/tests/worker.py b/tests/worker.py deleted file mode 100644 index cf121f38..00000000 --- a/tests/worker.py +++ /dev/null @@ -1,148 +0,0 @@ -#!/usr/bin/env python - -import unittest -import pexpect - -from re import search, escape -from os import kill -from signal import SIGINT, SIGUSR1 - -class WorkerTests(unittest.TestCase): - - __exec_echo_lol_output__ = '0\r\n4\r\nlol\r\n\r\n0\r\n\r\n' - - def prompt(self, n): - return ('#%i#\r\n' % n) - - def interrupted(self, n): - return ('#%i# INTERRUPTED\r\n' % n) - - def setUp(self): - self.worker = pexpect.spawn('python ../worker/worker.py') - # worker should always be very fast to span - self.expect(self.prompt(0)) - - 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_when_running(self): - self.sendlines(['exec', 'sleep 1; exec echo lol']) - self.sig(SIGINT) - self.expect(self.interrupted(0)+self.prompt(1)) - self.test_exit() - - def test_ctrlc_on_cmd_prompt(self): - self.sig(SIGINT) - self.expect(self.interrupted(0)+self.prompt(1)) - self.test_exit() - - def test_ctrlc_when_entering_command(self): - # "Mon clavier se blo" -- French reference - self.send('glo') - self.sig(SIGINT) - self.expect(self.interrupted(0)+self.prompt(1)) - self.test_exit() - - def test_ctrlc_on_readparms_drops(self): - self.sendlines(['exec']) - self.sig(SIGINT) - self.expect(self.interrupted(0)+self.prompt(1)) - self.sendlines(['glob']) - self.sig(SIGINT) - self.expect(self.interrupted(1)+self.prompt(2)) - self.test_exit() - - def test_basic_noop(self): - self.sendlines(['noop']) - self.expect(self.prompt(1)) - self.test_exit() - - def test_basic_ping(self): - self.sendlines(['ping']) - self.expect('ALIVE\r\n' + self.prompt(1)) - 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'+self.prompt(1)) - self.test_exit() - - def test_empty_glob(self): - self.sendlines(['glob', '/?kyzh?']) - self.expect('0\r\n'+self.prompt(1)) - self.test_exit() - - def test_basic_sigusr1(self): - self.sig(SIGUSR1) - self.expect('ALIVE\r\n') - self.test_exit() - - def test_sigusr1_when_entering_command(self): - self.worker.send('pin') - self.sig(SIGUSR1) - self.expect('pinALIVE\r\n') - self.sendlines(['g']) - self.expect('ALIVE\r\n' + self.prompt(1)) - self.test_exit() - - def test_sigusr1_on_readparms(self): - self.worker.send('exec\nech') - self.sig(SIGUSR1) - self.worker.send('o lol\n') - self.expect('exec\r\nechALIVE\r\no lol\r\n'+ - self.__exec_echo_lol_output__ + self.prompt(1)) - self.test_exit() - - def test_exec_continues_after_sigusr1(self): - self.worker.send('exec\nsleep 0.1; exec echo lol\n') - self.sig(SIGUSR1) - self.expect('exec\r\nsleep 0.1; exec echo lol\r\nALIVE\r\n'+ - self.__exec_echo_lol_output__ + self.prompt(1)) - self.test_exit() - - def test_increasing_counter(self): - for req_counter in range(1, 5): - self.sendlines(['noop']) - self.expect(self.prompt(req_counter)) - for req_counter in range(5, 10): - self.sendlines(['ping']) - self.expect('ALIVE\r\n'+self.prompt(req_counter)) - self.test_exit() - - def test_queuecommands(self): - self.worker.send('ping\n'*5) - self.worker.send('exec\necho lol\n'*5) - for req_counter in range(1,6): - self.lose_expect('ALIVE\r\n'+self.prompt(req_counter)) - for req_counter in range(6,11): - self.lose_expect(self.__exec_echo_lol_output__+self.prompt(req_counter)) - self.test_exit() - -if __name__ == '__main__': - unittest.main() |