def test_get_workflow_by_id(self): workflow = Workflow('wf_name', 0) playbook = Playbook('test', [workflow]) self.assertEqual(playbook.get_workflow_by_id(workflow.id), workflow)
def test_get_workflow_by_id_no_workflows(self): playbook = Playbook('test', []) self.assertIsNone(playbook.get_workflow_by_id('anything'))