1
0
mirror of https://github.com/ansible/awx.git synced 2024-10-30 22:21:13 +03:00

update workflow unit tests to new launch_type addition

This commit is contained in:
AlanCoding 2016-11-19 10:55:48 -05:00
parent 517c5a0188
commit c717bf090d

View File

@ -168,20 +168,24 @@ class TestWorkflowJobNodeJobKWARGS:
Tests for building the keyword arguments that go into creating and Tests for building the keyword arguments that go into creating and
launching a new job that corresponds to a workflow node. launching a new job that corresponds to a workflow node.
""" """
kwargs_base = {'launch_type': 'workflow'}
def test_null_kwargs(self, job_node_no_prompts): def test_null_kwargs(self, job_node_no_prompts):
assert job_node_no_prompts.get_job_kwargs() == {} assert job_node_no_prompts.get_job_kwargs() == self.kwargs_base
def test_inherit_workflow_job_extra_vars(self, job_node_no_prompts): def test_inherit_workflow_job_extra_vars(self, job_node_no_prompts):
workflow_job = job_node_no_prompts.workflow_job workflow_job = job_node_no_prompts.workflow_job
workflow_job.extra_vars = '{"a": 84}' workflow_job.extra_vars = '{"a": 84}'
assert job_node_no_prompts.get_job_kwargs() == {'extra_vars': {'a': 84}} assert job_node_no_prompts.get_job_kwargs() == dict(
extra_vars={'a': 84}, **self.kwargs_base)
def test_char_prompts_and_res_node_prompts(self, job_node_with_prompts): def test_char_prompts_and_res_node_prompts(self, job_node_with_prompts):
assert job_node_with_prompts.get_job_kwargs() == dict( expect_kwargs = dict(
inventory=job_node_with_prompts.inventory.pk, inventory=job_node_with_prompts.inventory.pk,
credential=job_node_with_prompts.credential.pk, credential=job_node_with_prompts.credential.pk,
**example_prompts) **example_prompts)
expect_kwargs.update(self.kwargs_base)
assert job_node_with_prompts.get_job_kwargs() == expect_kwargs
def test_reject_some_node_prompts(self, job_node_with_prompts): def test_reject_some_node_prompts(self, job_node_with_prompts):
job_node_with_prompts.unified_job_template.ask_inventory_on_launch = False job_node_with_prompts.unified_job_template.ask_inventory_on_launch = False
@ -189,13 +193,14 @@ class TestWorkflowJobNodeJobKWARGS:
expect_kwargs = dict(inventory=job_node_with_prompts.inventory.pk, expect_kwargs = dict(inventory=job_node_with_prompts.inventory.pk,
credential=job_node_with_prompts.credential.pk, credential=job_node_with_prompts.credential.pk,
**example_prompts) **example_prompts)
expect_kwargs.update(self.kwargs_base)
expect_kwargs.pop('inventory') expect_kwargs.pop('inventory')
expect_kwargs.pop('job_type') expect_kwargs.pop('job_type')
assert job_node_with_prompts.get_job_kwargs() == expect_kwargs assert job_node_with_prompts.get_job_kwargs() == expect_kwargs
def test_no_accepted_project_node_prompts(self, job_node_no_prompts, project_unit): def test_no_accepted_project_node_prompts(self, job_node_no_prompts, project_unit):
job_node_no_prompts.unified_job_template = project_unit job_node_no_prompts.unified_job_template = project_unit
assert job_node_no_prompts.get_job_kwargs() == {} assert job_node_no_prompts.get_job_kwargs() == self.kwargs_base
class TestWorkflowWarnings: class TestWorkflowWarnings: