diff --git a/apps/common/serializers.py b/apps/common/serializers.py index 7d13730..5516b0c 100644 --- a/apps/common/serializers.py +++ b/apps/common/serializers.py @@ -57,12 +57,18 @@ class Meta: "cpc", "description", "common_name", + "display_name", "scientific_name", "unit_of_measure", "kcals_per_unit", "aliases", ] + display_name = serializers.SerializerMethodField() + + def get_display_name(self, obj): + return obj.display_name() + class UserSerializer(serializers.ModelSerializer): class Meta: diff --git a/apps/common/tests/test_viewsets.py b/apps/common/tests/test_viewsets.py index 285c3c1..65b9991 100644 --- a/apps/common/tests/test_viewsets.py +++ b/apps/common/tests/test_viewsets.py @@ -181,6 +181,7 @@ def test_filter_by_common_name(self): result = json.loads(response.content.decode("utf-8")) self.assertEqual(len(result), 1) self.assertEqual(result[0]["common_name"], self.product2.common_name) + self.assertEqual(result[0]["display_name"], self.product2.display_name()) def test_search_fields(self): response = self.client.get(self.url, {"search": "Product Description"})