1
0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00

Wrote tests for secrets API

This commit is contained in:
Jeremy Stretch
2017-03-17 16:01:57 -04:00
parent 79a9ac3bc8
commit 130ff27f26
2 changed files with 239 additions and 0 deletions

View File

@@ -1,4 +1,5 @@
from rest_framework import serializers
from rest_framework.validators import UniqueTogetherValidator
from dcim.api.serializers import NestedDeviceSerializer
from secrets.models import Secret, SecretRole
@@ -42,3 +43,14 @@ class WritableSecretSerializer(serializers.ModelSerializer):
class Meta:
model = Secret
fields = ['id', 'device', 'role', 'name', 'plaintext']
validators = []
def validate(self, data):
# Validate uniqueness of name if one has been provided.
if data.get('name', None):
validator = UniqueTogetherValidator(queryset=Secret.objects.all(), fields=('device', 'role', 'name'))
validator.set_context(self)
validator(data)
return data