import re import time import subprocess import unittest from unit.applications.lang.python import TestApplicationPython class TestPythonProcman(TestApplicationPython): prerequisites = ['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_negative(self): self.assertIn( 'error', self.conf( {"spare": -1}, 'applications/' + self.app_name + '/processes' ), 'negative spare', ) def test_python_processes_max_negative(self): self.assertIn( 'error', self.conf( {"max": -1}, 'applications/' + self.app_name + '/processes' ), 'negative max', ) def test_python_processes_idle_timeout_negative(self): self.assertIn( 'error', self.conf( {"idle_timeout": -1}, 'applications/' + self.app_name + '/processes', ), 'negative idle_timeout', ) 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, read_timeout=1, ) 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__': TestPythonProcman.main()