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):