import re import time import subprocess import unittest import unit class TestUnitPythonProcman(unit.TestUnitApplicationPython): def setUpClass(): unit.TestUnit().check_modules('python') def pids_for_process(self): time.sleep(0.2) output = subprocess.check_output(['ps', 'ax']) pids = set() for m in re.findall('.*' + self.app_name, output.decode()): pids.add(re.search('^\s*(\d+)', m).group(1)) return pids def setUp(self): super().setUp() self.app_name = "app-" + self.testdir.split('/')[-1] self.load('empty', self.app_name) def test_python_processes_access(self): self.conf('1', '/applications/' + self.app_name + '/processes') self.assertIn('error', self.conf_get('/applications/' + self.app_name + '/processes/max'), 'max no access') self.assertIn('error', self.conf_get('/applications/' + self.app_name + '/processes/spare'), 'spare no access') self.assertIn('error', self.conf_get('/applications/' + self.app_name + '/processes/idle_timeout'), 'idle_timeout no access') def test_python_processes_spare_gt_max_default(self): self.assertIn('error', self.conf({"spare": 2}, '/applications/' + self.app_name + '/processes'), 'spare greater than max default') def test_python_processes_spare_gt_max(self): self.assertIn('error', self.conf({ "spare": 2, "max": 1, "idle_timeout": 1 }, '/applications/' + self.app_name + '/processes'), 'spare greater than max') def test_python_processes_max_zero(self): self.assertIn('error', self.conf({ "spare": 0, "max": 0, "idle_timeout": 1 }, '/applications/' + self.app_name + '/processes'), 'max 0') def test_python_processes_idle_timeout_zero(self): self.conf({ "spare": 0, "max": 2, "idle_timeout": 0 }, '/applications/' + self.app_name + '/processes') self.get() self.assertEqual(len(self.pids_for_process()), 0, 'idle timeout 0') def test_python_prefork(self): self.conf('2', '/applications/' + self.app_name + '/processes') pids = self.pids_for_process() self.assertEqual(len(pids), 2, 'prefork 2') self.get() self.assertSetEqual(self.pids_for_process(), pids, 'prefork still 2') self.conf('4', '/applications/' + self.app_name + '/processes') pids = self.pids_for_process() self.assertEqual(len(pids), 4, 'prefork 4') self.get() self.assertSetEqual(self.pids_for_process(), pids, 'prefork still 4') self.stop_all() @unittest.expectedFailure def test_python_prefork_same_processes(self): self.conf('2', '/applications/' + self.app_name + '/processes') pids = self.pids_for_process() self.conf('4', '/applications/' + self.app_name + '/processes') pids_new = self.pids_for_process() self.assertTrue(pids.issubset(pids_new), 'prefork same processes') def test_python_ondemand(self): self.conf({ "spare": 0, "max": 8, "idle_timeout": 1 }, '/applications/' + self.app_name + '/processes') self.assertEqual(len(self.pids_for_process()), 0, 'on-demand 0') self.get() pids = self.pids_for_process() self.assertEqual(len(pids), 1, 'on-demand 1') self.get() self.assertSetEqual(self.pids_for_process(), pids, 'on-demand still 1') time.sleep(1) self.assertEqual(len(self.pids_for_process()), 0, 'on-demand stop idle') self.stop_all() def test_python_scale_updown(self): self.conf({ "spare": 2, "max": 8, "idle_timeout": 1 }, '/applications/' + self.app_name + '/processes') pids = self.pids_for_process() self.assertEqual(len(pids), 2, 'updown 2') self.get() pids_new = self.pids_for_process() self.assertEqual(len(pids_new), 3, 'updown 3') self.assertTrue(pids.issubset(pids_new), 'updown 3 only 1 new') self.get() self.assertSetEqual(self.pids_for_process(), pids_new, 'updown still 3') time.sleep(1) pids = self.pids_for_process() self.assertEqual(len(pids), 2, 'updown stop idle') self.get() pids_new = self.pids_for_process() self.assertEqual(len(pids_new), 3, 'updown again 3') self.assertTrue(pids.issubset(pids_new), 'updown again 3 only 1 new') self.stop_all() def test_python_reconfigure(self): self.conf({ "spare": 2, "max": 6, "idle_timeout": 1 }, '/applications/' + self.app_name + '/processes') pids = self.pids_for_process() self.assertEqual(len(pids), 2, 'reconf 2') self.get() pids_new = self.pids_for_process() self.assertEqual(len(pids_new), 3, 'reconf 3') self.assertTrue(pids.issubset(pids_new), 'reconf 3 only 1 new') self.conf('6', '/applications/' + self.app_name + '/processes/spare') pids = self.pids_for_process() self.assertEqual(len(pids), 6, 'reconf 6') self.get() self.assertSetEqual(self.pids_for_process(), pids, 'reconf still 6') self.stop_all() def test_python_idle_timeout(self): self.conf({ "spare": 0, "max": 6, "idle_timeout": 2 }, '/applications/' + self.app_name + '/processes') self.get() pids = self.pids_for_process() self.assertEqual(len(pids), 1, 'idle timeout 1') time.sleep(1) self.get() time.sleep(1) pids_new = self.pids_for_process() self.assertEqual(len(pids_new), 1, 'idle timeout still 1') self.assertSetEqual(self.pids_for_process(), pids, 'idle timeout still 1 same pid') time.sleep(1) self.assertEqual(len(self.pids_for_process()), 0, 'idle timed out') def test_python_processes_connection_keepalive(self): self.conf({ "spare": 0, "max": 6, "idle_timeout": 2 }, '/applications/' + self.app_name + '/processes') (resp, sock) = self.get(headers={ 'Host': 'localhost', 'Connection': 'keep-alive' }, start=True) self.assertEqual(len(self.pids_for_process()), 1, 'keepalive connection 1') time.sleep(2) self.assertEqual(len(self.pids_for_process()), 0, 'keepalive connection 0') sock.close() def stop_all(self): self.conf({ "listeners": {}, "applications": {} }) self.assertEqual(len(self.pids_for_process()), 0, 'stop all') if __name__ == '__main__': unittest.main()