b0d0c31361
- Collect all tests in one folder 'mistralclient/tests'. - Create new folder 'functional' in the mistralclient/tests for integration functional tests. - Restructure tests, create folder 'v1' for current tests (version 1). - Create new folder 'unit' in the mistralclient/tests and put all unit tests here. - Correct imports in the file with unit tests. Change-Id: If92c11640d7cb61ee8ffcdc7ba99d7ab1061afa7
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)
|