Test your AI Agents with the all-new Agent to Agent Testing Platform.Learn More

How to use get_service_manager_from_init_and_run method in avocado

Best Python code snippet using avocado_python

service_unittest.py

Source:service_unittest.py

...99    def test_set_target_is_not_present_in_SpecifcServiceManager(self):100        assert not hasattr(self.service_manager, "set_target")101class TestServiceManager(unittest.TestCase):102    @staticmethod103    def get_service_manager_from_init_and_run(init_name, run_mock):104        command_generator = service._command_generators[init_name]105        result_parser = service._result_parsers[init_name]106        service_manager = service._service_managers[init_name]107        service_command_generator = service._ServiceCommandGenerator(108            command_generator)109        service_result_parser = service._ServiceResultParser(result_parser)110        return service_manager(service_command_generator, service_result_parser, run_mock)111class TestSystemdServiceManager(TestServiceManager):112    def setUp(self):113        self.run_mock = MagicMock()114        self.init_name = "systemd"115        self.service_manager = super(TestSystemdServiceManager,116                                     self).get_service_manager_from_init_and_run(self.init_name,117                                                                                 self.run_mock)118    def test_start(self):119        service = "lldpad"120        self.service_manager.start(service)121        assert self.run_mock.call_args[0][122            0] == "systemctl start %s.service" % service123    def test_list(self):124        list_result_mock = MagicMock(exit_status=0, stdout="sshd.service enabled\n"125                                                           "vsftpd.service disabled\n"126                                                           "systemd-sysctl.service static\n")127        run_mock = MagicMock(return_value=list_result_mock)128        service_manager = super(TestSystemdServiceManager,129                                self).get_service_manager_from_init_and_run(self.init_name,130                                                                            run_mock)131        list_result = service_manager.list(ignore_status=False)132        assert run_mock.call_args[0][133            0] == "systemctl list-unit-files --type=service --no-pager --full"134        assert list_result == {'sshd': "enabled",135                               'vsftpd': "disabled",136                               'systemd-sysctl': "static"}137    def test_set_default_runlevel(self):138        runlevel = service.convert_sysv_runlevel(3)139        mktemp_mock = MagicMock(return_value="temp_filename")140        symlink_mock = MagicMock()141        rename_mock = MagicMock()142        @patch.object(service, "mktemp", mktemp_mock)143        @patch("os.symlink", symlink_mock)144        @patch("os.rename", rename_mock)145        def _():146            self.service_manager.change_default_runlevel(runlevel)147            assert mktemp_mock.called148            assert symlink_mock.call_args[0][149                0] == "/usr/lib/systemd/system/multi-user.target"150            assert rename_mock.call_args[0][151                1] == "/etc/systemd/system/default.target"152        _()153    def test_unknown_runlevel(self):154        self.assertRaises(ValueError,155                          service.convert_systemd_target_to_runlevel, "unknown")156    def test_runlevels(self):157        assert service.convert_sysv_runlevel(0) == "poweroff.target"158        assert service.convert_sysv_runlevel(1) == "rescue.target"159        assert service.convert_sysv_runlevel(2) == "multi-user.target"160        assert service.convert_sysv_runlevel(5) == "graphical.target"161        assert service.convert_sysv_runlevel(6) == "reboot.target"162class TestSysVInitServiceManager(TestServiceManager):163    def setUp(self):164        self.run_mock = MagicMock()165        self.init_name = "init"166        self.service_manager = super(TestSysVInitServiceManager,167                                     self).get_service_manager_from_init_and_run(self.init_name,168                                                                                 self.run_mock)169    def test_list(self):170        list_result_mock = MagicMock(exit_status=0,171                                     stdout="sshd             0:off   1:off   2:off   3:off   4:off   5:off   6:off\n"172                                            "vsftpd           0:off   1:off   2:off   3:off   4:off   5:on   6:off\n"173                                            "xinetd based services:\n"174                                            "        amanda:         off\n"175                                            "        chargen-dgram:  on\n")176        run_mock = MagicMock(return_value=list_result_mock)177        service_manager = super(TestSysVInitServiceManager,178                                self).get_service_manager_from_init_and_run(self.init_name,179                                                                            run_mock)180        list_result = service_manager.list(ignore_status=False)181        assert run_mock.call_args[0][182            0] == "chkconfig --list"183        assert list_result == {'sshd': {0: "off", 1: "off", 2: "off", 3: "off", 4: "off", 5: "off", 6: "off"},184                               'vsftpd': {0: "off", 1: "off", 2: "off", 3: "off", 4: "off", 5: "on", 6: "off"},185                               'xinetd': {'amanda': "off", 'chargen-dgram': "on"}}186    def test_enable(self):187        service = "lldpad"188        self.service_manager.enable(service)189        assert self.run_mock.call_args[0][0] == "chkconfig lldpad on"190    def test_unknown_runlevel(self):191        self.assertRaises(ValueError,192                          service.convert_sysv_runlevel, "unknown")...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.