Просмотр исходного кода

Add test for VM interface type choices

Jeremy Stretch 6 лет назад
Родитель
Сommit
c202c1325b
1 измененных файлов с 4 добавлено и 1 удалено
  1. 4 1
      netbox/virtualization/tests/test_api.py

+ 4 - 1
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):