fix tests

This commit is contained in:
Arthur Hanson 2024-10-22 12:19:48 -07:00
parent 1b11b2613a
commit 25d523b671
3 changed files with 19 additions and 4 deletions

View File

@ -1,4 +1,5 @@
from django.contrib.contenttypes.models import ContentType
from django.core.exceptions import ObjectDoesNotExist
from django.utils.translation import gettext_lazy as _
from circuits.choices import CircuitCommitRateChoices, CircuitTerminationPortSpeedChoices
@ -8,9 +9,11 @@ from dcim.models import Site
from ipam.models import ASN
from netbox.forms import NetBoxModelForm
from tenancy.forms import TenancyForm
from utilities.forms import get_field_value
from utilities.forms.fields import CommentField, ContentTypeChoiceField, DynamicModelChoiceField, DynamicModelMultipleChoiceField, SlugField
from utilities.forms.rendering import FieldSet, InlineFields, TabbedGroups
from utilities.forms.widgets import DatePicker, HTMXSelect, NumberWithOptions
from utilities.templatetags.builtins.filters import bettertitle
__all__ = (
'CircuitForm',

View File

@ -1,5 +1,7 @@
import datetime
import json
from django.contrib.contenttypes.models import ContentType
from django.test import override_settings
from django.urls import reverse
@ -201,16 +203,25 @@ class CircuitTestCase(ViewTestCases.PrimaryObjectViewTestCase):
"terminations": [
{
"term_side": "A",
"site": "Site 1"
"scope_type": "dcim.site",
"scope_id": "1"
},
{
"term_side": "Z",
"site": "Site 1"
"scope_type": "dcim.site",
"scope_id": "1"
}
]
}
]
"""
# Fix up the scope site id
site = Site.objects.first()
data = json.loads(json_data)
data[0]["terminations"][0]["scope_id"] = data[0]["terminations"][1]["scope_id"] = site.id
json_data = json.dumps(data)
initial_count = self._get_queryset().count()
data = {
'data': json_data,
@ -370,7 +381,8 @@ class CircuitTerminationTestCase(ViewTestCases.PrimaryObjectViewTestCase):
cls.form_data = {
'circuit': circuits[2].pk,
'term_side': 'A',
'site': sites[2].pk,
'scope_type': ContentType.objects.get_for_model(Site).pk,
'scope': sites[2].pk,
'description': 'New description',
}

View File

@ -5117,7 +5117,7 @@ class CableTestCase(TestCase, ChangeLoggedFilterSetTests):
provider = Provider.objects.create(name='Provider 1', slug='provider-1')
circuit_type = CircuitType.objects.create(name='Circuit Type 1', slug='circuit-type-1')
circuit = Circuit.objects.create(cid='Circuit 1', provider=provider, type=circuit_type)
circuit_termination = CircuitTermination.objects.create(circuit=circuit, term_side='A', site=sites[0])
circuit_termination = CircuitTermination.objects.create(circuit=circuit, term_side='A', scope=sites[0])
# Cables
cables = (