def test_add_workload(self, mock_subtask_create, mock_workload):
mock_subtask_create.return_value = self.subtask
subtask = objects.Subtask("bar", title="foo")
name = "w"
description = "descr"
position = 0
runner = {"type": "runner"}
context = {"users": {}}
sla = {"failure_rate": {"max": 0}}
args = {"arg": "xxx"}
hooks = [{"config": {"foo": "bar"}}]
workload = subtask.add_workload(name,
description=description,
position=position,
runner=runner,
context=context,
sla=sla,
args=args,
hooks=hooks)
mock_workload.assert_called_once_with(
task_uuid=self.subtask["task_uuid"],
subtask_uuid=self.subtask["uuid"],
name=name,
description=description,
position=position,
runner=runner,
context=context,
sla=sla,
args=args,
hooks=hooks)
self.assertIs(workload, mock_workload.return_value)
def test_init(self, mock_subtask_create):
mock_subtask_create.return_value = self.subtask
subtask = objects.Subtask("bar", title="foo")
mock_subtask_create.assert_called_once_with("bar",
title="foo",
context=None,
description=None)
self.assertEqual(subtask["uuid"], self.subtask["uuid"])
def test_add_workload(self, mock_subtask_create, mock_workload):
mock_subtask_create.return_value = self.subtask
subtask = objects.Subtask("bar", title="foo")
workload = subtask.add_workload({"bar": "baz"})
mock_workload.assert_called_once_with(self.subtask["task_uuid"],
self.subtask["uuid"],
{"bar": "baz"})
self.assertIs(workload, mock_workload.return_value)
def test_update_status(self, mock_subtask_create, mock_subtask_update):
mock_subtask_create.return_value = self.subtask
subtask = objects.Subtask("bar", title="foo")
subtask.update_status(consts.SubtaskStatus.FINISHED)
mock_subtask_update.assert_called_once_with(
self.subtask["uuid"], {"status": consts.SubtaskStatus.FINISHED})
