7699 fix tests

This commit is contained in:
Arthur Hanson 2024-10-24 09:38:04 -07:00
parent 9108915604
commit c73902c088
2 changed files with 7 additions and 6 deletions

View File

@ -601,11 +601,12 @@ class DeviceTestCase(TestCase):
Site.objects.bulk_create(sites)
clusters = (
Cluster(name='Cluster 1', type=cluster_type, site=sites[0]),
Cluster(name='Cluster 2', type=cluster_type, site=sites[1]),
Cluster(name='Cluster 3', type=cluster_type, site=None),
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),
)
Cluster.objects.bulk_create(clusters)
for cluster in clusters:
cluster.save()
device_type = DeviceType.objects.first()
device_role = DeviceRole.objects.first()

View File

@ -274,7 +274,7 @@ class ConfigContextTest(TestCase):
name="Cluster",
group=cluster_group,
type=cluster_type,
site=site,
scope=site,
)
region_context = ConfigContext.objects.create(
@ -366,7 +366,7 @@ class ConfigContextTest(TestCase):
"""
site = Site.objects.first()
cluster_type = ClusterType.objects.create(name="Cluster Type")
cluster = Cluster.objects.create(name="Cluster", type=cluster_type, site=site)
cluster = Cluster.objects.create(name="Cluster", type=cluster_type, scope=site)
vm_role = DeviceRole.objects.first()
# Create a ConfigContext associated with the site