3eddcfccd8
F401 'dnf' imported but unused F401 'imp' imported but unused F401 'os' imported but unused F401 'subprocess' imported but unused F401 'sys' imported but unused F401 'yum' imported but unused JIRA: COMPOSE-4108 Signed-off-by: Haibo Lin <hlin@redhat.com>
90 lines
2.6 KiB
Python
90 lines
2.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
try:
|
|
import unittest2 as unittest
|
|
except ImportError:
|
|
import unittest
|
|
|
|
import six
|
|
|
|
from pungi.wrappers.lorax import LoraxWrapper
|
|
|
|
|
|
class LoraxWrapperTest(unittest.TestCase):
|
|
def setUp(self):
|
|
self.lorax = LoraxWrapper()
|
|
|
|
def test_get_command_with_minimal_arguments(self):
|
|
cmd = self.lorax.get_lorax_cmd(
|
|
"product", "version", "release", "/mnt/repo_baseurl", "/mnt/output_dir"
|
|
)
|
|
|
|
self.assertEqual(cmd[0], "lorax")
|
|
six.assertCountEqual(
|
|
self,
|
|
cmd[1:],
|
|
[
|
|
"--product=product",
|
|
"--version=version",
|
|
"--release=release",
|
|
"--source=file:///mnt/repo_baseurl",
|
|
"/mnt/output_dir",
|
|
],
|
|
)
|
|
|
|
def test_get_command_with_all_arguments(self):
|
|
cmd = self.lorax.get_lorax_cmd(
|
|
"product",
|
|
"version",
|
|
"release",
|
|
"/mnt/repo_baseurl",
|
|
"/mnt/output_dir",
|
|
variant="Server",
|
|
bugurl="http://example.com/",
|
|
nomacboot=True,
|
|
noupgrade=True,
|
|
is_final=True,
|
|
buildarch="x86_64",
|
|
volid="VOLUME_ID",
|
|
buildinstallpackages=["bash", "vim"],
|
|
add_template=["t1", "t2"],
|
|
add_arch_template=["ta1", "ta2"],
|
|
add_template_var=["v1", "v2"],
|
|
add_arch_template_var=["va1", "va2"],
|
|
log_dir="/tmp",
|
|
dracut_args=["--foo", "bar"],
|
|
)
|
|
|
|
self.assertEqual(cmd[0], "lorax")
|
|
six.assertCountEqual(
|
|
self,
|
|
cmd[1:],
|
|
[
|
|
"--product=product",
|
|
"--version=version",
|
|
"--release=release",
|
|
"--variant=Server",
|
|
"--source=file:///mnt/repo_baseurl",
|
|
"--bugurl=http://example.com/",
|
|
"--buildarch=x86_64",
|
|
"--volid=VOLUME_ID",
|
|
"--nomacboot",
|
|
"--noupgrade",
|
|
"--isfinal",
|
|
"--installpkgs=bash",
|
|
"--installpkgs=vim",
|
|
"--add-template=t1",
|
|
"--add-template=t2",
|
|
"--add-arch-template=ta1",
|
|
"--add-arch-template=ta2",
|
|
"--add-template-var=v1",
|
|
"--add-template-var=v2",
|
|
"--add-arch-template-var=va1",
|
|
"--add-arch-template-var=va2",
|
|
"--logfile=/tmp/lorax.log",
|
|
"--dracut-arg=--foo",
|
|
"--dracut-arg=bar",
|
|
"/mnt/output_dir",
|
|
],
|
|
)
|