wip

Author: ottocho

packages/ui-extensions-react/src/surfaces/customer-account/hooks/tests/authenticated-account.test.tsx

@@ -9,20 +9,17 @@ import {mount, createMockStatefulRemoteSubscribable} from './mount';
 
 function createMockCustomer() {
   return {
-    id: `gid://shopify/Customer/${faker.datatype.number({
-      min: 1,
-      precision: 1,
-    })}`,
+    id: `gid://shopify/Customer/${faker.number.int({min: 1})}`,
   };
 }
 
 function createMockPurchasingCompany() {
   return {
     company: {
-      id: `gid://shopify/Company/${faker.datatype.number({
-        min: 1,
-        precision: 1,
-      })}`,
+      id: `gid://shopify/Company/${faker.number.int({min: 1})}`,
+    },
+    location: {
+      id: `gid://shopify/CompanyLocation/${faker.number.int({min: 1})}`,
     },
   };
 }
@@ -60,7 +57,9 @@ describe('account Hooks', () => {
       );
       expect(purchasingCompany).toBeDefined();
       expect(purchasingCompany.company.id).toBeDefined();
+      expect(purchasingCompany.location.id).toBeDefined();
       expect(value.company.id).toBe(purchasingCompany.company.id);
+      expect(value.location.id).toBe(purchasingCompany.location.id);
     });
   });
 });

packages/ui-extensions/src/surfaces/customer-account/api/shared.ts

@@ -369,6 +369,11 @@ export interface PurchasingCompany {
    * Include information of the company of the logged in business customer.
    */
   company: Company;
+
+  /**
+   * Include information of the company location of the logged in business customer.
+   */
+  location?: CompanyLocation;
 }
 
 export interface Company {
@@ -378,6 +383,13 @@ export interface Company {
   id: string;
 }
 
+export interface CompanyLocation {
+  /**
+   * Company location ID.
+   */
+  id: string;
+}
+
 export interface Ui {
   /**
    * An overlay is a contextual element on top of the main interface that provides additional information or functionality.