@itwin/reports-config-widget-react 0.2.1 → 0.4.0

package/src/test/ReportMappings.test.tsx

@@ -3,53 +3,42 @@
 * See LICENSE.md in the project root for license terms and full copyright notice.
 *--------------------------------------------------------------------------------------------*/
 import React from "react";
-import faker from "@faker-js/faker";
 import "@testing-library/jest-dom";
-import type {
-  IModelConnection,
-  SelectionSet,
-  SelectionSetEvent,
-} from "@itwin/core-frontend";
-import { NoRenderApp } from "@itwin/core-frontend";
 import { ReportsConfigWidget } from "../ReportsConfigWidget";
 import {
   act,
+  mockITwinId,
   render,
   screen,
-  TestUtils,
+  waitFor,
   waitForElementToBeRemoved,
   within,
 } from "./test-utils";
 import * as moq from "typemoq";
 import type {
+  MappingsClient,
   MappingSingle,
   Report,
   ReportMappingCollection,
 } from "@itwin/insights-client";
-import type { ReportMappingAndMapping } from "../widget/components/ReportMappings";
+import {
+  ExtractionClient,
+  ReportsClient,
+} from "@itwin/insights-client";
 import { ReportMappings } from "../widget/components/ReportMappings";
-import type { GetSingleIModelParams, IModelOperations, OperationOptions } from "@itwin/imodels-client-management";
+import type { GetSingleIModelParams, IModelOperations, IModelsClient, OperationOptions } from "@itwin/imodels-client-management";
 import { IModelState } from "@itwin/imodels-client-management";
-import type {
-  SelectionManager,
-  SelectionScopesManager,
-} from "@itwin/presentation-frontend";
-import {
-  Presentation,
-  SelectionChangeEvent,
-} from "@itwin/presentation-frontend";
-import type { BeEvent } from "@itwin/core-bentley";
-import type BulkExtractor from "../widget/components/BulkExtractor";
+import { BulkExtractor } from "../widget/components/BulkExtractor";
 import type { ReportMappingHorizontalTileProps } from "../widget/components/ReportMappingHorizontalTile";
 import { Text } from "@itwin/itwinui-react";
 import { EmptyLocalization } from "@itwin/core-common";
+import type { AddMappingsModalProps } from "../widget/components/AddMappingsModal";
 
-const mockITwinId = faker.datatype.uuid();
 // Lets work with two iModels for now.
-const mockIModelId1 = faker.datatype.uuid();
-const mockIModelId2 = faker.datatype.uuid();
+const mockIModelId1 = "mockIModelId1";
+const mockIModelId2 = "mockIModelId2";
 
-const mockReportId = faker.datatype.uuid();
+const mockReportId = "mockReportId";
 
 const mockIModelsResponse = [
   {
@@ -131,7 +120,7 @@ const mockReportMappingsFactory = (): ReportMappingCollection => {
     mappings: [
       {
         reportId: mockReportId,
-        mappingId: faker.datatype.uuid(),
+        mappingId: "mockMappingId1",
         imodelId: mockIModelId1,
         _links: {
           report: {
@@ -147,7 +136,7 @@ const mockReportMappingsFactory = (): ReportMappingCollection => {
       },
       {
         reportId: mockReportId,
-        mappingId: faker.datatype.uuid(),
+        mappingId: "mockMappingId2",
         imodelId: mockIModelId2,
         _links: {
           report: {
@@ -198,19 +187,6 @@ const mockMappingsFactory = (
   return mockMappings;
 };
 
-const connectionMock = moq.Mock.ofType<IModelConnection>();
-const mockBulkExtractor = moq.Mock.ofType<BulkExtractor>();
-const selectionManagerMock = moq.Mock.ofType<SelectionManager>();
-const selectionScopesManagerMock = moq.Mock.ofType<SelectionScopesManager>();
-const mockIModelsClient = moq.Mock.ofType<IModelOperations<OperationOptions>>();
-
-interface AddMappingsModalProps {
-  reportId: string;
-  existingMappings: ReportMappingAndMapping[];
-  show: boolean;
-  returnFn: () => Promise<void>;
-}
-
 jest.mock("../widget/components/Constants.ts", () => ({
   STATUS_CHECK_INTERVAL: 10,
 }));
@@ -224,25 +200,18 @@ jest.mock("../widget/components/ReportMappingHorizontalTile", () => ({
   },
 }));
 
-jest.mock("@itwin/appui-react", () => ({
-  ...jest.requireActual("@itwin/appui-react"),
-  useActiveIModelConnection: () => connectionMock.object,
-}));
-
-let returnFn: () => Promise<void>;
+let onClose: () => Promise<void>;
 jest.mock("../widget/components/AddMappingsModal", () => ({
   ...jest.requireActual("../widget/components/AddMappingsModal"),
   AddMappingsModal: (props: AddMappingsModalProps) => {
-    returnFn = props.returnFn;
-    return <div data-testid="add-mappings-modal"/>;
+    onClose = props.onClose;
+    return <div data-testid="add-mappings-modal" />;
   },
 }));
 
 jest.mock("@itwin/imodels-client-management", () => ({
   ...jest.requireActual("@itwin/imodels-client-management"),
-  IModelsClient: jest.fn().mockImplementation(() => ({
-    iModels: mockIModelsClient.object,
-  })),
+
   toArray: jest.fn().mockImplementation(async () => {
     return mockProjectIModels.iModels;
   }),
@@ -251,60 +220,30 @@ jest.mock("@itwin/imodels-client-management", () => ({
 const mockGetMapping = jest.fn();
 const mockGetMappings = jest.fn();
 const mockGetReportMappings = jest.fn();
-const mockDeleteReportMapping = jest.fn();
 
-jest.mock("@itwin/insights-client", () => ({
-  ...jest.requireActual("@itwin/insights-client"),
-  MappingsClient: jest.fn().mockImplementation(() => ({
-    getMapping: mockGetMapping,
-    getMappings: mockGetMappings,
-  })),
-  ReportsClient: jest.fn().mockImplementation(() => ({
-    getReportMappings: mockGetReportMappings,
-    deleteReportMapping: mockDeleteReportMapping,
-  })),
-}));
+const mockIModelsClient = moq.Mock.ofType<IModelsClient>();
+const mockIModelsClientOperations = moq.Mock.ofType<IModelOperations<OperationOptions>>();
+const mockReportsClient = moq.Mock.ofType<ReportsClient>();
+const mockMappingsClient = moq.Mock.ofType<MappingsClient>();
 
 beforeAll(async () => {
-  await NoRenderApp.startup({localization: new EmptyLocalization()});
-  await Presentation.initialize();
-  const selectionSet = moq.Mock.ofType<SelectionSet>();
-  const onChanged = moq.Mock.ofType<BeEvent<(ev: SelectionSetEvent) => void>>();
-  selectionSet.setup((x) => x.elements).returns(() => new Set([]));
-  selectionSet.setup((x) => x.onChanged).returns(() => onChanged.object);
-  connectionMock
-    .setup((x) => x.selectionSet)
-    .returns(() => selectionSet.object);
-  connectionMock.setup((x) => x.iModelId).returns(() => mockIModelId1);
-  connectionMock.setup((x) => x.iTwinId).returns(() => mockITwinId);
-
-  selectionManagerMock
-    .setup((x) => x.selectionChange)
-    .returns(() => new SelectionChangeEvent());
-
-  selectionScopesManagerMock
-    .setup(async (x) => x.getSelectionScopes(connectionMock.object))
-    .returns(async () => []);
-  selectionManagerMock
-    .setup((x) => x.scopes)
-    .returns(() => selectionScopesManagerMock.object);
-
-  Presentation.setSelectionManager(selectionManagerMock.object);
-  await TestUtils.initializeUiFramework(connectionMock.object);
-  await ReportsConfigWidget.initialize();
-});
-
-afterAll(() => {
-  TestUtils.terminateUiFramework();
+  const localization = new EmptyLocalization();
+  await ReportsConfigWidget.initialize(localization);
+  mockIModelsClientOperations.setup(async (x) => x.getSingle(moq.It.isObjectWith<GetSingleIModelParams>({ iModelId: mockIModelId1 })))
+    .returns(async () => mockIModelsResponse[0].iModel);
+  mockIModelsClientOperations.setup(async (x) => x.getSingle(moq.It.isObjectWith<GetSingleIModelParams>({ iModelId: mockIModelId2 })))
+    .returns(async () => mockIModelsResponse[1].iModel);
+  mockIModelsClient.setup((x) => x.iModels)
+    .returns(() => mockIModelsClientOperations.object);
+  mockReportsClient.setup(async (x) => x.getReportMappings(moq.It.isAny(), moq.It.isAny())).returns(mockGetReportMappings);
+  mockMappingsClient.setup(async (x) => x.getMappings(moq.It.isAny(), moq.It.isAny())).returns(mockGetMappings);
+  mockMappingsClient.setup(async (x) => x.getMapping(moq.It.isAny(), moq.It.isAny(), moq.It.isAny())).returns(mockGetMapping);
 });
 
 afterEach(() => {
   mockGetMapping.mockReset();
   mockGetMappings.mockReset();
   mockGetReportMappings.mockReset();
-  mockDeleteReportMapping.mockReset();
-  mockIModelsClient.reset();
-  mockBulkExtractor.reset();
 });
 
 describe("Report Mappings View", () => {
@@ -312,16 +251,10 @@ describe("Report Mappings View", () => {
     const mockReportMappings = mockReportMappingsFactory();
     const mockMappings = mockMappingsFactory(mockReportMappings);
 
-    mockIModelsClient.setup(async (x) => x.getSingle(moq.It.isObjectWith<GetSingleIModelParams>({ iModelId: mockIModelId1 })))
-      .returns(async () => mockIModelsResponse[0].iModel);
-
-    mockIModelsClient.setup(async (x) => x.getSingle(moq.It.isObjectWith<GetSingleIModelParams>({ iModelId: mockIModelId2 })))
-      .returns(async () => mockIModelsResponse[1].iModel);
-
     mockGetMapping.mockReturnValueOnce(mockMappings[0].mapping).mockReturnValueOnce(mockMappings[1].mapping);
     mockGetReportMappings.mockReturnValueOnce(mockReportMappings.mappings);
 
-    render(<ReportMappings report={mockReport} bulkExtractor={mockBulkExtractor.object} goBack={jest.fn()} />);
+    render(<ReportMappings report={mockReport} onClickClose={jest.fn()} />, { reportsClient: mockReportsClient.object, mappingsClient: mockMappingsClient.object, iModelsClient: mockIModelsClient.object });
 
     await waitForElementToBeRemoved(() => screen.getByText(/loading/i));
 
@@ -333,23 +266,15 @@ describe("Report Mappings View", () => {
     const mockReportMappings = mockReportMappingsFactory();
     const mockMappings = mockMappingsFactory(mockReportMappings);
 
-    mockIModelsClient.setup(async (x) => x.getSingle(moq.It.isObjectWith<GetSingleIModelParams>({ iModelId: mockIModelId1 })))
-      .returns(async () => mockIModelsResponse[0].iModel);
-
-    mockIModelsClient.setup(async (x) => x.getSingle(moq.It.isObjectWith<GetSingleIModelParams>({ iModelId: mockIModelId2 })))
-      .returns(async () => mockIModelsResponse[1].iModel);
     mockGetMapping.mockReturnValueOnce(mockMappings[0].mapping).mockReturnValueOnce(mockMappings[1].mapping);
     mockGetReportMappings.mockReturnValueOnce(mockReportMappings.mappings);
 
-    const { user } = render(
-      <ReportMappings report={mockReport} bulkExtractor={mockBulkExtractor.object} goBack={jest.fn()} />
-    );
+    const { user } = render(<ReportMappings report={mockReport} onClickClose={jest.fn()} />, { reportsClient: mockReportsClient.object, mappingsClient: mockMappingsClient.object, iModelsClient: mockIModelsClient.object });
 
     await waitForElementToBeRemoved(() => screen.getByText(/loading/i));
 
-    const searchButton = within(screen.getByTestId(/rcw-search-bar/i)).getByRole(
-      "button"
-    );
+    const searchButton = screen.getByTestId(/rcw-search-button/i);
+
     await user.click(searchButton);
     const searchInput = screen.getByRole("textbox", {
       name: /search\-textbox/i,
@@ -387,18 +312,10 @@ describe("Report Mappings View", () => {
     const mockReportMappings = mockReportMappingsFactory();
     const mockMappings = mockMappingsFactory(mockReportMappings);
 
-    mockIModelsClient.setup(async (x) => x.getSingle(moq.It.isObjectWith<GetSingleIModelParams>({ iModelId: mockIModelId1 })))
-      .returns(async () => mockIModelsResponse[0].iModel);
-
-    mockIModelsClient.setup(async (x) => x.getSingle(moq.It.isObjectWith<GetSingleIModelParams>({ iModelId: mockIModelId2 })))
-      .returns(async () => mockIModelsResponse[1].iModel);
-
     mockGetMapping.mockReturnValueOnce(mockMappings[0].mapping).mockReturnValueOnce(mockMappings[1].mapping);
     mockGetReportMappings.mockReturnValueOnce(mockReportMappings.mappings);
 
-    const { user } = render(
-      <ReportMappings report={mockReport} bulkExtractor={mockBulkExtractor.object} goBack={jest.fn()} />
-    );
+    const { user } = render(<ReportMappings report={mockReport} onClickClose={jest.fn()} />, { reportsClient: mockReportsClient.object, mappingsClient: mockMappingsClient.object, iModelsClient: mockIModelsClient.object });
 
     await waitForElementToBeRemoved(() => screen.getByText(/loading/i));
 
@@ -414,7 +331,7 @@ describe("Report Mappings View", () => {
     expect(addMappingsModal).toBeInTheDocument();
 
     await act(async () => {
-      await returnFn();
+      await onClose();
     });
 
     const horizontalTiles = screen.getAllByTestId("horizontal-tile");
@@ -424,9 +341,7 @@ describe("Report Mappings View", () => {
   });
 
   it("odata feed url", async () => {
-    const { user } = render(
-      <ReportMappings report={mockReport} bulkExtractor={mockBulkExtractor.object} goBack={jest.fn()} />
-    );
+    const { user } = render(<ReportMappings report={mockReport} onClickClose={jest.fn()} />, { reportsClient: mockReportsClient.object, mappingsClient: mockMappingsClient.object, iModelsClient: mockIModelsClient.object });
 
     await waitForElementToBeRemoved(() => screen.getByText(/loading/i));
 
@@ -452,19 +367,16 @@ describe("Report Mappings View", () => {
     const mockReportMappings = mockReportMappingsFactory();
     const mockMappings = mockMappingsFactory(mockReportMappings);
 
-    mockIModelsClient.setup(async (x) => x.getSingle(moq.It.isObjectWith<GetSingleIModelParams>({ iModelId: mockIModelId1 })))
-      .returns(async () => mockIModelsResponse[0].iModel);
-
-    mockIModelsClient.setup(async (x) => x.getSingle(moq.It.isObjectWith<GetSingleIModelParams>({ iModelId: mockIModelId2 })))
-      .returns(async () => mockIModelsResponse[1].iModel);
-
     mockGetMapping.mockReturnValueOnce(mockMappings[0].mapping).mockReturnValueOnce(mockMappings[1].mapping);
     mockGetReportMappings.mockReturnValueOnce(mockReportMappings.mappings);
 
     const iModels = mockIModelsResponse.map((iModel) => iModel.iModel.id);
-    mockBulkExtractor.setup(async (x) => x.runIModelExtractions(iModels)).returns(async () => Promise.resolve());
 
-    const { user } = render(<ReportMappings report={mockReport} bulkExtractor={mockBulkExtractor.object} goBack={jest.fn()} />);
+    const bulkExtractor = new BulkExtractor(new ReportsClient(), new ExtractionClient(), jest.fn().mockResolvedValue("mockAccessToken"), jest.fn, jest.fn);
+
+    const runIModelExtractionsMock = jest.spyOn(bulkExtractor, "runIModelExtractions").mockImplementation(async () => Promise.resolve());
+
+    const { user } = render(<ReportMappings report={mockReport} onClickClose={jest.fn()} />, { reportsClient: mockReportsClient.object, mappingsClient: mockMappingsClient.object, iModelsClient: mockIModelsClient.object, bulkExtractor });
 
     await waitForElementToBeRemoved(() => screen.getByText(/loading/i));
 
@@ -472,7 +384,10 @@ describe("Report Mappings View", () => {
     expect(extractAllButton).toBeInTheDocument();
 
     await user.click(extractAllButton);
-    mockBulkExtractor.verify(async (x) => x.runIModelExtractions(iModels), moq.Times.once());
+    await waitFor(() => expect(extractAllButton).toBeEnabled());
+
+    // Check that the mocked method was called with the expected arguments
+    expect(runIModelExtractionsMock).toHaveBeenCalledWith(iModels);
   });
 
   const assertHorizontalTiles = (horizontalTiles: HTMLElement[], mockMappings: MappingSingle[]) => {

package/src/test/Reports.test.tsx

@@ -7,37 +7,18 @@ import "@testing-library/jest-dom";
 import {
   render,
   screen,
-  TestUtils,
   waitForElementToBeRemoved,
   within,
 } from "../test/test-utils";
 import { Reports } from "../widget/components/Reports";
-import type {
-  IModelConnection,
-  SelectionSet,
-  SelectionSetEvent,
-} from "@itwin/core-frontend";
-import { NoRenderApp } from "@itwin/core-frontend";
 import { ReportsConfigWidget } from "../ReportsConfigWidget";
 import { rest } from "msw";
 import { setupServer } from "msw/node";
-import * as moq from "typemoq";
-import type {
-  SelectionManager,
-  SelectionScopesManager,
-} from "@itwin/presentation-frontend";
-import {
-  Presentation,
-  SelectionChangeEvent,
-} from "@itwin/presentation-frontend";
 import faker from "@faker-js/faker";
 import type { ReportCollection } from "@itwin/insights-client";
 import userEvent from "@testing-library/user-event";
 import { REPORTS_CONFIG_BASE_URL } from "../widget/ReportsConfigUiProvider";
-import type { BeEvent } from "@itwin/core-bentley";
-
-const mockITwinId = faker.datatype.uuid();
-const mockIModelId = faker.datatype.uuid();
+import { EmptyLocalization } from "@itwin/core-common";
 
 const reportsFactory = (): ReportCollection => ({
   reports: Array.from(
@@ -62,56 +43,15 @@ const reportsFactory = (): ReportCollection => ({
   },
 });
 
-const connectionMock = moq.Mock.ofType<IModelConnection>();
-const selectionManagerMock = moq.Mock.ofType<SelectionManager>();
-const selectionScopesManagerMock = moq.Mock.ofType<SelectionScopesManager>();
-
-jest.mock("@itwin/appui-react", () => ({
-  ...jest.requireActual("@itwin/appui-react"),
-  useActiveIModelConnection: () => connectionMock.object,
-}));
-
-jest.mock("../widget/components/ReportMappings", () => ({
-  ReportMappings: () => "MockReportMappings",
-}));
-
 const server = setupServer();
 
 beforeAll(async () => {
-  // This is required by the i18n module within iTwin.js
-  (global as any).XMLHttpRequest = require("xmlhttprequest").XMLHttpRequest; // eslint-disable-line @typescript-eslint/no-var-requires
-
-  await NoRenderApp.startup();
-  await Presentation.initialize();
-  const selectionSet = moq.Mock.ofType<SelectionSet>();
-  const onChanged = moq.Mock.ofType<BeEvent<(ev: SelectionSetEvent) => void>>();
-  selectionSet.setup((x) => x.elements).returns(() => new Set([]));
-  selectionSet.setup((x) => x.onChanged).returns(() => onChanged.object);
-  connectionMock
-    .setup((x) => x.selectionSet)
-    .returns(() => selectionSet.object);
-  connectionMock.setup((x) => x.iModelId).returns(() => mockIModelId);
-  connectionMock.setup((x) => x.iTwinId).returns(() => mockITwinId);
-
-  selectionManagerMock
-    .setup((x) => x.selectionChange)
-    .returns(() => new SelectionChangeEvent());
-
-  selectionScopesManagerMock
-    .setup(async (x) => x.getSelectionScopes(connectionMock.object))
-    .returns(async () => []);
-  selectionManagerMock
-    .setup((x) => x.scopes)
-    .returns(() => selectionScopesManagerMock.object);
-
-  Presentation.setSelectionManager(selectionManagerMock.object);
-  await TestUtils.initializeUiFramework(connectionMock.object);
-  await ReportsConfigWidget.initialize();
+  const localization = new EmptyLocalization();
+  await ReportsConfigWidget.initialize(localization);
   server.listen();
 });
 
 afterAll(() => {
-  TestUtils.terminateUiFramework();
   server.close();
 });
 
@@ -132,14 +72,16 @@ describe("Reports View", () => {
       )
     );
 
-    const { user } = render(<Reports />);
+    const onClickAddMock = jest.fn();
+
+    const { user } = render(<Reports onClickAddReport={onClickAddMock} />);
 
     await waitForElementToBeRemoved(() => screen.getByText(/loading/i));
     const ctaButton = screen.getByRole("button", {
       name: /createonereportcta/i,
     });
     await user.click(ctaButton);
-    expect(screen.getByText(/addreport/i)).toBeInTheDocument();
+    expect(onClickAddMock).toBeCalled();
   });
 
   it("be able to add new report", async () => {
@@ -156,14 +98,15 @@ describe("Reports View", () => {
       )
     );
 
-    const { user } = render(<Reports />);
+    const onClickAddMock = jest.fn();
+    const { user } = render(<Reports onClickAddReport={onClickAddMock} />);
 
     await waitForElementToBeRemoved(() => screen.getByText(/loading/i));
     const newButton = screen.getByRole("button", {
       name: /new/i,
     });
     await user.click(newButton);
-    expect(screen.getByText(/addreport/i)).toBeInTheDocument();
+    expect(onClickAddMock).toBeCalled();
   });
 
   it("list all reports", async () => {
@@ -209,8 +152,8 @@ describe("Reports View", () => {
         }
       )
     );
-
-    const { user } = render(<Reports />);
+    const onClickModifyMock = jest.fn();
+    const { user } = render(<Reports onClickReportModify={onClickModifyMock} />);
 
     await waitForElementToBeRemoved(() => screen.getByText(/loading/i));
 
@@ -221,7 +164,7 @@ describe("Reports View", () => {
     const modifyButton = screen.getByRole("menuitem", { name: /modify/i });
     await user.click(modifyButton);
 
-    expect(screen.getByText(/modifyreport/i)).toBeInTheDocument();
+    expect(onClickModifyMock).toBeCalled();
   });
 
   it("remove a report", async () => {
@@ -291,9 +234,8 @@ describe("Reports View", () => {
 
     await waitForElementToBeRemoved(() => screen.getByText(/loading/i));
 
-    const searchButton = within(screen.getByTestId(/rcw-search-bar/i)).getByRole(
-      "button"
-    );
+    const searchButton = screen.getByTestId(/rcw-search-button/i);
+
     await user.click(searchButton);
     const searchInput = screen.getByRole("textbox", {
       name: /search\-textbox/i,
@@ -331,8 +273,8 @@ describe("Reports View", () => {
         }
       )
     );
-
-    const { user } = render(<Reports />);
+    const onClickModifyMock = jest.fn();
+    const { user } = render(<Reports onClickReportModify={onClickModifyMock} />);
 
     await waitForElementToBeRemoved(() => screen.getByText(/loading/i));
 
@@ -342,8 +284,7 @@ describe("Reports View", () => {
     await user.click(firstMenuDropdown);
     const modifyButton = screen.getByRole("menuitem", { name: /modify/i });
     await user.click(modifyButton);
-    // Modify report should appear
-    expect(screen.getByText(/modifyreport/i)).toBeInTheDocument();
+    expect(onClickModifyMock).toBeCalled();
   });
 
   it("click a report", async () => {
@@ -356,8 +297,8 @@ describe("Reports View", () => {
         }
       )
     );
-
-    const { user } = render(<Reports />);
+    const onClickTitleMock = jest.fn();
+    const { user } = render(<Reports onClickReportTitle={onClickTitleMock} />);
 
     await waitForElementToBeRemoved(() => screen.getByText(/loading/i));
 
@@ -365,6 +306,6 @@ describe("Reports View", () => {
       mockedReports.reports[0].displayName
     );
     await user.click(reportName);
-    expect(screen.getByText(/MockReportMappings/i)).toBeInTheDocument();
+    expect(onClickTitleMock).toBeCalled();
   });
 });

package/src/test/WidgetHeader.test.tsx

@@ -7,8 +7,15 @@ import "@testing-library/jest-dom";
 import { render, screen } from "../test/test-utils";
 import { WidgetHeader } from "../widget/components/utils";
 import { faker } from "@faker-js/faker";
+import { EmptyLocalization } from "@itwin/core-common";
+import { ReportsConfigWidget } from "../ReportsConfigWidget";
 
 describe("Widget Header Component", () => {
+  beforeAll(async () => {
+    const localization = new EmptyLocalization();
+    await ReportsConfigWidget.initialize(localization);
+  });
+
   it("title renders", async () => {
     const fakeTitle = faker.word.interjection();
     render(<WidgetHeader title={fakeTitle} />);
@@ -20,9 +27,9 @@ describe("Widget Header Component", () => {
 
   it("header allows return callback", async () => {
     const mockReturnFn = jest.fn();
-    const { user } = render(<WidgetHeader title="" returnFn={mockReturnFn} />);
+    const { user, container } = render(<WidgetHeader title="" returnFn={mockReturnFn} />);
 
-    const returnBtn = screen.getByRole("button");
+    const returnBtn = container.getElementsByClassName("rcw-chevron")[0];
     await user.click(returnBtn);
     expect(mockReturnFn).toHaveBeenCalledTimes(1);
   });

package/src/test/test-utils.tsx

@@ -4,35 +4,26 @@
 *--------------------------------------------------------------------------------------------*/
 import * as React from "react";
 import { render as rtlRender } from "@testing-library/react";
-import { ReportsApiConfigContext } from "../widget/context/ReportsApiConfigContext";
-import type { Store } from "redux";
-import { combineReducers, createStore } from "redux";
-import {
-  FrameworkReducer,
-  SyncUiEventDispatcher,
-  UiFramework,
-} from "@itwin/appui-react";
-import type { IModelConnection } from "@itwin/core-frontend";
-import { ReportsConfigWidget } from "../ReportsConfigWidget";
 import userEvent from "@testing-library/user-event";
-import { REPORTS_CONFIG_BASE_URL } from "../widget/ReportsConfigUiProvider";
+import type { ReportsConfigContextProps } from "../widget/components/ReportsConfigContext";
+import { ReportsConfigContext } from "../widget/components/ReportsConfigContext";
 
 export const mockAccessToken = async () => "Bearer eyJhbGci";
 
 interface WrapperProps {
   children?: React.ReactNode;
 }
+export const mockITwinId: string = "mockiTwinId";
 
-function render(ui: React.ReactElement, { ...options } = {}) {
+function render(ui: React.ReactElement, contextProps?: Partial<ReportsConfigContextProps>, { ...options } = {}) {
   const Wrapper = ({ children }: WrapperProps) => (
-    <ReportsApiConfigContext.Provider
-      value={{
-        getAccessToken: mockAccessToken,
-        baseUrl: REPORTS_CONFIG_BASE_URL,
-      }}
+    <ReportsConfigContext
+      getAccessToken={mockAccessToken}
+      iTwinId={contextProps?.iTwinId ?? mockITwinId}
+      {...contextProps}
     >
       {children}
-    </ReportsApiConfigContext.Provider>
+    </ReportsConfigContext>
   );
 
   return {
@@ -44,34 +35,3 @@ function render(ui: React.ReactElement, { ...options } = {}) {
 export * from "@testing-library/react";
 // override React Testing Library's render with our own
 export { render };
-
-export class TestUtils {
-  public static createAppStore(): Store {
-    const rootReducer = combineReducers({
-      frameworkState: FrameworkReducer,
-    });
-
-    return createStore(
-      rootReducer,
-      (window as any).__REDUX_DEVTOOLS_EXTENSION__ &&
-      (window as any).__REDUX_DEVTOOLS_EXTENSION__()
-    );
-  }
-
-  public static async initializeUiFramework(_?: IModelConnection) {
-    const store = this.createAppStore();
-
-    await UiFramework.initialize(store);
-
-    // Set the iModelConnection in the Redux store
-    // Can not get it to function as intended. Store does not update with the iModel connection
-    // if (imodel)
-    //   UiFramework.setIModelConnection(imodel);
-
-    SyncUiEventDispatcher.setTimeoutPeriod(0); // disables non-immediate event processing.
-  }
-
-  public static terminateUiFramework() {
-    ReportsConfigWidget.terminate();
-  }
-}