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