fixed unit tests

business-registry-dashboard/tests/unit/stores/affiliations.test.ts

@@ -145,6 +145,12 @@ mockNuxtImport('useConnectLaunchdarklyStore', () => {
   )
 })
 
+// Add mock for useRoute
+const mockRoute = { params: {} }
+mockNuxtImport('useRoute', () => {
+  return () => mockRoute
+})
+
 describe('useAffiliationsStore', () => {
   let store: any
 
@@ -169,25 +175,48 @@ describe('useAffiliationsStore', () => {
     vi.clearAllMocks()
     vi.resetAllMocks()
     mockAuthenticated = true
+    mockRoute.params = {} // Reset route params after each test
   })
 
   describe('loadAffiliations', () => {
     it('should fetch and set affiliations correctly', async () => {
-      mockGetStoredFlag.mockReturnValue(true) // set LDFlags.AffiliationInvitationRequestAccess = true - allow invitations fetch
+      mockGetStoredFlag.mockReturnValue(true)
       mockAuthApi
-        .mockResolvedValueOnce(mockEntities) // affiliations
-        .mockResolvedValueOnce({ // invitations
+        .mockResolvedValueOnce(mockEntities)
+        .mockResolvedValueOnce({
           affiliationInvitations: []
         })
 
       const affStore = useAffiliationsStore()
 
       await affStore.loadAffiliations()
+      await flushPromises()
+
+      // Should use accountStore.currentAccount.id since not staff
+      expect(mockAuthApi).toHaveBeenCalledWith('/orgs/123/affiliations?new=true')
+      // results mapped to business object
+      expect(affStore.affiliations.results).toEqual(processedAffiliations)
+      expect(affStore.affiliations.count).toEqual(4)
+    })
+
+    it('should use route param orgId when user is staff', async () => {
+      store.currentAccount.accountType = 'STAFF' // Make user staff
+      mockRoute.params.orgId = '456' // Set route param
+
+      mockGetStoredFlag.mockReturnValue(true)
+      mockAuthApi
+        .mockResolvedValueOnce(mockEntities)
+        .mockResolvedValueOnce({
+          affiliationInvitations: []
+        })
 
-      // wait for getAffiliatedEntities to finish
+      const affStore = useAffiliationsStore()
+
+      await affStore.loadAffiliations()
       await flushPromises()
 
-      expect(mockAuthApi).toHaveBeenCalledTimes(2) // once for affiliations and another for invitations
+      // Should use route.params.orgId since user is staff
+      expect(mockAuthApi).toHaveBeenCalledWith('/orgs/456/affiliations?new=true')
       // results mapped to business object
       expect(affStore.affiliations.results).toEqual(processedAffiliations)
       expect(affStore.affiliations.count).toEqual(4)