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

#6732: Add asns relationship to SiteSerializer and extend tests

Browse Source
This commit is contained in:
jeremystretch
2021-11-03 15:15:14 -04:00
parent 7a97d5d4eb
commit 7a55832a22
6 changed files with 40 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
netbox/dcim/tests/test_api.py
View File

@ -6,7 +6,7 @@ from rest_framework import status
from dcim.choices import *
from dcim.constants import *
from dcim.models import *
from ipam.models import VLAN
from ipam.models import ASN, RIR, VLAN
from utilities.testing import APITestCase, APIViewTestCases
from virtualization.models import Cluster, ClusterType
@ -143,6 +143,13 @@ class SiteTest(APIViewTestCases.APIViewTestCase):
)
Site.objects.bulk_create(sites)
rir = RIR.objects.create(name='RFC 6996', is_private=True)
asns = [
ASN(asn=65000 + i, rir=rir) for i in range(8)
]
ASN.objects.bulk_create(asns)
cls.create_data = [
{
'name': 'Site 4',
@ -150,6 +157,7 @@ class SiteTest(APIViewTestCases.APIViewTestCase):
'region': regions[1].pk,
'group': groups[1].pk,
'status': SiteStatusChoices.STATUS_ACTIVE,
'asns': [asns[0].pk, asns[1].pk],
},
{
'name': 'Site 5',
@ -157,6 +165,7 @@ class SiteTest(APIViewTestCases.APIViewTestCase):
'region': regions[1].pk,
'group': groups[1].pk,
'status': SiteStatusChoices.STATUS_ACTIVE,
'asns': [asns[2].pk, asns[3].pk],
},
{
'name': 'Site 6',
@ -164,6 +173,7 @@ class SiteTest(APIViewTestCases.APIViewTestCase):
'region': regions[1].pk,
'group': groups[1].pk,
'status': SiteStatusChoices.STATUS_ACTIVE,
'asns': [asns[4].pk, asns[5].pk],
},
]