summaryrefslogblamecommitdiffhomepage
path: root/test/test_python_procman.py
blob: 461846e3ff0e8f568bdc576b48c29bf02d048786 (plain) (tree)
1
2
3
4
5
6
7
8
9
         
           
                 
               
                                                               
 
 
                                               
                     
                                                       
 
                               
                       
 
                                                      
 
                    
                                                                   
                                                        
 
                   
 

                       
 
                                                            
                                         
 
                                           
                                                                      
 


















                                                                               
 
                                                   






                                                                             

                                                 






                                                                           

                                                          







                                                               
 
                                                         






                                                                            
 
                                                 







                                                                

                                             







                                                               

                                                      



                                                           



                                                                           
                                  
                                                                      


                                                   

                  
                                                                             
 
                                                                      
 

                                                   

                  
                                                                             
 
                       
 

                                                 
                                                                      


                                      
                                                                      




                                                                          
                                   



                                                           
 
                                                                        

                  

                                                     

                  


                                                                               
 


                                                                  
 
                       

                                       



                                                           
 

                                                  

                  


                                                                       

                  


                                                               

                     
 

                                                          

                  


                                                                             
 
                       

                                      



                                                           
 

                                                  

                  


                                                                       
 
                                                                            
 

                                                  

                  
                                                                            
 

                       
                                       



                                                           












                                                                  


                                                                          





                                                                           












                                                                      


                     


                                                                     


                    
                       
                                                        
 
                                                                     
 
 
                          
                            
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()