NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-07-22 12:06:53 -06:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

Standardize VMInterfaceTest

Browse Source
This commit is contained in:
Jeremy Stretch 2020-06-29 12:06:36 -04:00
parent 89ff59d048
commit 617e20af0b
1 changed files with 56 additions and 151 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

207
netbox/virtualization/tests/test_api.py
View File

@ -1,7 +1,9 @@
from django.contrib.contenttypes.models import ContentType
from django.urls import reverse from django.urls import reverse
from rest_framework import status from rest_framework import status
from dcim.choices import InterfaceModeChoices from dcim.choices import InterfaceModeChoices
from extras.models import Graph
from ipam.models import VLAN from ipam.models import VLAN
from utilities.testing import APITestCase, APIViewTestCases from utilities.testing import APITestCase, APIViewTestCases
from virtualization.models import Cluster, ClusterGroup, ClusterType, VirtualMachine, VMInterface from virtualization.models import Cluster, ClusterGroup, ClusterType, VirtualMachine, VMInterface
@ -161,7 +163,7 @@ class VirtualMachineTest(APIViewTestCases.APIViewTestCase):
""" """
Check that config context data is included by default in the virtual machines list. Check that config context data is included by default in the virtual machines list.
""" """
virtualmachine = VirtualMachine.objects.first() virtualmachine = VirtualMachine.objects.unrestricted().first()
url = '{}?id={}'.format(reverse('virtualization-api:virtualmachine-list'), virtualmachine.pk) url = '{}?id={}'.format(reverse('virtualization-api:virtualmachine-list'), virtualmachine.pk)
self.add_permissions('virtualization.view_virtualmachine') self.add_permissions('virtualization.view_virtualmachine')
@ -184,7 +186,7 @@ class VirtualMachineTest(APIViewTestCases.APIViewTestCase):
""" """
data = { data = {
'name': 'Virtual Machine 1', 'name': 'Virtual Machine 1',
'cluster': Cluster.objects.first().pk, 'cluster': Cluster.objects.unrestricted().first().pk,
} }
url = reverse('virtualization-api:virtualmachine-list') url = reverse('virtualization-api:virtualmachine-list')
self.add_permissions('virtualization.add_virtualmachine') self.add_permissions('virtualization.add_virtualmachine')
@ -193,169 +195,72 @@ class VirtualMachineTest(APIViewTestCases.APIViewTestCase):
self.assertHttpStatus(response, status.HTTP_400_BAD_REQUEST) self.assertHttpStatus(response, status.HTTP_400_BAD_REQUEST)
# TODO: Standardize InterfaceTest (pending #4721) class VMInterfaceTest(APIViewTestCases.APIViewTestCase):
class VMInterfaceTest(APITestCase): model = VMInterface
brief_fields = ['id', 'name', 'url', 'virtual_machine']
def setUp(self): @classmethod
def setUpTestData(cls):
super().setUp()
clustertype = ClusterType.objects.create(name='Test Cluster Type 1', slug='test-cluster-type-1') clustertype = ClusterType.objects.create(name='Test Cluster Type 1', slug='test-cluster-type-1')
cluster = Cluster.objects.create(name='Test Cluster 1', type=clustertype) cluster = Cluster.objects.create(name='Test Cluster 1', type=clustertype)
self.virtualmachine = VirtualMachine.objects.create(cluster=cluster, name='Test VM 1') virtualmachine = VirtualMachine.objects.create(cluster=cluster, name='Test VM 1')
self.interface1 = VMInterface.objects.create(
virtual_machine=self.virtualmachine, interfaces = (
name='Test Interface 1' VMInterface(virtual_machine=virtualmachine, name='Interface 1'),
VMInterface(virtual_machine=virtualmachine, name='Interface 2'),
VMInterface(virtual_machine=virtualmachine, name='Interface 3'),
) )
self.interface2 = VMInterface.objects.create( VMInterface.objects.bulk_create(interfaces)
virtual_machine=self.virtualmachine,
name='Test Interface 2' vlans = (
) VLAN(name='VLAN 1', vid=1),
self.interface3 = VMInterface.objects.create( VLAN(name='VLAN 2', vid=2),
virtual_machine=self.virtualmachine, VLAN(name='VLAN 3', vid=3),
name='Test Interface 3'
) )
VLAN.objects.bulk_create(vlans)
self.vlan1 = VLAN.objects.create(name="Test VLAN 1", vid=1) cls.create_data = [
self.vlan2 = VLAN.objects.create(name="Test VLAN 2", vid=2) {
self.vlan3 = VLAN.objects.create(name="Test VLAN 3", vid=3) 'virtual_machine': virtualmachine.pk,
'name': 'Interface 4',
def test_get_interface(self):
url = reverse('virtualization-api:vminterface-detail', kwargs={'pk': self.interface1.pk})
self.add_permissions('virtualization.view_vminterface')
response = self.client.get(url, **self.header)
self.assertEqual(response.data['name'], self.interface1.name)
def test_list_interfaces(self):
url = reverse('virtualization-api:vminterface-list')
self.add_permissions('virtualization.view_vminterface')
response = self.client.get(url, **self.header)
self.assertEqual(response.data['count'], 3)
def test_list_interfaces_brief(self):
url = reverse('virtualization-api:vminterface-list')
self.add_permissions('virtualization.view_vminterface')
response = self.client.get('{}?brief=1'.format(url), **self.header)
self.assertEqual(
sorted(response.data['results'][0]),
['id', 'name', 'url', 'virtual_machine']
)
def test_create_interface(self):
data = {
'virtual_machine': self.virtualmachine.pk,
'name': 'Test Interface 4',
}
url = reverse('virtualization-api:vminterface-list')
self.add_permissions('virtualization.add_vminterface')
response = self.client.post(url, data, format='json', **self.header)
self.assertHttpStatus(response, status.HTTP_201_CREATED)
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'])
def test_create_interface_with_802_1q(self):
data = {
'virtual_machine': self.virtualmachine.pk,
'name': 'Test Interface 4',
'mode': InterfaceModeChoices.MODE_TAGGED, 'mode': InterfaceModeChoices.MODE_TAGGED,
'untagged_vlan': self.vlan3.id, 'tagged_vlans': [vlans[0].pk, vlans[1].pk],
'tagged_vlans': [self.vlan1.id, self.vlan2.id], 'untagged_vlan': vlans[2].pk,
}
url = reverse('virtualization-api:vminterface-list')
self.add_permissions('virtualization.add_vminterface')
response = self.client.post(url, data, format='json', **self.header)
self.assertHttpStatus(response, status.HTTP_201_CREATED)
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'])
self.assertEqual([v['id'] for v in response.data['tagged_vlans']], data['tagged_vlans'])
def test_create_interface_bulk(self):
data = [
{
'virtual_machine': self.virtualmachine.pk,
'name': 'Test Interface 4',
}, },
{ {
'virtual_machine': self.virtualmachine.pk, 'virtual_machine': virtualmachine.pk,
'name': 'Test Interface 5', 'name': 'Interface 5',
'mode': InterfaceModeChoices.MODE_TAGGED,
'tagged_vlans': [vlans[0].pk, vlans[1].pk],
'untagged_vlan': vlans[2].pk,
}, },
{ {
'virtual_machine': self.virtualmachine.pk, 'virtual_machine': virtualmachine.pk,
'name': 'Test Interface 6', 'name': 'Interface 6',
'mode': InterfaceModeChoices.MODE_TAGGED,
'tagged_vlans': [vlans[0].pk, vlans[1].pk],
'untagged_vlan': vlans[2].pk,
}, },
] ]
url = reverse('virtualization-api:vminterface-list')
self.add_permissions('virtualization.add_vminterface')
response = self.client.post(url, data, format='json', **self.header) def test_get_interface_graphs(self):
self.assertHttpStatus(response, status.HTTP_201_CREATED) """
self.assertEqual(VMInterface.objects.count(), 6) Test retrieval of Graphs assigned to VM interfaces.
self.assertEqual(response.data[0]['name'], data[0]['name']) """
self.assertEqual(response.data[1]['name'], data[1]['name']) ct = ContentType.objects.get_for_model(VMInterface)
self.assertEqual(response.data[2]['name'], data[2]['name']) graphs = (
Graph(type=ct, name='Graph 1', source='http://example.com/graphs.py?interface={{ obj.name }}&foo=1'),
Graph(type=ct, name='Graph 2', source='http://example.com/graphs.py?interface={{ obj.name }}&foo=2'),
Graph(type=ct, name='Graph 3', source='http://example.com/graphs.py?interface={{ obj.name }}&foo=3'),
)
Graph.objects.bulk_create(graphs)
def test_create_interface_802_1q_bulk(self): self.add_permissions('virtualization.view_vminterface')
data = [ url = reverse('virtualization-api:vminterface-graphs', kwargs={
{ 'pk': VMInterface.objects.unrestricted().first().pk
'virtual_machine': self.virtualmachine.pk, })
'name': 'Test Interface 4', response = self.client.get(url, **self.header)
'mode': InterfaceModeChoices.MODE_TAGGED,
'untagged_vlan': self.vlan2.id,
'tagged_vlans': [self.vlan1.id],
},
{
'virtual_machine': self.virtualmachine.pk,
'name': 'Test Interface 5',
'mode': InterfaceModeChoices.MODE_TAGGED,
'untagged_vlan': self.vlan2.id,
'tagged_vlans': [self.vlan1.id],
},
{
'virtual_machine': self.virtualmachine.pk,
'name': 'Test Interface 6',
'mode': InterfaceModeChoices.MODE_TAGGED,
'untagged_vlan': self.vlan2.id,
'tagged_vlans': [self.vlan1.id],
},
]
url = reverse('virtualization-api:vminterface-list')
self.add_permissions('virtualization.add_vminterface')
response = self.client.post(url, data, format='json', **self.header) self.assertEqual(len(response.data), 3)
self.assertHttpStatus(response, status.HTTP_201_CREATED) self.assertEqual(response.data[0]['embed_url'], 'http://example.com/graphs.py?interface=Interface 1&foo=1')
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'])
self.assertEqual(response.data[i]['untagged_vlan']['id'], data[i]['untagged_vlan'])
def test_update_interface(self):
data = {
'virtual_machine': self.virtualmachine.pk,
'name': 'Test Interface X',
}
url = reverse('virtualization-api:vminterface-detail', kwargs={'pk': self.interface1.pk})
self.add_permissions('virtualization.change_vminterface')
response = self.client.put(url, data, format='json', **self.header)
self.assertHttpStatus(response, status.HTTP_200_OK)
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):
url = reverse('virtualization-api:vminterface-detail', kwargs={'pk': self.interface1.pk})
self.add_permissions('virtualization.delete_vminterface')
response = self.client.delete(url, **self.header)
self.assertHttpStatus(response, status.HTTP_204_NO_CONTENT)
self.assertEqual(VMInterface.objects.count(), 2)