awx/awx/main/tests/unit/models/test_job_template_unit.py

import pytest
import json


def test_missing_project_error(job_template_factory):
    objects = job_template_factory(
        'missing-project-jt',
        organization='org1',
        inventory='inventory1',
        credential='cred1',
        persisted=False)
    obj = objects.job_template
    assert 'project' in obj.resources_needed_to_start
    validation_errors, resources_needed_to_start = obj.resource_validation_data()
    assert 'project' in validation_errors


def test_inventory_credential_need_to_start(job_template_factory):
    objects = job_template_factory(
        'job-template-few-resources',
        project='project1',
        persisted=False)
    obj = objects.job_template
    assert 'inventory' in obj.resources_needed_to_start
    assert 'credential' in obj.resources_needed_to_start


def test_inventory_credential_contradictions(job_template_factory):
    objects = job_template_factory(
        'job-template-paradox',
        project='project1',
        persisted=False)
    obj = objects.job_template
    obj.ask_inventory_on_launch = False
    obj.ask_credential_on_launch = False
    validation_errors, resources_needed_to_start = obj.resource_validation_data()
    assert 'inventory' in validation_errors
    assert 'credential' in validation_errors


def test_survey_answers_as_string(job_template_factory):
    objects = job_template_factory(
        'job-template-with-survey',
        survey=['var1'],
        persisted=False)
    jt = objects.job_template
    user_extra_vars = json.dumps({'var1': 'asdf'})
    accepted, ignored = jt._accept_or_ignore_job_kwargs(extra_vars=user_extra_vars)
    assert 'var1' in accepted['extra_vars']


@pytest.mark.survey
def test_job_template_survey_password_redaction(job_template_with_survey_passwords_unit):
    """Tests the JobTemplate model's funciton to redact passwords from
    extra_vars - used when creating a new job"""
    assert job_template_with_survey_passwords_unit.survey_password_variables() == ['secret_key', 'SSN']


def test_job_template_survey_variable_validation(job_template_factory):
    objects = job_template_factory(
        'survey_variable_validation',
        organization='org1',
        inventory='inventory1',
        credential='cred1',
        persisted=False,
    )
    obj = objects.job_template
    obj.survey_spec = {
        "description": "",
        "spec": [
            {
                "required": True,
                "min": 0,
                "default": "5",
                "max": 1024,
                "question_description": "",
                "choices": "",
                "variable": "a",
                "question_name": "Whosyourdaddy",
                "type": "text"
            }
        ],
        "name": ""
    }
    obj.survey_enabled = True
    assert obj.survey_variable_validation({"a": 5}) == ["Value 5 for 'a' expected to be a string."]


def test_job_template_survey_mixin(job_template_factory):
    objects = job_template_factory(
        'survey_mixin_test',
        organization='org1',
        inventory='inventory1',
        credential='cred1',
        persisted=False,
    )
    obj = objects.job_template
    obj.survey_enabled = True
    obj.survey_spec = {'spec': [{'default':'my_default', 'type':'password', 'variable':'my_variable'}]}
    kwargs = obj._update_unified_job_kwargs(extra_vars={'my_variable':'$encrypted$'})
    assert kwargs['extra_vars'] == '{"my_variable": "my_default"}'


def test_job_template_survey_mixin_length(job_template_factory):
    objects = job_template_factory(
        'survey_mixin_test',
        organization='org1',
        inventory='inventory1',
        credential='cred1',
        persisted=False,
    )
    obj = objects.job_template
    obj.survey_enabled = True
    obj.survey_spec = {'spec': [{'default':'my_default', 'type':'password', 'variable':'my_variable'},
                                {'type':'password', 'variable':'my_other_variable'}]}
    kwargs = obj._update_unified_job_kwargs(extra_vars={'my_variable':'$encrypted$'})
    assert kwargs['extra_vars'] == '{"my_variable": "my_default"}'


def test_job_template_can_start_with_callback_extra_vars_provided(job_template_factory):
    objects = job_template_factory(
        'callback_extra_vars_test',
        organization='org1',
        inventory='inventory1',
        credential='cred1',
        persisted=False,
    )
    obj = objects.job_template
    obj.ask_variables_on_launch = True
    assert obj.can_start_without_user_input(callback_extra_vars='{"foo": "bar"}') is True