238 lines
8.3 KiB
Python
238 lines
8.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from unittest import mock
|
|
import os
|
|
|
|
from pungi.runroot import Runroot
|
|
from tests import helpers
|
|
|
|
|
|
class TestRunrootOpenSSH(helpers.PungiTestCase):
|
|
def setUp(self):
|
|
super(TestRunrootOpenSSH, self).setUp()
|
|
self.compose = helpers.DummyCompose(
|
|
self.topdir,
|
|
{
|
|
"runroot": True,
|
|
"runroot_method": "openssh",
|
|
"runroot_ssh_user": "root",
|
|
"runroot_ssh_hostnames": {"x86_64": "localhost"},
|
|
"runroot_tag": "f28-build",
|
|
},
|
|
)
|
|
|
|
self.runroot = Runroot(self.compose)
|
|
|
|
def test_get_runroot_method(self):
|
|
method = self.runroot.get_runroot_method()
|
|
self.assertEqual(method, "openssh")
|
|
|
|
def test_get_runroot_method_phase(self):
|
|
self.compose.conf["runroot_method"] = {"ostree": "openssh"}
|
|
method = self.runroot.get_runroot_method(phase="ostree")
|
|
self.assertEqual(method, "openssh")
|
|
|
|
# Test fallback to local
|
|
method = self.runroot.get_runroot_method(phase="createiso")
|
|
self.assertEqual(method, "local")
|
|
|
|
def test_get_runroot_method_global_runroot_method(self):
|
|
self.compose.conf["runroot_method"] = {"ostree": "openssh"}
|
|
self.compose.conf["global_runroot_method"] = "koji"
|
|
method = self.runroot.get_runroot_method(phase="ostree")
|
|
self.assertEqual(method, "openssh")
|
|
|
|
# Test global_runroot_method
|
|
method = self.runroot.get_runroot_method(phase="createiso")
|
|
self.assertEqual(method, "koji")
|
|
|
|
def _ssh_call(self, cmd, suffix=None):
|
|
"""
|
|
Helper method returning default SSH mock.call with given command `cmd`.
|
|
"""
|
|
logfile = ("/foo/runroot." + suffix + ".log") if suffix else "/foo/runroot.log"
|
|
return mock.call(
|
|
["ssh", "-oBatchMode=yes", "-n", "-l", "root", "localhost", cmd],
|
|
logfile=logfile,
|
|
show_cmd=True,
|
|
)
|
|
|
|
@mock.patch("pungi.runroot.run")
|
|
def test_run(self, run):
|
|
run.return_value = (0, "dummy output\n")
|
|
self.runroot.run("df -h", log_file="/foo/runroot.log", arch="x86_64")
|
|
run.assert_has_calls(
|
|
[
|
|
self._ssh_call("df -h"),
|
|
self._ssh_call(
|
|
"rpm -qa --qf='%{name}-%{version}-%{release}.%{arch}\n'",
|
|
suffix="rpms",
|
|
),
|
|
]
|
|
)
|
|
|
|
@mock.patch("pungi.runroot.run")
|
|
def test_get_buildroot_rpms(self, run):
|
|
# Run the runroot task at first.
|
|
run.return_value = (0, "foo-1-1.fc29.noarch\nbar-1-1.fc29.noarch\n")
|
|
self.runroot.run("df -h", log_file="/foo/runroot.log", arch="x86_64")
|
|
|
|
rpms = self.runroot.get_buildroot_rpms()
|
|
self.assertEqual(set(rpms), set(["foo-1-1.fc29.noarch", "bar-1-1.fc29.noarch"]))
|
|
|
|
@mock.patch("pungi.runroot.run")
|
|
def test_run_templates(self, run):
|
|
self.compose.conf[
|
|
"runroot_ssh_init_template"
|
|
] = "/usr/sbin/init_runroot {runroot_tag}"
|
|
self.compose.conf[
|
|
"runroot_ssh_install_packages_template"
|
|
] = "install {runroot_key} {packages}"
|
|
self.compose.conf["runroot_ssh_run_template"] = "run {runroot_key} {command}"
|
|
|
|
run.return_value = (0, "key\n")
|
|
self.runroot.run(
|
|
"df -h",
|
|
log_file="/foo/runroot.log",
|
|
arch="x86_64",
|
|
packages=["lorax", "automake"],
|
|
)
|
|
run.assert_has_calls(
|
|
[
|
|
self._ssh_call("/usr/sbin/init_runroot f28-build", suffix="init"),
|
|
self._ssh_call("install key lorax automake", suffix="install_packages"),
|
|
self._ssh_call("run key df -h"),
|
|
self._ssh_call(
|
|
"run key rpm -qa --qf='%{name}-%{version}-%{release}.%{arch}\n'",
|
|
suffix="rpms",
|
|
),
|
|
]
|
|
)
|
|
|
|
@mock.patch("pungi.runroot.run")
|
|
def test_run_templates_no_init(self, run):
|
|
self.compose.conf[
|
|
"runroot_ssh_install_packages_template"
|
|
] = "install {packages}"
|
|
self.compose.conf["runroot_ssh_run_template"] = "run {command}"
|
|
|
|
run.return_value = (0, "key\n")
|
|
self.runroot.run(
|
|
"df -h",
|
|
log_file="/foo/runroot.log",
|
|
arch="x86_64",
|
|
packages=["lorax", "automake"],
|
|
)
|
|
run.assert_has_calls(
|
|
[
|
|
self._ssh_call("install lorax automake", suffix="install_packages"),
|
|
self._ssh_call("run df -h"),
|
|
self._ssh_call(
|
|
"run rpm -qa --qf='%{name}-%{version}-%{release}.%{arch}\n'",
|
|
suffix="rpms",
|
|
),
|
|
]
|
|
)
|
|
|
|
@mock.patch("pungi.runroot.run")
|
|
def test_run_templates_no_packages(self, run):
|
|
self.compose.conf[
|
|
"runroot_ssh_install_packages_template"
|
|
] = "install {packages}"
|
|
self.compose.conf["runroot_ssh_run_template"] = "run {command}"
|
|
|
|
run.return_value = (0, "key\n")
|
|
self.runroot.run("df -h", log_file="/foo/runroot.log", arch="x86_64")
|
|
run.assert_has_calls(
|
|
[
|
|
self._ssh_call("run df -h"),
|
|
self._ssh_call(
|
|
"run rpm -qa --qf='%{name}-%{version}-%{release}.%{arch}\n'",
|
|
suffix="rpms",
|
|
),
|
|
]
|
|
)
|
|
|
|
@mock.patch("pungi.runroot.run")
|
|
def test_run_templates_no_install_packages(self, run):
|
|
self.compose.conf["runroot_ssh_run_template"] = "run {command}"
|
|
|
|
run.return_value = (0, "key\n")
|
|
self.runroot.run(
|
|
"df -h",
|
|
log_file="/foo/runroot.log",
|
|
arch="x86_64",
|
|
packages=["lorax", "automake"],
|
|
)
|
|
run.assert_has_calls(
|
|
[
|
|
self._ssh_call("run df -h"),
|
|
self._ssh_call(
|
|
"run rpm -qa --qf='%{name}-%{version}-%{release}.%{arch}\n'",
|
|
suffix="rpms",
|
|
),
|
|
]
|
|
)
|
|
|
|
@mock.patch("pungi.runroot.run")
|
|
def test_run_templates_output_dir(self, run):
|
|
self.compose.conf["runroot_ssh_run_template"] = "run {command}"
|
|
|
|
run.return_value = (0, "key\n")
|
|
self.runroot.run(
|
|
"df -h",
|
|
log_file="/foo/runroot.log",
|
|
arch="x86_64",
|
|
packages=["lorax", "automake"],
|
|
chown_paths=["/mnt/foo/compose", "/mnt/foo/x"],
|
|
)
|
|
run.assert_has_calls(
|
|
[
|
|
self._ssh_call(
|
|
"run df -h ; EXIT_CODE=$? ; "
|
|
"chmod -R a+r /mnt/foo/compose /mnt/foo/x ; "
|
|
"chown -R %d /mnt/foo/compose /mnt/foo/x ; exit $EXIT_CODE"
|
|
% os.getuid()
|
|
),
|
|
self._ssh_call(
|
|
"run rpm -qa --qf='%{name}-%{version}-%{release}.%{arch}\n'",
|
|
suffix="rpms",
|
|
),
|
|
]
|
|
)
|
|
|
|
|
|
class TestRunrootKoji(helpers.PungiTestCase):
|
|
def setUp(self):
|
|
super(TestRunrootKoji, self).setUp()
|
|
self.compose = helpers.DummyCompose(
|
|
self.topdir,
|
|
{"runroot": True, "runroot_tag": "f28-build"},
|
|
)
|
|
|
|
self.runroot = Runroot(self.compose)
|
|
|
|
def test_has_losetup_error(self):
|
|
self.assertFalse(self.runroot._has_losetup_error(None))
|
|
|
|
with mock.patch("pungi.runroot.open", mock.mock_open(read_data="")):
|
|
self.assertFalse(self.runroot._has_losetup_error("/foo_log_dir"))
|
|
|
|
with mock.patch(
|
|
"pungi.runroot.open",
|
|
mock.mock_open(read_data="losetup: cannot find an unused loop device"),
|
|
):
|
|
self.assertTrue(self.runroot._has_losetup_error("/bar_log_dir"))
|
|
|
|
@mock.patch("pungi.runroot.kojiwrapper.KojiWrapper")
|
|
def test_run_koji_retry(self, mock_kojiwrapper):
|
|
self.compose.conf["koji_profile"] = "test"
|
|
mock_kojiwrapper.return_value.get_runroot_cmd.return_value = ["df -h"]
|
|
mock_kojiwrapper.return_value.run_runroot_cmd.side_effect = [
|
|
{"retcode": 1, "task_id": 1},
|
|
{"retcode": 0, "task_id": 2},
|
|
]
|
|
self.runroot._has_losetup_error = mock.Mock(side_effect=[True, False])
|
|
self.runroot._run_koji("")
|
|
self.assertEqual(mock_kojiwrapper.return_value.run_runroot_cmd.call_count, 2)
|