mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
Rename Interface to VMInterface
This commit is contained in:
@@ -4,7 +4,7 @@ from rest_framework import status
|
||||
from dcim.choices import InterfaceModeChoices
|
||||
from ipam.models import VLAN
|
||||
from utilities.testing import APITestCase, APIViewTestCases
|
||||
from virtualization.models import Cluster, ClusterGroup, ClusterType, Interface, VirtualMachine
|
||||
from virtualization.models import Cluster, ClusterGroup, ClusterType, VirtualMachine, VMInterface
|
||||
|
||||
|
||||
class AppTest(APITestCase):
|
||||
@@ -203,15 +203,15 @@ class InterfaceTest(APITestCase):
|
||||
clustertype = ClusterType.objects.create(name='Test Cluster Type 1', slug='test-cluster-type-1')
|
||||
cluster = Cluster.objects.create(name='Test Cluster 1', type=clustertype)
|
||||
self.virtualmachine = VirtualMachine.objects.create(cluster=cluster, name='Test VM 1')
|
||||
self.interface1 = Interface.objects.create(
|
||||
self.interface1 = VMInterface.objects.create(
|
||||
virtual_machine=self.virtualmachine,
|
||||
name='Test Interface 1'
|
||||
)
|
||||
self.interface2 = Interface.objects.create(
|
||||
self.interface2 = VMInterface.objects.create(
|
||||
virtual_machine=self.virtualmachine,
|
||||
name='Test Interface 2'
|
||||
)
|
||||
self.interface3 = Interface.objects.create(
|
||||
self.interface3 = VMInterface.objects.create(
|
||||
virtual_machine=self.virtualmachine,
|
||||
name='Test Interface 3'
|
||||
)
|
||||
@@ -254,8 +254,8 @@ class InterfaceTest(APITestCase):
|
||||
|
||||
response = self.client.post(url, data, format='json', **self.header)
|
||||
self.assertHttpStatus(response, status.HTTP_201_CREATED)
|
||||
self.assertEqual(Interface.objects.count(), 4)
|
||||
interface4 = Interface.objects.get(pk=response.data['id'])
|
||||
self.assertEqual(VMInterface.objects.count(), 4)
|
||||
interface4 = VMInterface.objects.get(pk=response.data['id'])
|
||||
self.assertEqual(interface4.virtual_machine_id, data['virtual_machine'])
|
||||
self.assertEqual(interface4.name, data['name'])
|
||||
|
||||
@@ -272,7 +272,7 @@ class InterfaceTest(APITestCase):
|
||||
|
||||
response = self.client.post(url, data, format='json', **self.header)
|
||||
self.assertHttpStatus(response, status.HTTP_201_CREATED)
|
||||
self.assertEqual(Interface.objects.count(), 4)
|
||||
self.assertEqual(VMInterface.objects.count(), 4)
|
||||
self.assertEqual(response.data['virtual_machine']['id'], data['virtual_machine'])
|
||||
self.assertEqual(response.data['name'], data['name'])
|
||||
self.assertEqual(response.data['untagged_vlan']['id'], data['untagged_vlan'])
|
||||
@@ -298,7 +298,7 @@ class InterfaceTest(APITestCase):
|
||||
|
||||
response = self.client.post(url, data, format='json', **self.header)
|
||||
self.assertHttpStatus(response, status.HTTP_201_CREATED)
|
||||
self.assertEqual(Interface.objects.count(), 6)
|
||||
self.assertEqual(VMInterface.objects.count(), 6)
|
||||
self.assertEqual(response.data[0]['name'], data[0]['name'])
|
||||
self.assertEqual(response.data[1]['name'], data[1]['name'])
|
||||
self.assertEqual(response.data[2]['name'], data[2]['name'])
|
||||
@@ -332,7 +332,7 @@ class InterfaceTest(APITestCase):
|
||||
|
||||
response = self.client.post(url, data, format='json', **self.header)
|
||||
self.assertHttpStatus(response, status.HTTP_201_CREATED)
|
||||
self.assertEqual(Interface.objects.count(), 6)
|
||||
self.assertEqual(VMInterface.objects.count(), 6)
|
||||
for i in range(0, 3):
|
||||
self.assertEqual(response.data[i]['name'], data[i]['name'])
|
||||
self.assertEqual([v['id'] for v in response.data[i]['tagged_vlans']], data[i]['tagged_vlans'])
|
||||
@@ -348,8 +348,8 @@ class InterfaceTest(APITestCase):
|
||||
|
||||
response = self.client.put(url, data, format='json', **self.header)
|
||||
self.assertHttpStatus(response, status.HTTP_200_OK)
|
||||
self.assertEqual(Interface.objects.count(), 3)
|
||||
interface1 = Interface.objects.get(pk=response.data['id'])
|
||||
self.assertEqual(VMInterface.objects.count(), 3)
|
||||
interface1 = VMInterface.objects.get(pk=response.data['id'])
|
||||
self.assertEqual(interface1.name, data['name'])
|
||||
|
||||
def test_delete_interface(self):
|
||||
@@ -358,4 +358,4 @@ class InterfaceTest(APITestCase):
|
||||
|
||||
response = self.client.delete(url, **self.header)
|
||||
self.assertHttpStatus(response, status.HTTP_204_NO_CONTENT)
|
||||
self.assertEqual(Interface.objects.count(), 2)
|
||||
self.assertEqual(VMInterface.objects.count(), 2)
|
||||
|
Reference in New Issue
Block a user