def test_build_image_indirection(self):
"""Test _build_image_indirection function"""
expected = "Created task: %d" % self.task_id + "\n"
expected += "Task info: %s/taskinfo?taskID=%s" % (self.weburl, self.task_id) + "\n"
with mock.patch('sys.stdout', new_callable=six.StringIO) as stdout:
_build_image_indirection(
self.options, self.task_opts, self.session, [])
self.assert_console_message(stdout, expected)
def test_build_image_indirection(self):
"""Test _build_image_indirection function"""
expected = "Created task: %d" % self.task_id + "\n"
expected += "Task info: %s/taskinfo?taskID=%s" % (self.weburl,
self.task_id) + "\n"
with mock.patch('sys.stdout', new_callable=six.StringIO) as stdout:
_build_image_indirection(self.options, self.task_opts,
self.session, [])
self.assert_console_message(stdout, expected)
def test_build_image_indirection_with_noprogress(self):
"""Test _build_image_indirection function with noprogress option"""
expected = "\n"
expected += "Created task: %d" % self.task_id + "\n"
expected += "Task info: %s/taskinfo?taskID=%s" % (self.weburl, self.task_id) + "\n"
self.task_opts.indirection_template_url = None
self.task_opts.scratch = True
self.task_opts.noprogress = True
with mock.patch('sys.stdout', new_callable=six.StringIO) as stdout:
_build_image_indirection(
self.options, self.task_opts, self.session, [])
self.assert_console_message(stdout, expected)
args, kwargs = self.session.uploadWrapper.call_args
self.assertEqual(kwargs['callback'], None)
def test_build_image_indirection_with_noprogress(self):
"""Test _build_image_indirection function with noprogress option"""
expected = "\n"
expected += "Created task: %d" % self.task_id + "\n"
expected += "Task info: %s/taskinfo?taskID=%s" % (self.weburl,
self.task_id) + "\n"
self.task_opts.indirection_template_url = None
self.task_opts.scratch = True
self.task_opts.noprogress = True
with mock.patch('sys.stdout', new_callable=six.StringIO) as stdout:
_build_image_indirection(self.options, self.task_opts,
self.session, [])
self.assert_console_message(stdout, expected)
args, kwargs = self.session.uploadWrapper.call_args
self.assertEqual(kwargs['callback'], None)
def test_build_image_indirection_expections(self):
"""Test _build_image_indirection all exceptions"""
# Case 1. sanity check for utility image options
self.task_opts.utility_image_task = None
self.task_opts.utility_image_build = None
expected = "You must specify either a utility-image task or build ID/NVR"
with self.assertRaises(koji.GenericError) as cm:
_build_image_indirection(self.options, self.task_opts,
self.session, [])
self.assertEqual(str(cm.exception), expected)
self.activate_session.assert_not_called()
# Case 2. sanity check for base image options
self.task_opts.utility_image_build = 'image-utils-26.1'
self.task_opts.base_image_task = None
self.task_opts.base_image_build = None
expected = "You must specify either a base-image task or build ID/NVR"
with self.assertRaises(koji.GenericError) as cm:
_build_image_indirection(self.options, self.task_opts,
self.session, [])
self.assertEqual(str(cm.exception), expected)
self.activate_session.assert_not_called()
self.task_opts.base_image_build = 'image-base-26.1'
# Case 3. missing required options
required = [
'name', 'version', 'arch', 'target', 'indirection_template',
'results_loc'
]
for r in required:
orig = getattr(self.task_opts, r)
setattr(self.task_opts, r, None)
expected = "Missing the following required options: "
expected += "--" + r.replace('_', '-') + "\n"
with self.assertRaises(koji.GenericError) as cm, \
mock.patch('sys.stdout', new_callable=six.StringIO) as stdout:
_build_image_indirection(self.options, self.task_opts,
self.session, [])
self.assert_console_message(stdout, expected)
self.assertEqual(str(cm.exception),
"Missing required options specified above")
self.activate_session.assert_not_called()
setattr(self.task_opts, r, orig)
# Case 4. target not found error
self.session.getBuildTarget.return_value = {}
expected = "Unknown build target: %s" % {}
with self.assertRaises(koji.GenericError) as cm:
_build_image_indirection(self.options, self.task_opts,
self.session, [])
self.assertEqual(str(cm.exception), expected)
self.activate_session.assert_called_with(self.session, self.options)
# Case 5. tag not found error
self.session.getBuildTarget.return_value = self.build_target
self.session.getTag.return_value = {}
expected = "Unknown destination tag: %s" % self.build_target[
'dest_tag_name']
with self.assertRaises(koji.GenericError) as cm:
_build_image_indirection(self.options, self.task_opts,
self.session, [])
self.assertEqual(str(cm.exception), expected)
self.activate_session.assert_called_with(self.session, self.options)
# Case 6. --indirection-template-url is not given and --scatch is not set
self.session.getTag.return_value = self.dest_tag
self.task_opts.indirection_template_url = None
self.task_opts.scratch = None
expected = "Non-scratch builds must provide a URL for the indirection template"
with self.assertRaises(koji.GenericError) as cm:
_build_image_indirection(self.options, self.task_opts,
self.session, [])
self.assertEqual(str(cm.exception), expected)
self.activate_session.assert_called_with(self.session, self.options)
def test_build_image_indirection_expections(self):
"""Test _build_image_indirection all exceptions"""
# Case 1. sanity check for utility image options
self.task_opts.utility_image_task = None
self.task_opts.utility_image_build = None
expected = "You must specify either a utility-image task or build ID/NVR"
with self.assertRaises(koji.GenericError) as cm:
_build_image_indirection(
self.options, self.task_opts, self.session, [])
self.assertEqual(str(cm.exception), expected)
self.activate_session.assert_not_called()
# Case 2. sanity check for base image options
self.task_opts.utility_image_build = 'image-utils-26.1'
self.task_opts.base_image_task = None
self.task_opts.base_image_build = None
expected = "You must specify either a base-image task or build ID/NVR"
with self.assertRaises(koji.GenericError) as cm:
_build_image_indirection(
self.options, self.task_opts, self.session, [])
self.assertEqual(str(cm.exception), expected)
self.activate_session.assert_not_called()
self.task_opts.base_image_build = 'image-base-26.1'
# Case 3. missing required options
required = ['name', 'version', 'arch', 'target',
'indirection_template', 'results_loc']
for r in required:
orig = getattr(self.task_opts, r)
setattr(self.task_opts, r, None)
expected = "Missing the following required options: "
expected += "--" + r.replace('_', '-') + "\n"
with self.assertRaises(koji.GenericError) as cm:
with mock.patch('sys.stdout', new_callable=six.StringIO) as stdout:
_build_image_indirection(
self.options, self.task_opts, self.session, [])
self.assert_console_message(stdout, expected)
self.assertEqual(
str(cm.exception), "Missing required options specified above")
self.activate_session.assert_not_called()
setattr(self.task_opts, r, orig)
# Case 4. target not found error
self.session.getBuildTarget.return_value = {}
expected = "Unknown build target: %s" % {}
with self.assertRaises(koji.GenericError) as cm:
_build_image_indirection(
self.options, self.task_opts, self.session, [])
self.assertEqual(str(cm.exception), expected)
self.activate_session.assert_called_with(self.session, self.options)
# Case 5. tag not found error
self.session.getBuildTarget.return_value = self.build_target
self.session.getTag.return_value = {}
expected = "Unknown destination tag: %s" % self.build_target['dest_tag_name']
with self.assertRaises(koji.GenericError) as cm:
_build_image_indirection(
self.options, self.task_opts, self.session, [])
self.assertEqual(str(cm.exception), expected)
self.activate_session.assert_called_with(self.session, self.options)
# Case 6. --indirection-template-url is not given and --scatch is not set
self.session.getTag.return_value = self.dest_tag
self.task_opts.indirection_template_url = None
self.task_opts.scratch = None
expected = "Non-scratch builds must provide a URL for the indirection template"
with self.assertRaises(koji.GenericError) as cm:
_build_image_indirection(
self.options, self.task_opts, self.session, [])
self.assertEqual(str(cm.exception), expected)
self.activate_session.assert_called_with(self.session, self.options)