import re
import time
import subprocess
import unittest
from unit.applications.lang.python import TestApplicationPython
class TestPythonProcman(TestApplicationPython):
def setUpClass():
TestApplicationPython().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_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()