def test_drainer(self): output = MagicMock() outputs = [output] read_values = ['asdf', ''] def pop_values(*args): return read_values.pop(0) read_mock = MagicMock(side_effect=pop_values) @patch("os.read", read_mock) def _(): base_utils.AsyncJob._fd_drainer(MagicMock(), outputs, MagicMock()) _()
def test_list(self): list_result_mock = MagicMock(exit_status=0, stdout="sshd.service enabled\n" "vsftpd.service disabled\n" "systemd-sysctl.service static\n") run_mock = MagicMock(return_value=list_result_mock) service_manager = TestServiceManager(self.init_name, run_mock).get_service_manager() list_result = service_manager.list(ignore_status=False) assert run_mock.call_args[0][ 0] == "systemctl list-unit-files --type=service --no-pager --full" assert list_result == { 'sshd': "enabled", 'vsftpd': "disabled", 'systemd-sysctl': "static" }
def test_add_dir_splits_on_eq(): @patch.object(sys.modules['os'], "stat", side_effect={ "asdf": MagicMock(st_ino=2, st_dev=2), "qwer": MagicMock(st_ino=3, st_dev=3) }.get) def _test(stat_mock): from autotest.client.os_dep import Ldconfig l = Ldconfig() l._add_dir("asdf=glibc2") assert l.lddirs[0] == Ldconfig.DirEntry("asdf", "glibc2", 2, 2) l._add_dir("qwer%s =glibc2" % os.sep) assert l.lddirs[1] == Ldconfig.DirEntry("qwer", "glibc2", 3, 3) # pylint: disable=E1120 _test()
def _(lock_mock, thread_mock, popen_mock): job = base_utils.AsyncJob("echo", stdin=MagicMock(spec=file)) assert lock_mock.called assert job.stdout_thread.kwargs['name'] == 'echo-stdout' assert job.stderr_thread.kwargs['name'] == 'echo-stderr' assert hasattr(job, "stdout_thread") assert hasattr(job, "stdout_lock") assert hasattr(job, "stderr_thread") assert hasattr(job, "stderr_lock") assert job.string_stdin == None
def test_list(self): list_result_mock = MagicMock( exit_status=0, stdout= "sshd 0:off 1:off 2:off 3:off 4:off 5:off 6:off\n" "vsftpd 0:off 1:off 2:off 3:off 4:off 5:on 6:off\n" "xinetd based services:\n" " amanda: off\n" " chargen-dgram: on\n") run_mock = MagicMock(return_value=list_result_mock) service_manager = TestServiceManager(self.init_name, run_mock).get_service_manager() list_result = service_manager.list(ignore_status=False) assert run_mock.call_args[0][0] == "chkconfig --list" assert list_result == { 'sshd': { 0: "off", 1: "off", 2: "off", 3: "off", 4: "off", 5: "off", 6: "off" }, 'vsftpd': { 0: "off", 1: "off", 2: "off", 3: "off", 4: "off", 5: "on", 6: "off" }, 'xinetd': { 'amanda': "off", 'chargen-dgram': "on" } }
def test_add_dir_calls_stat(): @patch.object(sys.modules['os'], "stat", side_effect={"asdf": MagicMock(st_ino=2, st_dev=2)}.get) def _test(stat_mock): from autotest.client.os_dep import Ldconfig ldconfig = Ldconfig() ldconfig._add_dir("asdf") assert ldconfig.lddirs[0] == Ldconfig.DirEntry("asdf", "", 2, 2) assert ldconfig.lddirs[0].path == "asdf" assert ldconfig.lddirs[0].flag == "" assert ldconfig.lddirs[0].ino == 2 assert ldconfig.lddirs[0].dev == 2 # pylint: disable=E1120 _test()
def _(): base_utils.AsyncJob._fd_drainer(MagicMock(), outputs, MagicMock())
# the file called "COPYING". import os import __builtin__ import unittest try: import autotest.common as common except ImportError: import common from autotest.client.shared.mock import patch, MagicMock, call from autotest.client.shared.file_module_loader import load_module_from_file job_mock = MagicMock() _p = patch.object(__builtin__, "job", job_mock, create=True) _p.start() try: sut_setup_control = load_module_from_file( os.path.join(os.path.dirname(__file__), "control")) finally: _p.stop() class TestSleeptestControl(unittest.TestCase): @staticmethod def test_sleeptest_control(): assert "autotest" in sut_setup_control.DOC assert job_mock.run_test.call_args_list == [ call("sleeptest", seconds=1)
def setUp(self): self.run_mock = MagicMock() self.init_name = "init"