From c202c1325b621748a373483efe478f5b24b513e3 Mon Sep 17 00:00:00 2001 From: Jeremy Stretch Date: Wed, 15 Jan 2020 10:04:12 -0500 Subject: [PATCH] Add test for VM interface type choices --- netbox/virtualization/tests/test_api.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/netbox/virtualization/tests/test_api.py b/netbox/virtualization/tests/test_api.py index a8a2f4340..ad9063417 100644 --- a/netbox/virtualization/tests/test_api.py +++ b/netbox/virtualization/tests/test_api.py @@ -2,7 +2,7 @@ from django.urls import reverse from netaddr import IPNetwork from rest_framework import status -from dcim.choices import InterfaceModeChoices, InterfaceTypeChoices +from dcim.choices import InterfaceModeChoices from dcim.models import Interface from ipam.models import IPAddress, VLAN from utilities.testing import APITestCase, choices_to_dict @@ -22,6 +22,9 @@ class ChoicesTest(APITestCase): # VirtualMachine self.assertEqual(choices_to_dict(response.data.get('virtual-machine:status')), VirtualMachineStatusChoices.as_dict()) + # Interface + self.assertEqual(choices_to_dict(response.data.get('interface:type')), VMInterfaceTypeChoices.as_dict()) + class ClusterTypeTest(APITestCase):