mirror of
https://github.com/ansible/awx.git
synced 2026-01-21 06:28:01 -03:30
103 lines
4.0 KiB
Python
103 lines
4.0 KiB
Python
from __future__ import absolute_import, division, print_function
|
|
|
|
__metaclass__ = type
|
|
|
|
import pytest
|
|
|
|
from awx.main.models import WorkflowJobTemplate, JobTemplate, Project, InventorySource, Inventory, WorkflowJobTemplateNode
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_create_workflow_job_template(run_module, admin_user, organization, survey_spec, silence_deprecation):
|
|
result = run_module(
|
|
'tower_workflow_template',
|
|
{
|
|
'name': 'foo-workflow',
|
|
'organization': organization.name,
|
|
'extra_vars': {'foo': 'bar', 'another-foo': {'barz': 'bar2'}},
|
|
'survey': survey_spec,
|
|
'survey_enabled': True,
|
|
'state': 'present',
|
|
},
|
|
admin_user,
|
|
)
|
|
|
|
wfjt = WorkflowJobTemplate.objects.get(name='foo-workflow')
|
|
assert wfjt.extra_vars == '{"foo": "bar", "another-foo": {"barz": "bar2"}}'
|
|
|
|
result.pop('invocation', None)
|
|
assert result == {"workflow_template": "foo-workflow", "state": "present", "id": wfjt.id, "changed": True} # TODO: remove after refactor
|
|
|
|
assert wfjt.organization_id == organization.id
|
|
assert wfjt.survey_spec == survey_spec
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_with_nested_workflow(run_module, admin_user, organization, silence_deprecation):
|
|
wfjt1 = WorkflowJobTemplate.objects.create(name='first', organization=organization)
|
|
|
|
result = run_module(
|
|
'tower_workflow_template',
|
|
{'name': 'foo-workflow', 'organization': organization.name, 'schema': [{'workflow': wfjt1.name}], 'state': 'present'},
|
|
admin_user,
|
|
)
|
|
assert not result.get('failed', False), result.get('msg', result)
|
|
|
|
wfjt = WorkflowJobTemplate.objects.get(name='foo-workflow')
|
|
node = wfjt.workflow_nodes.first()
|
|
assert node is not None
|
|
assert node.unified_job_template == wfjt1
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_schema_with_branches(run_module, admin_user, organization, silence_deprecation):
|
|
|
|
proj = Project.objects.create(organization=organization, name='Ansible Examples')
|
|
inv = Inventory.objects.create(organization=organization, name='test-inv')
|
|
jt = JobTemplate.objects.create(project=proj, playbook='helloworld.yml', inventory=inv, name='Hello world')
|
|
inv_src = InventorySource.objects.create(inventory=inv, name='AWS servers', source='ec2')
|
|
|
|
result = run_module(
|
|
'tower_workflow_template',
|
|
{
|
|
'name': 'foo-workflow',
|
|
'organization': organization.name,
|
|
'schema': [
|
|
{
|
|
'job_template': 'Hello world',
|
|
'failure': [{'inventory_source': 'AWS servers', 'success': [{'project': 'Ansible Examples', 'always': [{'job_template': "Hello world"}]}]}],
|
|
}
|
|
],
|
|
'state': 'present',
|
|
},
|
|
admin_user,
|
|
)
|
|
assert not result.get('failed', False), result.get('msg', result)
|
|
|
|
wfjt = WorkflowJobTemplate.objects.get(name='foo-workflow')
|
|
root_nodes = wfjt.workflow_nodes.filter(
|
|
**{
|
|
'%ss_success__isnull' % WorkflowJobTemplateNode.__name__.lower(): True,
|
|
'%ss_failure__isnull' % WorkflowJobTemplateNode.__name__.lower(): True,
|
|
'%ss_always__isnull' % WorkflowJobTemplateNode.__name__.lower(): True,
|
|
}
|
|
)
|
|
assert len(root_nodes) == 1
|
|
node = root_nodes[0]
|
|
assert node.unified_job_template == jt
|
|
second = node.failure_nodes.first()
|
|
assert second.unified_job_template == inv_src
|
|
third = second.success_nodes.first()
|
|
assert third.unified_job_template == proj
|
|
fourth = third.always_nodes.first()
|
|
assert fourth.unified_job_template == jt
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_with_missing_ujt(run_module, admin_user, organization, silence_deprecation):
|
|
result = run_module(
|
|
'tower_workflow_template', {'name': 'foo-workflow', 'organization': organization.name, 'schema': [{'foo': 'bar'}], 'state': 'present'}, admin_user
|
|
)
|
|
assert result.get('failed', False), result
|
|
assert 'You should provide exactly one of the attributes job_template,' in result['msg']
|