unit test for node prompt values

This commit is contained in:
AlanCoding 2016-09-27 09:17:51 -04:00
parent 1604e2fd02
commit 549273e90f
3 changed files with 43 additions and 4 deletions

View File

@ -2290,7 +2290,7 @@ class WorkflowJobTemplateNodeSerializer(WorkflowNodeBaseSerializer):
def extract_char_prompts(self, data):
char_prompts = {}
for fd in ['job_type', 'job_tags', 'skip_tags', 'limit', 'skip_tags']:
for fd in ['job_type', 'job_tags', 'skip_tags', 'limit']:
if data.get(fd, None):
char_prompts[fd] = data[fd]
return char_prompts

View File

@ -27,7 +27,7 @@ import json
__all__ = ['WorkflowJobTemplate', 'WorkflowJob', 'WorkflowJobOptions', 'WorkflowJobNode', 'WorkflowJobTemplateNode',]
CHAR_PROMPTS_LIST = ['job_type', 'job_tags', 'skip_tags', 'limit', 'skip_tags']
CHAR_PROMPTS_LIST = ['job_type', 'job_tags', 'skip_tags', 'limit']
class WorkflowNodeBase(CreatedModifiedModel):
class Meta:
@ -173,7 +173,7 @@ class WorkflowJobNode(WorkflowNodeBase):
WJ_json_extra_vars = yaml.safe_load(self.workflow_job.extra_vars)
except yaml.YAMLError:
WJ_json_extra_vars = {}
extra_vars.update(WJ_json_extra_vars)
extra_vars.update(WJ_json_extra_vars)
# TODO: merge artifacts, add ancestor_artifacts to kwargs
if extra_vars:
data['extra_vars'] = extra_vars

View File

@ -1,7 +1,12 @@
import pytest
from awx.main.models.jobs import JobTemplate
from awx.main.models.workflow import WorkflowJobTemplateNode, WorkflowJobInheritNodesMixin, WorkflowJobNode
from awx.main.models.inventory import Inventory
from awx.main.models.credential import Credential
from awx.main.models.workflow import (
WorkflowJobTemplateNode, WorkflowJobInheritNodesMixin,
WorkflowJob, WorkflowJobNode
)
class TestWorkflowJobInheritNodesMixin():
class TestCreateWorkflowJobNodes():
@ -79,3 +84,37 @@ class TestWorkflowJobInheritNodesMixin():
job_nodes[i].success_nodes.add.assert_any_call(job_nodes[i + 1])
class TestWorkflowJobHelperMethods:
@pytest.fixture
def workflow_job_unit(self):
return WorkflowJob(name='workflow', status='new')
@pytest.fixture
def workflow_job_node_unit(self, workflow_job_unit, job_template_factory):
# note: factory sets ask_inventory_on_launch to true when not provided
jt = job_template_factory(name='example-jt', persisted=False).job_template
return WorkflowJobNode(workflow_job=workflow_job_unit, unified_job_template=jt)
def test_null_kwargs(self, workflow_job_node_unit):
assert workflow_job_node_unit.get_job_kwargs() == {}
def test_inherit_workflow_job_extra_vars(self, workflow_job_node_unit):
workflow_job = workflow_job_node_unit.workflow_job
workflow_job.extra_vars = '{"a": 84}'
assert workflow_job_node_unit.get_job_kwargs() == {'extra_vars': {'a': 84}}
def test_char_prompts_and_res_node_prompts(self, workflow_job_node_unit):
barnyard_kwargs = dict(
job_type='scan',
job_tags='quack',
limit='duck',
skip_tags='oink'
)
workflow_job_node_unit.char_prompts = barnyard_kwargs
inv = Inventory(name='example-inv')
cred = Credential(name='example-inv', kind='ssh', username='asdf', password='asdf')
workflow_job_node_unit.inventory = inv
workflow_job_node_unit.credential = cred
assert workflow_job_node_unit.get_job_kwargs() == dict(
inventory=inv, credential=cred, **barnyard_kwargs)