#5377: include country

ustaxcourt · Sep 18, 2024 · d7569e8 · d7569e8
1 parent 23475b9
commit d7569e8
Show file tree

Hide file tree

Showing 2 changed files with 58 additions and 0 deletions.
diff --git a/web-api/src/lambdas/v2/marshallers/marshallContact.test.ts b/web-api/src/lambdas/v2/marshallers/marshallContact.test.ts
@@ -5,13 +5,35 @@ const MOCK_CONTACT = Object.assign({}, getContactPrimary(MOCK_CASE), {
   serviceIndicator: 'Electronic',
 });
 
+const MOCK_CONTACT_INTERNATIONAL = Object.assign({}, MOCK_CONTACT, {
+  country: 'Brazil',
+  serviceIndicator: 'Electronic',
+});
+
 describe('marshallContact', () => {
   it('returns a contact object with the expected properties', () => {
     expect(Object.keys(marshallContact(MOCK_CONTACT)).sort()).toEqual([
       'address1',
       'address2',
       'address3',
       'city',
+      'country',
+      'email',
+      'name',
+      'phone',
+      'postalCode',
+      'serviceIndicator',
+      'state',
+    ]);
+
+    expect(
+      Object.keys(marshallContact(MOCK_CONTACT_INTERNATIONAL)).sort(),
+    ).toEqual([
+      'address1',
+      'address2',
+      'address3',
+      'city',
+      'country',
       'email',
       'name',
       'phone',
@@ -37,13 +59,48 @@ describe('marshallContact', () => {
     expect(mock.postalCode).toBeDefined();
     expect(mock.serviceIndicator).toBeDefined();
     expect(mock.state).toBeDefined();
+    expect(mock.country).toBeUndefined(); // undefined
+
+    const marshalled = marshallContact(mock);
+
+    expect(marshalled.address1).toEqual(mock.address1);
+    expect(marshalled.address2).toEqual(mock.address2);
+    expect(marshalled.address3).toEqual(mock.address3);
+    expect(marshalled.city).toEqual(mock.city);
+    expect(marshalled.email).toEqual(mock.email);
+    expect(marshalled.name).toEqual(mock.name);
+    expect(marshalled.phone).toEqual(mock.phone);
+    expect(marshalled.postalCode).toEqual(mock.postalCode);
+    expect(marshalled.serviceIndicator).toEqual(mock.serviceIndicator);
+    expect(marshalled.state).toEqual(mock.state);
+    expect(marshalled.country).toEqual(mock.country);
+  });
+
+  it('marshalls from the current contact format for users with a country', () => {
+    const mock = Object.assign({}, MOCK_CONTACT_INTERNATIONAL, {
+      address2: 'address2',
+      address3: 'address3',
+    });
+
+    expect(mock.address1).toBeDefined();
+    expect(mock.address2).toBeDefined();
+    expect(mock.address3).toBeDefined();
+    expect(mock.city).toBeDefined();
+    expect(mock.country).toBeDefined();
+    expect(mock.email).toBeDefined();
+    expect(mock.name).toBeDefined();
+    expect(mock.phone).toBeDefined();
+    expect(mock.postalCode).toBeDefined();
+    expect(mock.serviceIndicator).toBeDefined();
+    expect(mock.state).toBeDefined();
 
     const marshalled = marshallContact(mock);
 
     expect(marshalled.address1).toEqual(mock.address1);
     expect(marshalled.address2).toEqual(mock.address2);
     expect(marshalled.address3).toEqual(mock.address3);
     expect(marshalled.city).toEqual(mock.city);
+    expect(marshalled.country).toEqual(mock.country);
     expect(marshalled.email).toEqual(mock.email);
     expect(marshalled.name).toEqual(mock.name);
     expect(marshalled.phone).toEqual(mock.phone);

diff --git a/web-api/src/lambdas/v2/marshallers/marshallContact.ts b/web-api/src/lambdas/v2/marshallers/marshallContact.ts
@@ -11,6 +11,7 @@ export const marshallContact = contactObject => {
     address2: contactObject.address2,
     address3: contactObject.address3,
     city: contactObject.city,
+    country: contactObject.country,
     email: contactObject.email,
     name: contactObject.name,
     phone: contactObject.phone,