aboutsummaryrefslogtreecommitdiffstats
path: root/tests/worker.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/worker.py')
-rw-r--r--tests/worker.py148
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()