def test_updateWorkflow(self):
self.c.create_workflow_from_template('emptyWorkflow', 'emptyWorkflow')
self.c.update_workflow_name('emptyWorkflow', 'emptyWorkflow',
'newPlaybookName', 'newWorkflowName')
old_key = _WorkflowKey('emptyWorkflow', 'emptyWorkflow')
new_key = _WorkflowKey('newPlaybookName', 'newWorkflowName')
self.assertEqual(len(self.c.workflows), 3)
self.assertNotIn(old_key, self.c.workflows)
self.assertIn(new_key, self.c.workflows)
def test_load_workflows_from_file(self):
path = config.test_workflows_path + "tieredWorkflow.workflow"
playbook_name = "testPlaybook"
self.controller.load_workflows_from_file(
path=path, playbook_override=playbook_name)
key = _WorkflowKey(playbook_name, "parentWorkflow")
self.assertTrue(key in self.controller.workflows)
key = _WorkflowKey(playbook_name, "childWorkflow")
self.assertTrue(key in self.controller.workflows)
def setUp(self):
self.c = controller.Controller()
self.c.load_workflows_from_file(path=config.test_workflows_path +
'basicWorkflowTest.playbook')
self.workflow_name = _WorkflowKey('basicWorkflowTest',
'helloWorldWorkflow')
self.testWorkflow = self.c.workflows[self.workflow_name]
def setUp(self):
self.c = controller.Controller()
self.c.loadWorkflowsFromFile(path=config.testWorkflowsPath +
'basicWorkflowTest.workflow')
self.workflow_name = _WorkflowKey('basicWorkflowTest',
'helloWorldWorkflow')
self.testWorkflow = self.c.workflows[self.workflow_name]
def test_removeWorkflow(self):
initial_workflows = list(self.c.workflows.keys())
self.c.create_workflow_from_template('emptyWorkflow', 'emptyWorkflow')
self.assertEqual(len(self.c.workflows), 3)
success = self.c.removeWorkflow('emptyWorkflow', 'emptyWorkflow')
self.assertTrue(success)
self.assertEqual(len(self.c.workflows), 2)
key = _WorkflowKey('emptyWorkflow', 'emptyWorkflow')
self.assertNotIn(key, self.c.workflows)
orderless_list_compare(self, list(self.c.workflows.keys()),
initial_workflows)
def test_load_workflow_from_file(self):
path = '{0}{1}{2}.workflow'.format(paths.workflows_path, sep, "test")
playbook_name = "testPlaybook"
workflow_name = "helloWorldWorkflow"
key = _WorkflowKey(playbook_name, workflow_name)
ret = self.controller.load_workflow_from_file(
path=path,
workflow_name=workflow_name,
playbook_override=playbook_name)
self.assertTrue(ret)
self.assertTrue(key in self.controller.workflows)
def test_load_all_workflows_from_directory(self):
path = config.test_workflows_path
workflows = helpers.locate_workflows_in_directory(path)
keys = []
for workflow in workflows:
for name in helpers.get_workflow_names_from_file(
os.path.join(config.test_workflows_path, workflow)):
playbook_name = workflow.split('.')[0]
keys.append(_WorkflowKey(playbook_name, name))
self.controller.load_all_workflows_from_directory(path=path)
for key in keys:
self.assertTrue(key in self.controller.workflows)
def test_load_workflow_from_file(self):
path = '{0}{1}.playbook'.format(
config.test_workflows_path_with_generated, "test")
playbook_name = "testPlaybook"
workflow_name = "helloWorldWorkflow"
key = _WorkflowKey(playbook_name, workflow_name)
ret = self.controller.load_workflow_from_file(
path=path,
workflow_name=workflow_name,
playbook_override=playbook_name)
self.assertTrue(ret)
self.assertTrue(key in self.controller.workflows)
def test_load_all_workflows_from_directory(self):
path = config.test_workflows_path
workflows = helpers.locate_workflows_in_directory(path)
keys = []
invalid_workflows = [
'invalidActionWorkflow', 'invalidAppWorkflow',
'tooManyStepInputsWorkflow', 'invalidInputWorkflow'
]
for workflow in workflows:
for name in helpers.get_workflow_names_from_file(
os.path.join(config.test_workflows_path, workflow)):
playbook_name = workflow.split('.')[0]
if playbook_name not in invalid_workflows:
keys.append(_WorkflowKey(playbook_name, name))
self.controller.load_all_workflows_from_directory(path=path)
for key in keys:
self.assertTrue(key in self.controller.workflows)
