7699 fix ipam tests

This commit is contained in:
Arthur Hanson 2024-10-24 09:23:18 -07:00
parent 65295f6d79
commit 9108915604
2 changed files with 6 additions and 5 deletions

View File

@ -148,7 +148,7 @@ class VLANQuerySet(RestrictedQuerySet):
# Find all relevant VLANGroups
q = Q()
site = vm.site or vm.cluster.site
site = vm.site or vm.cluster._site
if vm.cluster:
# Add VLANGroups scoped to the assigned cluster (or its group)
q |= Q(

View File

@ -1674,11 +1674,12 @@ class VLANTestCase(TestCase, ChangeLoggedFilterSetTests):
cluster_type = ClusterType.objects.create(name='Cluster Type 1', slug='cluster-type-1')
clusters = (
Cluster(name='Cluster 1', type=cluster_type, group=cluster_groups[0], site=sites[0]),
Cluster(name='Cluster 2', type=cluster_type, group=cluster_groups[1], site=sites[1]),
Cluster(name='Cluster 3', type=cluster_type, group=cluster_groups[2], site=sites[2]),
Cluster(name='Cluster 1', type=cluster_type, group=cluster_groups[0], scope=sites[0]),
Cluster(name='Cluster 2', type=cluster_type, group=cluster_groups[1], scope=sites[1]),
Cluster(name='Cluster 3', type=cluster_type, group=cluster_groups[2], scope=sites[2]),
)
Cluster.objects.bulk_create(clusters)
for cluster in clusters:
cluster.save()
virtual_machines = (
VirtualMachine(name='Virtual Machine 1', cluster=clusters[0]),