mirror of
https://github.com/ansible/awx.git
synced 2024-11-01 08:21:15 +03:00
fix a few issues in credential type kind validation
- fix a typo from `network` to `net` - properly update OPTIONS for CredentialTypes to reflect allowed `kind` values for POST/PUT/ see: #6959
This commit is contained in:
parent
b79627d57c
commit
09055f9c2f
@ -17,7 +17,7 @@ from rest_framework.relations import RelatedField, ManyRelatedField
|
||||
from rest_framework.request import clone_request
|
||||
|
||||
# Ansible Tower
|
||||
from awx.main.models import InventorySource, NotificationTemplate
|
||||
from awx.main.models import InventorySource, NotificationTemplate, CredentialType
|
||||
|
||||
|
||||
class Metadata(metadata.SimpleMetadata):
|
||||
@ -149,6 +149,16 @@ class Metadata(metadata.SimpleMetadata):
|
||||
if field == 'type' and hasattr(serializer, 'get_type_choices'):
|
||||
meta['choices'] = serializer.get_type_choices()
|
||||
|
||||
# API-created/modified CredentialType kinds are limited to
|
||||
# `cloud` and `network`
|
||||
if method != 'GET' and \
|
||||
hasattr(serializer, 'Meta') and \
|
||||
getattr(serializer.Meta, 'model', None) is CredentialType:
|
||||
actions[method]['kind']['choices'] = filter(
|
||||
lambda choice: choice[0] in ('cloud', 'net'),
|
||||
actions[method]['kind']['choices']
|
||||
)
|
||||
|
||||
# For GET method, remove meta attributes that aren't relevant
|
||||
# when reading a field and remove write-only fields.
|
||||
if method == 'GET':
|
||||
|
@ -1904,9 +1904,9 @@ class CredentialTypeSerializer(BaseSerializer):
|
||||
)
|
||||
ret = super(CredentialTypeSerializer, self).validate(attrs)
|
||||
|
||||
if 'kind' in attrs and attrs['kind'] not in ('cloud', 'network'):
|
||||
if 'kind' in attrs and attrs['kind'] not in ('cloud', 'net'):
|
||||
raise serializers.ValidationError({
|
||||
"kind": _("Must be 'cloud' or 'network', not %s") % attrs['kind']
|
||||
"kind": _("Must be 'cloud' or 'net', not %s") % attrs['kind']
|
||||
})
|
||||
|
||||
fields = attrs.get('inputs', {}).get('fields', [])
|
||||
|
@ -143,6 +143,28 @@ def test_create_managed_by_tower_readonly(get, post, admin):
|
||||
assert response.data['results'][0]['managed_by_tower'] is False
|
||||
|
||||
|
||||
@pytest.mark.django_db
|
||||
@pytest.mark.parametrize('kind', ['cloud', 'net'])
|
||||
def test_create_valid_kind(kind, get, post, admin):
|
||||
response = post(reverse('api:credential_type_list'), {
|
||||
'kind': kind,
|
||||
'name': 'My Custom Type',
|
||||
'inputs': {
|
||||
'fields': [{
|
||||
'id': 'api_token',
|
||||
'label': 'API Token',
|
||||
'type': 'string',
|
||||
'secret': True
|
||||
}]
|
||||
},
|
||||
'injectors': {}
|
||||
}, admin)
|
||||
assert response.status_code == 201
|
||||
|
||||
response = get(reverse('api:credential_type_list'), admin)
|
||||
assert response.data['count'] == 1
|
||||
|
||||
|
||||
@pytest.mark.django_db
|
||||
@pytest.mark.parametrize('kind', ['ssh', 'vault', 'scm', 'insights'])
|
||||
def test_create_invalid_kind(kind, get, post, admin):
|
||||
|
Loading…
Reference in New Issue
Block a user