mirror of
https://github.com/netbox-community/netbox.git
synced 2025-07-19 09:53:34 -06:00
6dc75d8db1
* 7699 Add Scope to Cluster * 7699 Serializer * 7699 filterset * 7699 bulk_edit * 7699 bulk_import * 7699 model_form * 7699 graphql, tables * 7699 fixes * 7699 fixes * 7699 fixes * 7699 fixes * 7699 fix tests * 7699 fix graphql tests for clusters reference * 7699 fix dcim tests * 7699 fix ipam tests * 7699 fix tests * 7699 use mixin for model * 7699 change mixin name * 7699 scope form * 7699 scope form * 7699 scoped form, fitlerset * 7699 review changes * 7699 move ScopedFilterset * 7699 move CachedScopeMixin * 7699 review changes * 7699 review changes * 7699 refactor mixins * 7699 _sitegroup -> _site_group * 7699 update docstring * Misc cleanup * Update migrations --------- Co-authored-by: Jeremy Stretch <jstretch@netboxlabs.com>
122 lines
3.9 KiB
Python
122 lines
3.9 KiB
Python
from django.core.exceptions import ValidationError
|
|
from django.test import TestCase
|
|
|
|
from dcim.models import Site
|
|
from virtualization.models import *
|
|
from tenancy.models import Tenant
|
|
|
|
|
|
class VirtualMachineTestCase(TestCase):
|
|
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
cluster_type = ClusterType.objects.create(name='Cluster Type 1', slug='cluster-type-1')
|
|
Cluster.objects.create(name='Cluster 1', type=cluster_type)
|
|
|
|
def test_vm_duplicate_name_per_cluster(self):
|
|
vm1 = VirtualMachine(
|
|
cluster=Cluster.objects.first(),
|
|
name='Test VM 1'
|
|
)
|
|
vm1.save()
|
|
|
|
vm2 = VirtualMachine(
|
|
cluster=vm1.cluster,
|
|
name=vm1.name
|
|
)
|
|
|
|
# Two VMs assigned to the same Cluster and no Tenant should fail validation
|
|
with self.assertRaises(ValidationError):
|
|
vm2.full_clean()
|
|
|
|
tenant = Tenant.objects.create(name='Test Tenant 1', slug='test-tenant-1')
|
|
vm1.tenant = tenant
|
|
vm1.save()
|
|
vm2.tenant = tenant
|
|
|
|
# Two VMs assigned to the same Cluster and the same Tenant should fail validation
|
|
with self.assertRaises(ValidationError):
|
|
vm2.full_clean()
|
|
|
|
vm2.tenant = None
|
|
|
|
# Two VMs assigned to the same Cluster and different Tenants should pass validation
|
|
vm2.full_clean()
|
|
vm2.save()
|
|
|
|
def test_vm_mismatched_site_cluster(self):
|
|
cluster_type = ClusterType.objects.first()
|
|
|
|
sites = (
|
|
Site(name='Site 1', slug='site-1'),
|
|
Site(name='Site 2', slug='site-2'),
|
|
)
|
|
Site.objects.bulk_create(sites)
|
|
|
|
clusters = (
|
|
Cluster(name='Cluster 1', type=cluster_type, scope=sites[0]),
|
|
Cluster(name='Cluster 2', type=cluster_type, scope=sites[1]),
|
|
Cluster(name='Cluster 3', type=cluster_type, scope=None),
|
|
)
|
|
for cluster in clusters:
|
|
cluster.save()
|
|
|
|
# VM with site only should pass
|
|
VirtualMachine(name='vm1', site=sites[0]).full_clean()
|
|
|
|
# VM with site, cluster non-site should pass
|
|
VirtualMachine(name='vm1', site=sites[0], cluster=clusters[2]).full_clean()
|
|
|
|
# VM with non-site cluster only should pass
|
|
VirtualMachine(name='vm1', cluster=clusters[2]).full_clean()
|
|
|
|
# VM with mismatched site & cluster should fail
|
|
with self.assertRaises(ValidationError):
|
|
VirtualMachine(name='vm1', site=sites[0], cluster=clusters[1]).full_clean()
|
|
|
|
# VM with cluster site but no direct site should have its site set automatically
|
|
vm = VirtualMachine(name='vm1', site=None, cluster=clusters[0])
|
|
vm.save()
|
|
self.assertEqual(vm.site, sites[0])
|
|
|
|
def test_vm_name_case_sensitivity(self):
|
|
vm1 = VirtualMachine(
|
|
cluster=Cluster.objects.first(),
|
|
name='virtual machine 1'
|
|
)
|
|
vm1.save()
|
|
|
|
vm2 = VirtualMachine(
|
|
cluster=vm1.cluster,
|
|
name='VIRTUAL MACHINE 1'
|
|
)
|
|
|
|
# Uniqueness validation for name should ignore case
|
|
with self.assertRaises(ValidationError):
|
|
vm2.full_clean()
|
|
|
|
def test_disk_size(self):
|
|
vm = VirtualMachine(
|
|
cluster=Cluster.objects.first(),
|
|
name='Virtual Machine 1'
|
|
)
|
|
vm.save()
|
|
vm.refresh_from_db()
|
|
self.assertEqual(vm.disk, None)
|
|
|
|
# Create two VirtualDisks
|
|
VirtualDisk.objects.create(virtual_machine=vm, name='Virtual Disk 1', size=10)
|
|
VirtualDisk.objects.create(virtual_machine=vm, name='Virtual Disk 2', size=10)
|
|
vm.refresh_from_db()
|
|
self.assertEqual(vm.disk, 20)
|
|
|
|
# Delete one VirtualDisk
|
|
VirtualDisk.objects.first().delete()
|
|
vm.refresh_from_db()
|
|
self.assertEqual(vm.disk, 10)
|
|
|
|
# Attempt to manually overwrite the aggregate disk size
|
|
vm.disk = 30
|
|
with self.assertRaises(ValidationError):
|
|
vm.full_clean()
|