diff --git a/netbox/dcim/tests/test_api.py b/netbox/dcim/tests/test_api.py index c0ea5eca7..607637d1e 100644 --- a/netbox/dcim/tests/test_api.py +++ b/netbox/dcim/tests/test_api.py @@ -40,7 +40,7 @@ class AppTest(APITestCase): self.assertEqual(choices_to_dict(response.data.get('cable:status')), CableStatusChoices.as_dict()) content_types = ContentType.objects.filter(CABLE_TERMINATION_MODELS) cable_termination_choices = { - "{}.{}".format(ct.app_label, ct.model): ct.name for ct in content_types + "{}.{}".format(ct.app_label, ct.model): str(ct) for ct in content_types } self.assertEqual(choices_to_dict(response.data.get('cable:termination_a_type')), cable_termination_choices) self.assertEqual(choices_to_dict(response.data.get('cable:termination_b_type')), cable_termination_choices) diff --git a/netbox/extras/tests/test_api.py b/netbox/extras/tests/test_api.py index b15553a49..db2861f46 100644 --- a/netbox/extras/tests/test_api.py +++ b/netbox/extras/tests/test_api.py @@ -37,7 +37,7 @@ class AppTest(APITestCase): # Graph content_types = ContentType.objects.filter(GRAPH_MODELS) graph_type_choices = { - "{}.{}".format(ct.app_label, ct.model): ct.name for ct in content_types + "{}.{}".format(ct.app_label, ct.model): str(ct) for ct in content_types } self.assertEqual(choices_to_dict(response.data.get('graph:type')), graph_type_choices) self.assertEqual(choices_to_dict(response.data.get('graph:template_language')), TemplateLanguageChoices.as_dict())