NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-13 02:58:17 -06:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

add missing site

Browse Source
This commit is contained in:
Alex Gittings 2024-04-24 11:23:51 +01:00
parent ea3550ff07
commit da037d6dda
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
netbox/netbox/tests/test_staging.py
View File

@ -1,7 +1,7 @@
from django.test import TransactionTestCase
from circuits.models import Provider, Circuit, CircuitType
from dcim.models import Location
from dcim.models import Location, Site
from extras.choices import ChangeActionChoices
from extras.models import Branch, StagedChange, Tag
from ipam.models import ASN, RIR
@ -14,6 +14,8 @@ class StagingTestCase(TransactionTestCase):
def setUp(self):
create_tags('Alpha', 'Bravo', 'Charlie')
Site.objects.create(name="Site 1", slug="site-1")
rir = RIR.objects.create(name='RIR 1', slug='rir-1')
asns = (
ASN(asn=65001, rir=rir),
@ -47,6 +49,7 @@ class StagingTestCase(TransactionTestCase):
branch = Branch.objects.create(name='Branch 1')
tags = Tag.objects.all()
asns = ASN.objects.all()
site = Site.objects.first()
with checkout(branch):
provider = Provider.objects.create(name='Provider D', slug='provider-d')
@ -55,7 +58,7 @@ class StagingTestCase(TransactionTestCase):
circuit.tags.set(tags)
# Test MPTT Model
location = Location.objects.create(name='Location 1', slug='location-1')
location = Location.objects.create(name='Location 1', slug='location-1', site=site)
# Sanity-checking
self.assertEqual(Provider.objects.count(), 4)