pungi/tests/test_liveimagesphase.py
Haibo Lin c0193c9fca Fix flake8 complaints - E501
E501 line too long (92 > 88 characters)
E501 line too long (103 > 88 characters)
...

JIRA: COMPOSE-4108
Signed-off-by: Haibo Lin <hlin@redhat.com>
2020-02-07 14:36:46 +08:00

1054 lines
38 KiB
Python

# -*- coding: utf-8 -*-
import mock
import six
from pungi.phases.live_images import LiveImagesPhase, CreateLiveImageThread
from tests.helpers import DummyCompose, PungiTestCase, boom
class TestLiveImagesPhase(PungiTestCase):
@mock.patch("pungi.phases.live_images.ThreadPool")
def test_live_image_build(self, ThreadPool):
compose = DummyCompose(
self.topdir,
{
"live_images": [
(
"^Client$",
{
"amd64": {
"kickstart": "test.ks",
"repo": [
"http://example.com/repo/",
"Everything",
"Server-optional",
],
"release": None,
"target": "f27",
}
},
)
],
},
)
compose.setup_optional()
self.assertValidConfig(compose.conf)
phase = LiveImagesPhase(compose)
phase.run()
# assert at least one thread was started
self.assertTrue(phase.pool.add.called)
self.maxDiff = None
six.assertCountEqual(
self,
phase.pool.queue_put.mock_calls,
[
mock.call(
(
compose,
{
"ks_file": "test.ks",
"build_arch": "amd64",
"dest_dir": self.topdir + "/compose/Client/amd64/iso",
"scratch": False,
"repos": [
self.topdir + "/compose/Client/amd64/os",
"http://example.com/repo/",
self.topdir + "/compose/Everything/amd64/os",
self.topdir + "/compose/Server-optional/amd64/os",
],
"label": "",
"name": None,
"filename": "image-name",
"version": "25",
"specfile": None,
"sign": False,
"type": "live",
"release": "20151203.t.0",
"subvariant": "Client",
"failable_arches": [],
"ksurl": None,
"target": "f27",
},
compose.variants["Client"],
"amd64",
)
)
],
)
six.assertCountEqual(
self,
compose.get_image_name.mock_calls,
[
mock.call(
"amd64",
compose.variants["Client"],
disc_num=None,
disc_type="live",
format="%(compose_id)s-%(variant)s-%(arch)s-%(disc_type)s%(disc_num)s%(suffix)s", # noqa: E501
)
],
)
@mock.patch("pungi.phases.live_images.ThreadPool")
def test_live_image_build_single_repo_from(self, ThreadPool):
compose = DummyCompose(
self.topdir,
{
"live_images": [
(
"^Client$",
{
"amd64": {
"kickstart": "test.ks",
"repo": ["http://example.com/repo/", "Everything"],
"release": None,
"target": "f27",
}
},
)
],
},
)
self.assertValidConfig(compose.conf)
phase = LiveImagesPhase(compose)
phase.run()
# assert at least one thread was started
self.assertTrue(phase.pool.add.called)
self.maxDiff = None
six.assertCountEqual(
self,
phase.pool.queue_put.mock_calls,
[
mock.call(
(
compose,
{
"ks_file": "test.ks",
"build_arch": "amd64",
"dest_dir": self.topdir + "/compose/Client/amd64/iso",
"scratch": False,
"repos": [
self.topdir + "/compose/Client/amd64/os",
"http://example.com/repo/",
self.topdir + "/compose/Everything/amd64/os",
],
"label": "",
"name": None,
"filename": "image-name",
"version": "25",
"specfile": None,
"sign": False,
"type": "live",
"release": "20151203.t.0",
"subvariant": "Client",
"failable_arches": [],
"ksurl": None,
"target": "f27",
},
compose.variants["Client"],
"amd64",
)
)
],
)
@mock.patch("pungi.phases.live_images.ThreadPool")
def test_live_image_build_without_rename(self, ThreadPool):
compose = DummyCompose(
self.topdir,
{
"live_images_no_rename": True,
"live_images": [
(
"^Client$",
{
"amd64": {
"kickstart": "test.ks",
"repo": ["http://example.com/repo/", "Everything"],
"release": None,
"target": "f27",
}
},
)
],
},
)
self.assertValidConfig(compose.conf)
phase = LiveImagesPhase(compose)
phase.run()
# assert at least one thread was started
self.assertTrue(phase.pool.add.called)
self.maxDiff = None
six.assertCountEqual(
self,
phase.pool.queue_put.mock_calls,
[
mock.call(
(
compose,
{
"ks_file": "test.ks",
"build_arch": "amd64",
"dest_dir": self.topdir + "/compose/Client/amd64/iso",
"scratch": False,
"repos": [
self.topdir + "/compose/Client/amd64/os",
"http://example.com/repo/",
self.topdir + "/compose/Everything/amd64/os",
],
"label": "",
"name": None,
"filename": None,
"version": "25",
"specfile": None,
"sign": False,
"type": "live",
"release": "20151203.t.0",
"subvariant": "Client",
"failable_arches": [],
"ksurl": None,
"target": "f27",
},
compose.variants["Client"],
"amd64",
)
)
],
)
@mock.patch("pungi.phases.live_images.ThreadPool")
def test_live_image_build_two_images(self, ThreadPool):
compose = DummyCompose(
self.topdir,
{
"live_images": [
(
"^Client$",
{
"amd64": [
{
"kickstart": "test.ks",
"repo": ["http://example.com/repo/", "Everything"],
"target": "f27",
},
{
"kickstart": "another.ks",
"repo": ["http://example.com/repo/", "Everything"],
"target": "f27",
},
]
},
)
],
},
)
self.assertValidConfig(compose.conf)
phase = LiveImagesPhase(compose)
phase.run()
# assert at least one thread was started
self.assertTrue(phase.pool.add.called)
self.maxDiff = None
six.assertCountEqual(
self,
phase.pool.queue_put.mock_calls,
[
mock.call(
(
compose,
{
"ks_file": "test.ks",
"build_arch": "amd64",
"dest_dir": self.topdir + "/compose/Client/amd64/iso",
"scratch": False,
"repos": [
self.topdir + "/compose/Client/amd64/os",
"http://example.com/repo/",
self.topdir + "/compose/Everything/amd64/os",
],
"label": "",
"name": None,
"filename": "image-name",
"version": "25",
"specfile": None,
"sign": False,
"type": "live",
"release": None,
"subvariant": "Client",
"failable_arches": [],
"target": "f27",
"ksurl": None,
},
compose.variants["Client"],
"amd64",
)
),
mock.call(
(
compose,
{
"ks_file": "another.ks",
"build_arch": "amd64",
"dest_dir": self.topdir + "/compose/Client/amd64/iso",
"scratch": False,
"repos": [
self.topdir + "/compose/Client/amd64/os",
"http://example.com/repo/",
self.topdir + "/compose/Everything/amd64/os",
],
"label": "",
"name": None,
"filename": "image-name",
"version": "25",
"specfile": None,
"sign": False,
"type": "live",
"release": None,
"subvariant": "Client",
"failable_arches": [],
"target": "f27",
"ksurl": None,
},
compose.variants["Client"],
"amd64",
)
),
],
)
@mock.patch("pungi.phases.live_images.ThreadPool")
def test_spin_appliance(self, ThreadPool):
compose = DummyCompose(
self.topdir,
{
"live_images": [
(
"^Client$",
{
"amd64": {
"kickstart": "test.ks",
"ksurl": "https://git.example.com/kickstarts.git?#CAFEBABE", # noqa: E501
"repo": ["http://example.com/repo/", "Everything"],
"type": "appliance",
"target": "f27",
}
},
)
],
},
)
self.assertValidConfig(compose.conf)
phase = LiveImagesPhase(compose)
phase.run()
# assert at least one thread was started
self.assertTrue(phase.pool.add.called)
self.maxDiff = None
six.assertCountEqual(
self,
phase.pool.queue_put.mock_calls,
[
mock.call(
(
compose,
{
"ks_file": "test.ks",
"build_arch": "amd64",
"dest_dir": self.topdir + "/compose/Client/amd64/images",
"scratch": False,
"repos": [
self.topdir + "/compose/Client/amd64/os",
"http://example.com/repo/",
self.topdir + "/compose/Everything/amd64/os",
],
"label": "",
"name": None,
"filename": "image-name",
"version": "25",
"specfile": None,
"sign": False,
"type": "appliance",
"release": None,
"subvariant": "Client",
"failable_arches": [],
"target": "f27",
"ksurl": "https://git.example.com/kickstarts.git?#CAFEBABE",
},
compose.variants["Client"],
"amd64",
)
)
],
)
@mock.patch("pungi.phases.live_images.ThreadPool")
def test_spin_appliance_phase_global_settings(self, ThreadPool):
compose = DummyCompose(
self.topdir,
{
"live_images_ksurl": "https://git.example.com/kickstarts.git?#CAFEBABE",
"live_images_release": None,
"live_images_version": "Rawhide",
"live_images_target": "f27",
"live_images": [
(
"^Client$",
{
"amd64": {
"kickstart": "test.ks",
"repo": ["http://example.com/repo/", "Everything"],
"type": "appliance",
}
},
)
],
},
)
self.assertValidConfig(compose.conf)
phase = LiveImagesPhase(compose)
phase.run()
# assert at least one thread was started
self.assertTrue(phase.pool.add.called)
self.maxDiff = None
six.assertCountEqual(
self,
phase.pool.queue_put.mock_calls,
[
mock.call(
(
compose,
{
"ks_file": "test.ks",
"build_arch": "amd64",
"dest_dir": self.topdir + "/compose/Client/amd64/images",
"scratch": False,
"repos": [
self.topdir + "/compose/Client/amd64/os",
"http://example.com/repo/",
self.topdir + "/compose/Everything/amd64/os",
],
"label": "",
"name": None,
"filename": "image-name",
"version": "Rawhide",
"specfile": None,
"sign": False,
"type": "appliance",
"release": "20151203.t.0",
"subvariant": "Client",
"failable_arches": [],
"target": "f27",
"ksurl": "https://git.example.com/kickstarts.git?#CAFEBABE",
},
compose.variants["Client"],
"amd64",
)
)
],
)
@mock.patch("pungi.phases.live_images.ThreadPool")
def test_spin_appliance_global_settings(self, ThreadPool):
compose = DummyCompose(
self.topdir,
{
"global_ksurl": "https://git.example.com/kickstarts.git?#CAFEBABE",
"global_release": None,
"global_version": "Rawhide",
"global_target": "f27",
"live_images": [
(
"^Client$",
{
"amd64": {
"kickstart": "test.ks",
"repo": ["http://example.com/repo/", "Everything"],
"type": "appliance",
}
},
)
],
},
)
self.assertValidConfig(compose.conf)
phase = LiveImagesPhase(compose)
phase.run()
# assert at least one thread was started
self.assertTrue(phase.pool.add.called)
self.maxDiff = None
six.assertCountEqual(
self,
phase.pool.queue_put.mock_calls,
[
mock.call(
(
compose,
{
"ks_file": "test.ks",
"build_arch": "amd64",
"dest_dir": self.topdir + "/compose/Client/amd64/images",
"scratch": False,
"repos": [
self.topdir + "/compose/Client/amd64/os",
"http://example.com/repo/",
self.topdir + "/compose/Everything/amd64/os",
],
"label": "",
"name": None,
"filename": "image-name",
"version": "Rawhide",
"specfile": None,
"sign": False,
"type": "appliance",
"release": "20151203.t.0",
"subvariant": "Client",
"failable_arches": [],
"target": "f27",
"ksurl": "https://git.example.com/kickstarts.git?#CAFEBABE",
},
compose.variants["Client"],
"amd64",
)
)
],
)
@mock.patch("pungi.phases.live_images.ThreadPool")
def test_live_image_build_custom_type(self, ThreadPool):
compose = DummyCompose(
self.topdir,
{
"disc_types": {"live": "Live"},
"live_images": [
(
"^Client$",
{
"amd64": {
"kickstart": "test.ks",
"repo": ["http://example.com/repo/", "Everything"],
"release": None,
"target": "f27",
}
},
)
],
},
)
self.assertValidConfig(compose.conf)
phase = LiveImagesPhase(compose)
phase.run()
# assert at least one thread was started
self.assertTrue(phase.pool.add.called)
self.maxDiff = None
six.assertCountEqual(
self,
phase.pool.queue_put.mock_calls,
[
mock.call(
(
compose,
{
"ks_file": "test.ks",
"build_arch": "amd64",
"dest_dir": self.topdir + "/compose/Client/amd64/iso",
"scratch": False,
"repos": [
self.topdir + "/compose/Client/amd64/os",
"http://example.com/repo/",
self.topdir + "/compose/Everything/amd64/os",
],
"label": "",
"name": None,
"filename": "image-name",
"version": "25",
"specfile": None,
"sign": False,
"type": "live",
"release": "20151203.t.0",
"subvariant": "Client",
"failable_arches": [],
"target": "f27",
"ksurl": None,
},
compose.variants["Client"],
"amd64",
)
)
],
)
six.assertCountEqual(
self,
compose.get_image_name.mock_calls,
[
mock.call(
"amd64",
compose.variants["Client"],
disc_num=None,
disc_type="Live",
format="%(compose_id)s-%(variant)s-%(arch)s-%(disc_type)s%(disc_num)s%(suffix)s", # noqa: E501
)
],
)
class TestCreateLiveImageThread(PungiTestCase):
@mock.patch("pungi.phases.live_images.Image")
@mock.patch("shutil.copy2")
@mock.patch("pungi.phases.live_images.run")
@mock.patch("pungi.phases.live_images.KojiWrapper")
def test_process(self, KojiWrapper, run, copy2, Image):
compose = DummyCompose(self.topdir, {"koji_profile": "koji"})
pool = mock.Mock()
cmd = {
"ks_file": "/path/to/ks_file",
"build_arch": "amd64",
"dest_dir": self.topdir + "/compose/Client/amd64/iso",
"scratch": False,
"repos": [
"/repo/amd64/Client",
"http://example.com/repo/",
"/repo/amd64/Everything",
],
"label": "",
"name": None,
"filename": "image-name",
"version": None,
"specfile": None,
"type": "live",
"ksurl": "https://git.example.com/kickstarts.git?#CAFEBABE",
"release": None,
"subvariant": "Something",
"target": "f27",
}
koji_wrapper = KojiWrapper.return_value
koji_wrapper.get_create_image_cmd.return_value = "koji spin-livecd ..."
koji_wrapper.run_blocking_cmd.return_value = {
"retcode": 0,
"output": "some output",
"task_id": 123,
}
koji_wrapper.get_image_path.return_value = ["/path/to/image.iso"]
t = CreateLiveImageThread(pool)
with mock.patch("pungi.phases.live_images.get_file_size") as get_file_size:
get_file_size.return_value = 1024
with mock.patch("pungi.phases.live_images.get_mtime") as get_mtime:
get_mtime.return_value = 13579
with mock.patch("time.sleep"):
t.process((compose, cmd, compose.variants["Client"], "amd64"), 1)
self.assertEqual(
koji_wrapper.run_blocking_cmd.mock_calls,
[
mock.call(
"koji spin-livecd ...",
log_file=self.topdir
+ "/logs/amd64/liveimage-None-None-None.amd64.log",
)
],
)
self.assertEqual(koji_wrapper.get_image_path.mock_calls, [mock.call(123)])
self.assertEqual(
copy2.mock_calls,
[
mock.call(
"/path/to/image.iso",
self.topdir + "/compose/Client/amd64/iso/image-name",
)
],
)
write_manifest_cmd = " && ".join(
[
"cd " + self.topdir + "/compose/Client/amd64/iso",
"isoinfo -R -f -i image-name | grep -v '/TRANS.TBL$' | sort >> image-name.manifest", # noqa: E501
]
)
self.assertEqual(run.mock_calls, [mock.call(write_manifest_cmd)])
self.assertEqual(
koji_wrapper.get_create_image_cmd.mock_calls,
[
mock.call(
"test-Something-Live-amd64",
"20151203.0.t",
"f27",
"amd64",
"/path/to/ks_file",
[
"/repo/amd64/Client",
"http://example.com/repo/",
"/repo/amd64/Everything",
],
image_type="live",
archive=False,
specfile=None,
wait=True,
release=None,
ksurl="https://git.example.com/kickstarts.git?#CAFEBABE",
)
],
)
self.assertEqual(Image.return_value.type, "live")
self.assertEqual(Image.return_value.format, "iso")
self.assertEqual(Image.return_value.path, "Client/amd64/iso/image-name")
self.assertEqual(Image.return_value.size, 1024)
self.assertEqual(Image.return_value.mtime, 13579)
self.assertEqual(Image.return_value.arch, "amd64")
self.assertEqual(Image.return_value.disc_number, 1)
self.assertEqual(Image.return_value.disc_count, 1)
self.assertTrue(Image.return_value.bootable)
self.assertEqual(
compose.im.add.mock_calls,
[mock.call(variant="Client", arch="amd64", image=Image.return_value)],
)
@mock.patch("pungi.phases.live_images.Image")
@mock.patch("shutil.copy2")
@mock.patch("pungi.phases.live_images.run")
@mock.patch("pungi.phases.live_images.KojiWrapper")
def test_process_no_rename(self, KojiWrapper, run, copy2, Image):
compose = DummyCompose(self.topdir, {"koji_profile": "koji"})
pool = mock.Mock()
cmd = {
"ks_file": "/path/to/ks_file",
"build_arch": "amd64",
"dest_dir": self.topdir + "/compose/Client/amd64/iso",
"scratch": False,
"repos": [
"/repo/amd64/Client",
"http://example.com/repo/",
"/repo/amd64/Everything",
],
"label": "",
"name": None,
"filename": None,
"version": None,
"specfile": None,
"type": "live",
"ksurl": "https://git.example.com/kickstarts.git?#CAFEBABE",
"release": None,
"subvariant": "Client",
"target": "f27",
}
koji_wrapper = KojiWrapper.return_value
koji_wrapper.get_create_image_cmd.return_value = "koji spin-livecd ..."
koji_wrapper.run_blocking_cmd.return_value = {
"retcode": 0,
"output": "some output",
"task_id": 123,
}
koji_wrapper.get_image_path.return_value = ["/path/to/image.iso"]
t = CreateLiveImageThread(pool)
with mock.patch("pungi.phases.live_images.get_file_size") as get_file_size:
get_file_size.return_value = 1024
with mock.patch("pungi.phases.live_images.get_mtime") as get_mtime:
get_mtime.return_value = 13579
with mock.patch("time.sleep"):
t.process((compose, cmd, compose.variants["Client"], "amd64"), 1)
self.assertEqual(
koji_wrapper.run_blocking_cmd.mock_calls,
[
mock.call(
"koji spin-livecd ...",
log_file=self.topdir
+ "/logs/amd64/liveimage-None-None-None.amd64.log",
)
],
)
self.assertEqual(koji_wrapper.get_image_path.mock_calls, [mock.call(123)])
self.assertEqual(
copy2.mock_calls,
[
mock.call(
"/path/to/image.iso",
self.topdir + "/compose/Client/amd64/iso/image.iso",
)
],
)
write_manifest_cmd = " && ".join(
[
"cd " + self.topdir + "/compose/Client/amd64/iso",
"isoinfo -R -f -i image.iso | grep -v '/TRANS.TBL$' | sort >> image.iso.manifest", # noqa: E501
]
)
self.assertEqual(run.mock_calls, [mock.call(write_manifest_cmd)])
self.assertEqual(
koji_wrapper.get_create_image_cmd.mock_calls,
[
mock.call(
"test-Client-Live-amd64",
"20151203.0.t",
"f27",
"amd64",
"/path/to/ks_file",
[
"/repo/amd64/Client",
"http://example.com/repo/",
"/repo/amd64/Everything",
],
image_type="live",
archive=False,
specfile=None,
wait=True,
release=None,
ksurl="https://git.example.com/kickstarts.git?#CAFEBABE",
)
],
)
self.assertEqual(Image.return_value.type, "live")
self.assertEqual(Image.return_value.format, "iso")
self.assertEqual(Image.return_value.path, "Client/amd64/iso/image.iso")
self.assertEqual(Image.return_value.size, 1024)
self.assertEqual(Image.return_value.mtime, 13579)
self.assertEqual(Image.return_value.arch, "amd64")
self.assertEqual(Image.return_value.disc_number, 1)
self.assertEqual(Image.return_value.disc_count, 1)
self.assertTrue(Image.return_value.bootable)
self.assertEqual(
compose.im.add.mock_calls,
[mock.call(variant="Client", arch="amd64", image=Image.return_value)],
)
@mock.patch("pungi.phases.live_images.Image")
@mock.patch("shutil.copy2")
@mock.patch("pungi.phases.live_images.run")
@mock.patch("pungi.phases.live_images.KojiWrapper")
def test_process_applicance(self, KojiWrapper, run, copy2, Image):
compose = DummyCompose(self.topdir, {"koji_profile": "koji"})
pool = mock.Mock()
cmd = {
"ks_file": "/path/to/ks_file",
"build_arch": "amd64",
"dest_dir": self.topdir + "/compose/Client/amd64/iso",
"scratch": False,
"repos": [
"/repo/amd64/Client",
"http://example.com/repo/",
"/repo/amd64/Everything",
],
"label": "",
"name": None,
"filename": "image-name",
"version": None,
"specfile": None,
"type": "appliance",
"ksurl": None,
"release": None,
"subvariant": "Client",
"target": "f27",
}
koji_wrapper = KojiWrapper.return_value
koji_wrapper.get_create_image_cmd.return_value = "koji spin-livecd ..."
koji_wrapper.run_blocking_cmd.return_value = {
"retcode": 0,
"output": "some output",
"task_id": 123,
}
koji_wrapper.get_image_path.return_value = ["/path/to/image-a.b-sda.raw.xz"]
t = CreateLiveImageThread(pool)
with mock.patch("pungi.phases.live_images.get_file_size") as get_file_size:
get_file_size.return_value = 1024
with mock.patch("pungi.phases.live_images.get_mtime") as get_mtime:
get_mtime.return_value = 13579
with mock.patch("time.sleep"):
t.process((compose, cmd, compose.variants["Client"], "amd64"), 1)
self.assertEqual(
koji_wrapper.run_blocking_cmd.mock_calls,
[
mock.call(
"koji spin-livecd ...",
log_file=self.topdir
+ "/logs/amd64/liveimage-None-None-None.amd64.log",
)
],
)
self.assertEqual(koji_wrapper.get_image_path.mock_calls, [mock.call(123)])
self.assertEqual(
copy2.mock_calls,
[
mock.call(
"/path/to/image-a.b-sda.raw.xz",
self.topdir + "/compose/Client/amd64/iso/image-name",
)
],
)
self.assertEqual(run.mock_calls, [])
self.assertEqual(
koji_wrapper.get_create_image_cmd.mock_calls,
[
mock.call(
"test-Client-Disk-amd64",
"20151203.0.t",
"f27",
"amd64",
"/path/to/ks_file",
[
"/repo/amd64/Client",
"http://example.com/repo/",
"/repo/amd64/Everything",
],
image_type="appliance",
archive=False,
specfile=None,
wait=True,
release=None,
ksurl=None,
)
],
)
self.assertEqual(Image.return_value.type, "raw-xz")
self.assertEqual(Image.return_value.format, "raw.xz")
self.assertEqual(Image.return_value.path, "Client/amd64/iso/image-name")
self.assertEqual(Image.return_value.size, 1024)
self.assertEqual(Image.return_value.mtime, 13579)
self.assertEqual(Image.return_value.arch, "amd64")
self.assertEqual(Image.return_value.disc_number, 1)
self.assertEqual(Image.return_value.disc_count, 1)
self.assertTrue(Image.return_value.bootable)
self.assertEqual(
compose.im.add.mock_calls,
[mock.call(variant="Client", arch="amd64", image=Image.return_value)],
)
@mock.patch("shutil.copy2")
@mock.patch("pungi.phases.live_images.run")
@mock.patch("pungi.phases.live_images.KojiWrapper")
def test_process_handles_fail(self, KojiWrapper, run, copy2):
compose = DummyCompose(self.topdir, {"koji_profile": "koji"})
pool = mock.Mock()
cmd = {
"ks_file": "/path/to/ks_file",
"build_arch": "amd64",
"dest_dir": "/top/iso_dir/amd64/Client",
"scratch": False,
"repos": [
"/repo/amd64/Client",
"http://example.com/repo/",
"/repo/amd64/Everything",
],
"label": "",
"name": None,
"filename": "image-name",
"version": None,
"specfile": None,
"ksurl": None,
"subvariant": "Client",
"release": "xyz",
"type": "live",
"failable_arches": ["*"],
"target": "f27",
}
koji_wrapper = KojiWrapper.return_value
koji_wrapper.get_create_image_cmd.return_value = "koji spin-livecd ..."
koji_wrapper.run_blocking_cmd.return_value = {
"retcode": 1,
"output": "some output",
"task_id": 123,
}
t = CreateLiveImageThread(pool)
with mock.patch("time.sleep"):
t.process((compose, cmd, compose.variants["Client"], "amd64"), 1)
pool._logger.error.assert_has_calls(
[
mock.call(
"[FAIL] Live (variant Client, arch amd64, subvariant Client) failed, but going on anyway." # noqa: E501
),
mock.call(
"LiveImage task failed: 123. See %s/logs/amd64/liveimage-None-None-xyz.amd64.log for more details." # noqa: E501
% self.topdir
),
]
)
@mock.patch("shutil.copy2")
@mock.patch("pungi.phases.live_images.run")
@mock.patch("pungi.phases.live_images.KojiWrapper")
def test_process_handles_exception(self, KojiWrapper, run, copy2):
compose = DummyCompose(self.topdir, {"koji_profile": "koji"})
pool = mock.Mock()
cmd = {
"ks_file": "/path/to/ks_file",
"build_arch": "amd64",
"dest_dir": "/top/iso_dir/amd64/Client",
"scratch": False,
"repos": [
"/repo/amd64/Client",
"http://example.com/repo/",
"/repo/amd64/Everything",
],
"label": "",
"name": None,
"filename": "image-name",
"version": None,
"specfile": None,
"ksurl": None,
"subvariant": "Client",
"release": "xyz",
"type": "live",
"failable_arches": ["*"],
"target": "f27",
}
koji_wrapper = KojiWrapper.return_value
koji_wrapper.get_create_image_cmd.side_effect = boom
t = CreateLiveImageThread(pool)
with mock.patch("time.sleep"):
t.process((compose, cmd, compose.variants["Client"], "amd64"), 1)
pool._logger.error.assert_has_calls(
[
mock.call(
"[FAIL] Live (variant Client, arch amd64, subvariant Client) failed, but going on anyway." # noqa: E501
),
mock.call("BOOM"),
]
)