Complete viewset tests for VLANTranslationRule

This commit is contained in:
Brian Tiemann 2024-10-21 22:41:21 -04:00
parent 0d5961bf6e
commit 25c8aca1fb
2 changed files with 11 additions and 21 deletions

View File

@ -554,12 +554,19 @@ class VLANTranslationPolicyBulkEditForm(NetBoxModelBulkEditForm):
class VLANTranslationRuleBulkEditForm(NetBoxModelBulkEditForm):
policy = DynamicModelChoiceField(
label=_('Policy'),
queryset=VLANTranslationPolicy.objects.all(),
selector=True
)
local_vid = forms.IntegerField(required=False)
remote_vid = forms.IntegerField(required=False)
model = VLANTranslationRule
fieldsets = (
FieldSet('policy', 'local_vid', 'remote_vid'),
)
nullable_fields = ('description',)
fields = ('policy', 'local_vid', 'remote_vid')
class ServiceTemplateBulkEditForm(NetBoxModelBulkEditForm):

View File

@ -953,8 +953,6 @@ class VLANTranslationRuleTestCase(ViewTestCases.PrimaryObjectViewTestCase):
)
VLANTranslationRule.objects.bulk_create(vlan_translation_rules)
tags = create_tags('Alpha', 'Bravo', 'Charlie')
cls.form_data = {
'policy': vlan_translation_policies[0].pk,
'local_vid': 300,
@ -962,38 +960,23 @@ class VLANTranslationRuleTestCase(ViewTestCases.PrimaryObjectViewTestCase):
}
cls.csv_data = (
"policy_id,local_vid,remote_vid",
"policy,local_vid,remote_vid",
f"{vlan_translation_policies[0].pk},103,203",
f"{vlan_translation_policies[0].pk},104,204",
f"{vlan_translation_policies[1].pk},105,205",
)
cls.csv_update_data = (
"id,policy_id,local_vid,remote_vid",
"id,policy,local_vid,remote_vid",
f"{vlan_translation_rules[0].pk},{vlan_translation_policies[1].pk},105,205",
f"{vlan_translation_rules[1].pk},{vlan_translation_policies[1].pk},106,206",
f"{vlan_translation_rules[2].pk},{vlan_translation_policies[0].pk},107,207",
)
cls.bulk_edit_data = {
'tags': [t.pk for t in tags],
'policy': vlan_translation_policies[2].pk,
}
# def test_bulk_edit_objects_with_permission(self):
# pass
#
# def test_bulk_edit_objects_with_constrained_permission(self):
# pass
#
# def test_bulk_import_objects_with_permission(self):
# pass
#
# def test_bulk_import_objects_with_constrained_permission(self):
# pass
#
# def test_bulk_update_objects_with_permission(self):
# pass
class ServiceTemplateTestCase(ViewTestCases.PrimaryObjectViewTestCase):
model = ServiceTemplate