51a55e1f34
* Fixing all rules will be in subsequent commits Change-Id: If972c22297a7c41b72a9ee7051fad22eb1c44ed9
103 lines
4.1 KiB
Python
103 lines
4.1 KiB
Python
from base import MistralBase
|
|
|
|
|
|
class Workbooks(MistralBase):
|
|
|
|
def test_get_workbook_list(self):
|
|
wbs = self.mistral_client.workbooks.list()
|
|
self.assertIsInstance(wbs, list)
|
|
|
|
def test_create_workbook(self):
|
|
wbs = self.mistral_client.workbooks.list()
|
|
self.mistral_client.workbooks.create("new_wb")
|
|
wbs_with_new_wb = self.mistral_client.workbooks.list()
|
|
|
|
self.assertEqual(len(wbs) + 1, len(wbs_with_new_wb))
|
|
self.assertTrue(self.assert_item_in_list(
|
|
wbs_with_new_wb, name="new_wb"))
|
|
|
|
def test_get_workbook(self):
|
|
received_wb = self.mistral_client.workbooks.get("wb")
|
|
self.assertEqual(self.wb.name, received_wb.name)
|
|
|
|
def test_update_workbook(self):
|
|
updated_wb = self.mistral_client.workbooks.update(
|
|
"wb", "New Description", ["tags"])
|
|
|
|
self.assertEqual(self.wb.name, updated_wb.name)
|
|
self.assertEqual("New Description", updated_wb.description)
|
|
self.assertEqual(["tags"], updated_wb.tags)
|
|
|
|
def test_upload_get_definition(self):
|
|
self.mistral_client.workbooks.upload_definition("wb", self.definition)
|
|
received_definition = \
|
|
self.mistral_client.workbooks.get_definition("wb")
|
|
|
|
self.assertEqual(self.definition, received_definition)
|
|
|
|
|
|
class Executions(MistralBase):
|
|
|
|
def test_create_execution(self):
|
|
execution = self.create_execution()
|
|
self.assertEqual("wb", execution.workbook_name)
|
|
self.assertNotEqual(execution.id, None)
|
|
|
|
def test_update_execution(self):
|
|
execution = self.create_execution()
|
|
updated_exec = self.mistral_client.executions.update(
|
|
"wb", execution.id, "ERROR")
|
|
self.assertEqual("ERROR", updated_exec.state)
|
|
updated_exec = self.mistral_client.executions.update(
|
|
None, execution.id, "SUCCESS")
|
|
self.assertEqual("SUCCESS", updated_exec.state)
|
|
|
|
def test_list_executions(self):
|
|
execution = self.create_execution()
|
|
exec_list = self.mistral_client.executions.list(None)
|
|
self.assertTrue(self.assert_item_in_list(
|
|
exec_list, id=execution.id))
|
|
exec_list = self.mistral_client.executions.list("wb")
|
|
self.assertTrue(self.assert_item_in_list(
|
|
exec_list, id=execution.id, workbook_name="wb"))
|
|
|
|
def test_get_execution(self):
|
|
execution = self.create_execution()
|
|
received_exec = self.mistral_client.executions.get(None, execution.id)
|
|
self.assertEqual(execution.id, received_exec.id)
|
|
received_exec = self.mistral_client.executions.get("wb", execution.id)
|
|
self.assertEqual(execution.id, received_exec.id)
|
|
|
|
|
|
class Tasks(MistralBase):
|
|
|
|
def test_list_tasks(self):
|
|
execution = self.create_execution()
|
|
tasks_list = self.mistral_client.tasks.list(None, None)
|
|
self.assertIsInstance(tasks_list, list)
|
|
tasks_list = self.mistral_client.tasks.list(
|
|
execution.workbook_name, None)
|
|
self.assertIsInstance(tasks_list, list)
|
|
tasks_list = self.mistral_client.tasks.list(None, execution.id)
|
|
self.assertIsInstance(tasks_list, list)
|
|
tasks_list = self.mistral_client.tasks.list(
|
|
execution.workbook_name, execution.id)
|
|
self.assertIsInstance(tasks_list, list)
|
|
|
|
def test_get_task(self):
|
|
execution = self.create_execution()
|
|
task = self.mistral_client.tasks.list(None, None)[-1]
|
|
received_task = self.mistral_client.tasks.get(None, None, task.id)
|
|
self.assertIsNotNone(received_task)
|
|
task = self.mistral_client.tasks.list("wb", None)[-1]
|
|
received_task = self.mistral_client.tasks.get("wb", None, task.id)
|
|
self.assertIsNotNone(received_task)
|
|
task = self.mistral_client.tasks.list(None, execution.id)[-1]
|
|
received_task = self.mistral_client.tasks.get(
|
|
None, execution.id, task.id)
|
|
self.assertIsNotNone(received_task)
|
|
task = self.mistral_client.tasks.list("wb", execution.id)[-1]
|
|
received_task = self.mistral_client.tasks.get(
|
|
"wb", execution.id, task.id)
|
|
self.assertIsNotNone(received_task)
|