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"
