Added custom field API tests

2017-04-25 14:53:18 -04:00
parent 10823e1c37
commit 1c86b00b5c
3 changed files with 221 additions and 5 deletions
--- a/netbox/extras/models.py
+++ b/netbox/extras/models.py
@@ -156,10 +156,7 @@ class CustomField(models.Model):
            # Read date as YYYY-MM-DD
            return date(*[int(n) for n in serialized_value.split('-')])
        if self.type == CF_TYPE_SELECT:
-            try:
-                return self.choices.get(pk=int(serialized_value))
-            except CustomFieldChoice.DoesNotExist:
-                return None
+            return self.choices.get(pk=int(serialized_value))
        return serialized_value