External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-03-11 22:49:32 -02:30
Code Issues Packages Projects Releases Wiki Activity

flake8

Browse Source
This commit is contained in:
chris meyers
2018-11-13 13:39:41 -05:00
committed by mabashian
parent 8bb9cfd62a
commit 1b87e11d8f
1 changed files with 12 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
awx/main/tests/unit/scheduler/test_dag_workflow.py
View File

@@ -31,12 +31,14 @@ class WorkflowNodeUJT(WorkflowNodeDNR):
def WorkflowNodeClass(): def WorkflowNodeClass():
return WorkflowNodeBase return WorkflowNodeBase
@pytest.fixture @pytest.fixture
def wf_node_generator(mocker, WorkflowNodeClass): def wf_node_generator(mocker, WorkflowNodeClass):
def fn(**kwargs): def fn(**kwargs):
return WorkflowNodeClass(**kwargs) return WorkflowNodeClass(**kwargs)
return fn return fn
@pytest.fixture @pytest.fixture
def workflow_dag_1(wf_node_generator): def workflow_dag_1(wf_node_generator):
g = WorkflowDAG() g = WorkflowDAG()
@@ -64,6 +66,7 @@ def workflow_dag_1(wf_node_generator):
g.add_edge(nodes[3], nodes[2], "failure_nodes") g.add_edge(nodes[3], nodes[2], "failure_nodes")
return (g, nodes) return (g, nodes)
class TestWorkflowDAG(): class TestWorkflowDAG():
@pytest.fixture @pytest.fixture
def workflow_dag_root_children(self, wf_node_generator): def workflow_dag_root_children(self, wf_node_generator):
@@ -137,6 +140,7 @@ class TestDNR():
assert 1 == len(do_not_run_nodes) assert 1 == len(do_not_run_nodes)
assert nodes[3] == do_not_run_nodes[0] assert nodes[3] == do_not_run_nodes[0]
class TestIsWorkflowDone(): class TestIsWorkflowDone():
@pytest.fixture @pytest.fixture
def WorkflowNodeClass(self): def WorkflowNodeClass(self):
@@ -163,9 +167,9 @@ class TestIsWorkflowDone():
W2 W2
''' '''
nodes[0].job = Job(status='successful') nodes[0].job = Job(status='successful')
do_not_run_nodes = g.mark_dnr_nodes() g.mark_dnr_nodes()
nodes[1].job = Job(status='successful') nodes[1].job = Job(status='successful')
do_not_run_nodes = g.mark_dnr_nodes() g.mark_dnr_nodes()
nodes[2].job = Job(status='waiting') nodes[2].job = Job(status='waiting')
return (g, nodes) return (g, nodes)
@@ -188,9 +192,9 @@ class TestIsWorkflowDone():
F2 F2
''' '''
nodes[0].job = Job(status='successful') nodes[0].job = Job(status='successful')
do_not_run_nodes = g.mark_dnr_nodes() g.mark_dnr_nodes()
nodes[1].job = Job(status='successful') nodes[1].job = Job(status='successful')
do_not_run_nodes = g.mark_dnr_nodes() g.mark_dnr_nodes()
nodes[2].job = Job(status='failure') nodes[2].job = Job(status='failure')
return (g, nodes) return (g, nodes)
@@ -199,13 +203,13 @@ class TestIsWorkflowDone():
is_done, is_failed = g.is_workflow_done() is_done, is_failed = g.is_workflow_done()
assert False == is_done assert is_done is False
assert False == is_failed assert is_failed is False
def test_is_workflow_done_failed(self, workflow_dag_failed): def test_is_workflow_done_failed(self, workflow_dag_failed):
g = workflow_dag_failed[0] g = workflow_dag_failed[0]
is_done, is_failed = g.is_workflow_done() is_done, is_failed = g.is_workflow_done()
assert True == is_done assert is_done is True
assert True == is_failed assert is_failed is True