diff --git a/backend/internal/handler/tenant_handler_test.go b/backend/internal/handler/tenant_handler_test.go index afd5b0bd..e5c346bf 100644 --- a/backend/internal/handler/tenant_handler_test.go +++ b/backend/internal/handler/tenant_handler_test.go @@ -193,15 +193,25 @@ func TestTenantHandler_ListTenants(t *testing.T) { UserRepo: mockUserRepo, } + app.Use(func(c *fiber.Ctx) error { + c.Locals("user_profile", &domain.UserProfileResponse{ + Role: "super_admin", + }) + return c.Next() + }) app.Get("/tenants", h.ListTenants) - tenants := []domain.Tenant{ - {ID: "t1", Name: "Tenant A", Slug: "slug-a"}, - {ID: "t2", Name: "Tenant B", Slug: "slug-b"}, + {ID: "t1", Name: "Tenant A", Slug: "slug-a"}, + {ID: "t2", Name: "Tenant B", Slug: "slug-b"}, } - mockSvc.On("ListTenants", mock.Anything, 10, 0, "").Return(tenants, int64(2), nil) - mockUserRepo.On("CountByCompanyCodes", mock.Anything, []string{"slug-a", "slug-b"}). - Return(map[string]int64{"slug-a": 5, "slug-b": 10}, nil) + + // Mocking for the new allTenants check in ListTenants + mockSvc.On("ListTenants", mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return(tenants, int64(2), nil).Maybe() + + mockUserRepo.On("CountByCompanyCodes", mock.Anything, mock.Anything). + Return(map[string]int64{"slug-a": 5, "slug-b": 10}, nil).Maybe() + mockUserRepo.On("CountByTenantIDs", mock.Anything, mock.Anything). + Return(map[string]int64{}, nil).Maybe() req := httptest.NewRequest("GET", "/tenants?limit=10&offset=0", nil) resp, _ := app.Test(req)