diff --git a/awx_collection/test/awx/test_workflow_job_template_node.py b/awx_collection/test/awx/test_workflow_job_template_node.py index 7fdc606dc5..b603ea2fd8 100644 --- a/awx_collection/test/awx/test_workflow_job_template_node.py +++ b/awx_collection/test/awx/test_workflow_job_template_node.py @@ -4,7 +4,7 @@ __metaclass__ = type import pytest -from awx.main.models import WorkflowJobTemplateNode, WorkflowJobTemplate, JobTemplate +from awx.main.models import WorkflowJobTemplateNode, WorkflowJobTemplate, JobTemplate, UnifiedJobTemplate @pytest.fixture @@ -60,16 +60,17 @@ def test_create_workflow_job_template_node_no_template(run_module, admin_user, w 'identifier': this_identifier, 'workflow_job_template': wfjt.name, 'organization': wfjt.organization.name, - 'approval_node': {'name': 'foo-jt'} + 'approval_node': {'name': 'foo-jt-approval'} }, admin_user) assert not result.get('failed', False), result.get('msg', result) assert result.get('changed', False), result node = WorkflowJobTemplateNode.objects.get(identifier=this_identifier) + approval_node = UnifiedJobTemplate.objects.get(name='foo-jt-approval') assert node.identifier == this_identifier assert node.workflow_job_template_id == wfjt.id - assert node.unified_job_template_id is None + assert node.unified_job_template_id is approval_node.id @pytest.mark.django_db