IGUK-433 Add new EYB user data fields to activity stream serialiser

uktrade · Oct 2, 2024 · 93e7a03 · 93e7a03
1 parent dec6f6b
commit 93e7a03
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/activitystream/serializers.py b/activitystream/serializers.py
@@ -310,6 +310,12 @@ class ActivityStreamExpandYourBusinessUserDataSerializer(serializers.ModelSerial
     agreeInfoEmail = serializers.BooleanField(source='agree_info_email')  # noqa: N815
     landingTimeframe = serializers.CharField(source='landing_timeframe')  # noqa: N815
     companyWebsite = serializers.CharField(source='company_website')  # noqa: N815
+    dunsNumber = serializers.CharField(source='duns_number')  # noqa: N815
+    addressLine1 = serializers.CharField(source='address_line_1')  # noqa: N815
+    addressLine2 = serializers.CharField(source='address_line_2')  # noqa: N815
+    town = serializers.CharField(source='town')  # noqa: N815
+    county = serializers.CharField(source='county')  # noqa: N815
+    postcode = serializers.CharField(source='postcode')  # noqa: N815
 
     class Meta:
         model = UserData
@@ -328,6 +334,12 @@ class Meta:
             'created',
             'modified',
             'companyWebsite',
+            'dunsNumber',
+            'addressLine1',
+            'addressLine2',
+            'town',
+            'county',
+            'postcode',
         ]
 
     def to_representation(self, instance):

diff --git a/tests/unit/activitystream/test_serializers.py b/tests/unit/activitystream/test_serializers.py
@@ -377,6 +377,13 @@ def test_eyb_user_serializer():
     instance.modified = '2023-08-24 10:48:19.018536+00'
     instance.company_website = 'https://www.great.gov.uk'
 
+    instance.duns_number = 'DUNDUHDUNDUN'
+    instance.address_line_1 = '1 High street'
+    instance.address_line_2 = 'Electric avenue'
+    instance.town = 'Swansea'
+    instance.county = 'Glamorgan'
+    instance.postcode = 'SA4 4PP'
+
     serializer = ActivityStreamExpandYourBusinessUserDataSerializer()
 
     output = serializer.to_representation(instance)
@@ -399,6 +406,12 @@ def test_eyb_user_serializer():
             'created': instance.created,
             'modified': instance.modified,
             'companyWebsite': instance.company_website,
+            'dunsNumber': instance.duns_number,
+            'addressLine1': instance.address_line_1,
+            'addressLine2': instance.address_line_2,
+            'town': instance.town,
+            'county': instance.county,
+            'postcode': instance.postcode,
         },
     }
     assert output == expected