mirror/netbox-community-netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00
Code Activity

Fixes #10517: Automatically inherit site assignment from cluster when creating a virtual machine

Browse Source
This commit is contained in:
jeremystretch
2022-10-03 15:35:45 -04:00
parent 7feb86fe55
commit 7712b81ab9
3 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
netbox/virtualization/models.py
View File

@@ -347,10 +347,12 @@ class VirtualMachine(NetBoxModel, ConfigContextModel):
})
# Validate site for cluster & device
if self.cluster and self.cluster.site != self.site:
if self.cluster and self.site and self.cluster.site != self.site:
raise ValidationError({
'cluster': f'The selected cluster ({self.cluster} is not assigned to this site ({self.site}).'
})
elif self.cluster:
self.site = self.cluster.site
if self.device and self.device.site != self.site:
raise ValidationError({
'device': f'The selected device ({self.device} is not assigned to this site ({self.site}).'

7
netbox/virtualization/tests/test_models.py
View File

@@ -68,6 +68,7 @@ class VirtualMachineTestCase(TestCase):
with self.assertRaises(ValidationError):
VirtualMachine(name='vm1', site=sites[0], cluster=clusters[1]).full_clean()
# VM with cluster site but no direct site should fail
with self.assertRaises(ValidationError):
VirtualMachine(name='vm1', site=None, cluster=clusters[0]).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.full_clean()
self.assertEqual(vm.site, sites[0])