Format tests with black
JIRA: COMPOSE-4086 Signed-off-by: Haibo Lin <hlin@redhat.com>
This commit is contained in:
Haibo Lin
parent
ef33d00f5b
commit
38142d30ba
51 changed files with 13767 additions and 9180 deletions
|
|
@ -11,15 +11,16 @@ 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"
|
||||
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",
|
||||
},
|
||||
"runroot_tag": "f28-build",
|
||||
})
|
||||
)
|
||||
|
||||
self.runroot = Runroot(self.compose)
|
||||
|
||||
|
|
@ -52,7 +53,7 @@ class TestRunrootOpenSSH(helpers.PungiTestCase):
|
|||
"""
|
||||
logfile = ("/foo/runroot." + suffix + ".log") if suffix else "/foo/runroot.log"
|
||||
return mock.call(
|
||||
['ssh', '-oBatchMode=yes', '-n', '-l', 'root', 'localhost', cmd],
|
||||
["ssh", "-oBatchMode=yes", "-n", "-l", "root", "localhost", cmd],
|
||||
logfile=logfile,
|
||||
show_cmd=True,
|
||||
)
|
||||
|
|
@ -61,12 +62,15 @@ class TestRunrootOpenSSH(helpers.PungiTestCase):
|
|||
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"
|
||||
),
|
||||
])
|
||||
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):
|
||||
|
|
@ -75,92 +79,123 @@ class TestRunrootOpenSSH(helpers.PungiTestCase):
|
|||
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"]))
|
||||
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_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",
|
||||
),
|
||||
])
|
||||
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_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",
|
||||
),
|
||||
])
|
||||
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_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",
|
||||
),
|
||||
])
|
||||
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",
|
||||
),
|
||||
])
|
||||
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 && chmod -R a+r /mnt/foo/compose /mnt/foo/x && "
|
||||
"chown -R %d /mnt/foo/compose /mnt/foo/x" % os.getuid()),
|
||||
self._ssh_call(
|
||||
"run rpm -qa --qf='%{name}-%{version}-%{release}.%{arch}\n'",
|
||||
suffix="rpms",
|
||||
),
|
||||
])
|
||||
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 && chmod -R a+r /mnt/foo/compose /mnt/foo/x && "
|
||||
"chown -R %d /mnt/foo/compose /mnt/foo/x" % os.getuid()
|
||||
),
|
||||
self._ssh_call(
|
||||
"run rpm -qa --qf='%{name}-%{version}-%{release}.%{arch}\n'",
|
||||
suffix="rpms",
|
||||
),
|
||||
]
|
||||
)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue