Changed the become_method field into one that takes arbitrary input

related #2630

Signed-off-by: Jeff Bradberry <jeff.bradberry@gmail.com>
This commit is contained in:
Jeff Bradberry
2019-01-28 14:49:59 -05:00
parent 155c214df0
commit 0ecd6542bf
5 changed files with 10 additions and 22 deletions

View File

@@ -1,13 +1,11 @@
# Copyright (c) 2017 Ansible by Red Hat
# All Rights Reserved.
import itertools
import pytest
from django.core.exceptions import ValidationError
from awx.main.utils import decrypt_field
from awx.main.models import Credential, CredentialType, V1Credential
from awx.main.models import Credential, CredentialType
from rest_framework import serializers
@@ -206,10 +204,11 @@ def test_vault_validation(organization, inputs, valid):
@pytest.mark.django_db
@pytest.mark.parametrize('become_method, valid', list(zip(
dict(V1Credential.FIELDS['become_method'].choices).keys(),
itertools.repeat(True)
)) + [('invalid-choice', False)])
@pytest.mark.parametrize('become_method, valid', [
('', True),
('sudo', True),
('custom-plugin', True),
])
def test_choices_validity(become_method, valid, organization):
inputs = {'become_method': become_method}
cred_type = CredentialType.defaults['ssh']()