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

View File

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

View File

@ -5117,7 +5117,7 @@ class CableTestCase(TestCase, ChangeLoggedFilterSetTests):
provider = Provider.objects.create(name='Provider 1', slug='provider-1') provider = Provider.objects.create(name='Provider 1', slug='provider-1')
circuit_type = CircuitType.objects.create(name='Circuit Type 1', slug='circuit-type-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 = 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
cables = ( cables = (