@itwin/reports-config-widget-react 0.0.8 → 0.2.0

package/lib/cjs/test/ReportMappings.test.js

@@ -31,19 +31,13 @@ const faker_1 = __importDefault(require("@faker-js/faker"));
 require("@testing-library/jest-dom");
 const core_frontend_1 = require("@itwin/core-frontend");
 const ReportsConfigWidget_1 = require("../ReportsConfigWidget");
-const node_1 = require("msw/node");
 const test_utils_1 = require("./test-utils");
-const user_event_1 = __importDefault(require("@testing-library/user-event"));
 const moq = __importStar(require("typemoq"));
-const msw_1 = require("msw");
-const insights_client_1 = require("@itwin/insights-client");
 const ReportMappings_1 = require("../widget/components/ReportMappings");
 const imodels_client_management_1 = require("@itwin/imodels-client-management");
-const ReportsConfigUiProvider_1 = require("../widget/ReportsConfigUiProvider");
-const Extraction_1 = require("../widget/components/Extraction");
 const presentation_frontend_1 = require("@itwin/presentation-frontend");
-// For the extraction test
-jest.setTimeout(20000);
+const itwinui_react_1 = require("@itwin/itwinui-react");
+const core_common_1 = require("@itwin/core-common");
 const mockITwinId = faker_1.default.datatype.uuid();
 // Lets work with two iModels for now.
 const mockIModelId1 = faker_1.default.datatype.uuid();
@@ -184,23 +178,61 @@ const mockMappingsFactory = (mockReportMappings) => {
             },
         },
     }));
-    const iModelHandlers = mockMappings.map((mapping) => msw_1.rest.get(`${ReportsConfigUiProvider_1.REPORTS_CONFIG_BASE_URL}/insights/reporting/datasources/imodels/${mapping.mapping._links.imodel.href}/mappings/${mapping.mapping.id}`, async (_req, res, ctx) => {
-        return res(ctx.delay(), ctx.status(200), ctx.json(mapping));
-    }));
-    return [mockMappings, iModelHandlers];
+    return mockMappings;
 };
 const connectionMock = moq.Mock.ofType();
+const mockBulkExtractor = moq.Mock.ofType();
 const selectionManagerMock = moq.Mock.ofType();
 const selectionScopesManagerMock = moq.Mock.ofType();
+const mockIModelsClient = moq.Mock.ofType();
+jest.mock("../widget/components/Constants.ts", () => ({
+    STATUS_CHECK_INTERVAL: 10,
+}));
+jest.mock("../widget/components/ReportMappingHorizontalTile", () => ({
+    ReportMappingHorizontalTile: (props) => {
+        return (react_1.default.createElement("div", { "data-testid": "horizontal-tile" },
+            react_1.default.createElement(itwinui_react_1.Text, null, props.mapping.mappingName),
+            react_1.default.createElement(itwinui_react_1.Text, { title: props.mapping.mappingDescription }, props.mapping.iModelName)));
+    },
+}));
 jest.mock("@itwin/appui-react", () => ({
     ...jest.requireActual("@itwin/appui-react"),
     useActiveIModelConnection: () => connectionMock.object,
 }));
-const server = node_1.setupServer();
+let returnFn;
+jest.mock("../widget/components/AddMappingsModal", () => ({
+    ...jest.requireActual("../widget/components/AddMappingsModal"),
+    AddMappingsModal: (props) => {
+        returnFn = props.returnFn;
+        return react_1.default.createElement("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;
+    }),
+}));
+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,
+    })),
+}));
 beforeAll(async () => {
-    // This is required by the i18n module within iTwin.js
-    global.XMLHttpRequest = require("xmlhttprequest").XMLHttpRequest; // eslint-disable-line @typescript-eslint/no-var-requires
-    await core_frontend_1.NoRenderApp.startup();
+    await core_frontend_1.NoRenderApp.startup({ localization: new core_common_1.EmptyLocalization() });
     await presentation_frontend_1.Presentation.initialize();
     const selectionSet = moq.Mock.ofType();
     const onChanged = moq.Mock.ofType();
@@ -223,53 +255,46 @@ beforeAll(async () => {
     presentation_frontend_1.Presentation.setSelectionManager(selectionManagerMock.object);
     await test_utils_1.TestUtils.initializeUiFramework(connectionMock.object);
     await ReportsConfigWidget_1.ReportsConfigWidget.initialize();
-    server.listen();
 });
 afterAll(() => {
     test_utils_1.TestUtils.terminateUiFramework();
-    server.close();
 });
 afterEach(() => {
-    server.resetHandlers();
+    mockGetMapping.mockReset();
+    mockGetMappings.mockReset();
+    mockGetReportMappings.mockReset();
+    mockDeleteReportMapping.mockReset();
+    mockIModelsClient.reset();
+    mockBulkExtractor.reset();
 });
 describe("Report Mappings View", () => {
     it("shows all report mappings", async () => {
-        var _a, _b;
         const mockReportMappings = mockReportMappingsFactory();
-        const [mockMappings, iModelHandlers] = mockMappingsFactory(mockReportMappings);
-        server.use(msw_1.rest.get(`${ReportsConfigUiProvider_1.REPORTS_CONFIG_BASE_URL}/insights/reporting/reports/${mockReportId}/datasources/imodelMappings`, async (_req, res, ctx) => {
-            return res(ctx.delay(), ctx.status(200), ctx.json(mockReportMappings));
-        }), msw_1.rest.get(`${imodels_client_management_1.Constants.api.baseUrl}/${mockIModelId1}`, async (_req, res, ctx) => {
-            return res(ctx.delay(), ctx.status(200), ctx.json(mockIModelsResponse[0]));
-        }), msw_1.rest.get(`${imodels_client_management_1.Constants.api.baseUrl}/${mockIModelId2}`, async (_req, res, ctx) => {
-            return res(ctx.delay(), ctx.status(200), ctx.json(mockIModelsResponse[1]));
-        }), ...iModelHandlers);
-        test_utils_1.render(react_1.default.createElement(ReportMappings_1.ReportMappings, { report: mockReport, goBack: jest.fn() }));
+        const mockMappings = mockMappingsFactory(mockReportMappings);
+        mockIModelsClient.setup(async (x) => x.getSingle(moq.It.isObjectWith({ iModelId: mockIModelId1 })))
+            .returns(async () => mockIModelsResponse[0].iModel);
+        mockIModelsClient.setup(async (x) => x.getSingle(moq.It.isObjectWith({ iModelId: mockIModelId2 })))
+            .returns(async () => mockIModelsResponse[1].iModel);
+        mockGetMapping.mockReturnValueOnce(mockMappings[0].mapping).mockReturnValueOnce(mockMappings[1].mapping);
+        mockGetReportMappings.mockReturnValueOnce(mockReportMappings.mappings);
+        test_utils_1.render(react_1.default.createElement(ReportMappings_1.ReportMappings, { report: mockReport, bulkExtractor: mockBulkExtractor.object, goBack: jest.fn() }));
         await test_utils_1.waitForElementToBeRemoved(() => test_utils_1.screen.getByText(/loading/i));
         const horizontalTiles = test_utils_1.screen.getAllByTestId("horizontal-tile");
-        expect(horizontalTiles).toHaveLength(mockMappings.length);
-        for (const [index, horizontalTile] of horizontalTiles.entries()) {
-            const reportMappingTile = test_utils_1.within(horizontalTile);
-            const mockiModel = mockIModelsResponse.find((iModel) => iModel.iModel.id === mockMappings[index].mapping._links.imodel.href);
-            expect(reportMappingTile.getByText(mockMappings[index].mapping.mappingName)).toBeInTheDocument();
-            expect(reportMappingTile.getByTitle((_a = mockMappings[index].mapping.description) !== null && _a !== void 0 ? _a : "")).toBeInTheDocument();
-            expect(reportMappingTile.getByText((_b = mockiModel === null || mockiModel === void 0 ? void 0 : mockiModel.iModel.displayName) !== null && _b !== void 0 ? _b : "")).toBeInTheDocument();
-        }
+        assertHorizontalTiles(horizontalTiles, mockMappings);
     });
     it("search for a report mapping", async () => {
         var _a, _b, _c, _d;
         const mockReportMappings = mockReportMappingsFactory();
-        const [mockMappings, iModelHandlers] = mockMappingsFactory(mockReportMappings);
-        server.use(msw_1.rest.get(`${ReportsConfigUiProvider_1.REPORTS_CONFIG_BASE_URL}/insights/reporting/reports/${mockReportId}/datasources/imodelMappings`, async (_req, res, ctx) => {
-            return res(ctx.delay(), ctx.status(200), ctx.json(mockReportMappings));
-        }), msw_1.rest.get(`${imodels_client_management_1.Constants.api.baseUrl}/${mockIModelId1}`, async (_req, res, ctx) => {
-            return res(ctx.delay(), ctx.status(200), ctx.json(mockIModelsResponse[0]));
-        }), msw_1.rest.get(`${imodels_client_management_1.Constants.api.baseUrl}/${mockIModelId2}`, async (_req, res, ctx) => {
-            return res(ctx.delay(), ctx.status(200), ctx.json(mockIModelsResponse[1]));
-        }), ...iModelHandlers);
-        const { user } = test_utils_1.render(react_1.default.createElement(ReportMappings_1.ReportMappings, { report: mockReport, goBack: jest.fn() }));
+        const mockMappings = mockMappingsFactory(mockReportMappings);
+        mockIModelsClient.setup(async (x) => x.getSingle(moq.It.isObjectWith({ iModelId: mockIModelId1 })))
+            .returns(async () => mockIModelsResponse[0].iModel);
+        mockIModelsClient.setup(async (x) => x.getSingle(moq.It.isObjectWith({ iModelId: mockIModelId2 })))
+            .returns(async () => mockIModelsResponse[1].iModel);
+        mockGetMapping.mockReturnValueOnce(mockMappings[0].mapping).mockReturnValueOnce(mockMappings[1].mapping);
+        mockGetReportMappings.mockReturnValueOnce(mockReportMappings.mappings);
+        const { user } = test_utils_1.render(react_1.default.createElement(ReportMappings_1.ReportMappings, { report: mockReport, bulkExtractor: mockBulkExtractor.object, goBack: jest.fn() }));
         await test_utils_1.waitForElementToBeRemoved(() => test_utils_1.screen.getByText(/loading/i));
-        const searchButton = test_utils_1.within(test_utils_1.screen.getByTestId(/search-bar/i)).getByRole("button");
+        const searchButton = test_utils_1.within(test_utils_1.screen.getByTestId(/rcw-search-bar/i)).getByRole("button");
         await user.click(searchButton);
         const searchInput = test_utils_1.screen.getByRole("textbox", {
             name: /search\-textbox/i,
@@ -290,133 +315,33 @@ describe("Report Mappings View", () => {
         expect(test_utils_1.screen.getAllByTestId("horizontal-tile")).toHaveLength(1);
         expect(test_utils_1.screen.getByText((_d = iModel === null || iModel === void 0 ? void 0 : iModel.iModel.displayName) !== null && _d !== void 0 ? _d : "")).toBeInTheDocument();
     });
-    it("remove a report mapping", async () => {
-        const mockReportMappings = mockReportMappingsFactory();
-        const [_, iModelHandlers] = mockMappingsFactory(mockReportMappings);
-        const mockReportMappingsOriginalSize = mockReportMappings.mappings.length;
-        server.use(msw_1.rest.get(`${ReportsConfigUiProvider_1.REPORTS_CONFIG_BASE_URL}/insights/reporting/reports/${mockReportId}/datasources/imodelMappings`, async (_req, res, ctx) => {
-            return res(ctx.delay(), ctx.status(200), ctx.json(mockReportMappings));
-        }), msw_1.rest.get(`${imodels_client_management_1.Constants.api.baseUrl}/${mockIModelId1}`, async (_req, res, ctx) => {
-            return res(ctx.delay(), ctx.status(200), ctx.json(mockIModelsResponse[0]));
-        }), msw_1.rest.get(`${imodels_client_management_1.Constants.api.baseUrl}/${mockIModelId2}`, async (_req, res, ctx) => {
-            return res(ctx.delay(), ctx.status(200), ctx.json(mockIModelsResponse[1]));
-        }), ...iModelHandlers, msw_1.rest.delete(`${ReportsConfigUiProvider_1.REPORTS_CONFIG_BASE_URL}/insights/reporting/reports/${mockReportId}/datasources/imodelMappings/${mockReportMappings.mappings[0].mappingId}`, async (_req, res, ctx) => {
-            mockReportMappings.mappings = mockReportMappings.mappings.filter((mapping) => mapping.mappingId !== mockReportMappings.mappings[0].mappingId);
-            return res(ctx.delay(100), ctx.status(204));
-        }));
-        const { user } = test_utils_1.render(react_1.default.createElement(ReportMappings_1.ReportMappings, { report: mockReport, goBack: jest.fn() }));
-        await test_utils_1.waitForElementToBeRemoved(() => test_utils_1.screen.getByText(/loading/i));
-        const firstMenuDropdown = test_utils_1.within(test_utils_1.screen.getAllByTestId(/tile-action-button/i)[0]).getByRole("button");
-        await user.click(firstMenuDropdown);
-        const removeButton = test_utils_1.screen.getByRole("menuitem", { name: /remove/i });
-        await user.click(removeButton);
-        // Delete modal dialog should appear
-        expect(test_utils_1.screen.getByRole("dialog")).toBeInTheDocument();
-        const deleteButton = test_utils_1.screen.getByRole("button", {
-            name: /delete/i,
-        });
-        await user.click(deleteButton);
-        await test_utils_1.waitForElementToBeRemoved(() => test_utils_1.screen.getByTestId(/rcw-loading-delete/i));
-        await test_utils_1.waitForElementToBeRemoved(() => test_utils_1.screen.getByRole("dialog"));
-        // Should be one less mapping
-        expect(test_utils_1.screen.getAllByTestId("horizontal-tile")).toHaveLength(mockReportMappingsOriginalSize - 1);
-    });
     it("add mapping", async () => {
         const mockReportMappings = mockReportMappingsFactory();
-        const [mockMappings, iModelHandlers] = mockMappingsFactory(mockReportMappings);
-        // Adding an extra unmapped mapping.
-        const extraMappingId = faker_1.default.datatype.uuid();
-        const extraMappingName = "mOcKNaMeExTrA";
-        const extraMappingDescription = "mOcKDeScRiPtIoNeXtRa";
-        mockMappings.push({
-            mapping: {
-                id: extraMappingId,
-                mappingName: extraMappingName,
-                description: extraMappingDescription,
-                extractionEnabled: false,
-                createdOn: "",
-                createdBy: "",
-                modifiedOn: "",
-                modifiedBy: "",
-                _links: {
-                    imodel: {
-                        href: "",
-                    },
-                },
-            },
-        });
-        const mockMappingsResponse = {
-            // Type guarding
-            mappings: mockMappings
-                .map((mapping) => mapping.mapping)
-                .filter((mapping) => !!mapping),
-            _links: {
-                next: undefined,
-                self: {
-                    href: "",
-                },
-            },
-        };
-        server.use(msw_1.rest.get(`${ReportsConfigUiProvider_1.REPORTS_CONFIG_BASE_URL}/insights/reporting/reports/${mockReportId}/datasources/imodelMappings`, async (_req, res, ctx) => {
-            return res(ctx.delay(), ctx.status(200), ctx.json(mockReportMappings));
-        }), msw_1.rest.get(`${imodels_client_management_1.Constants.api.baseUrl}`, async (_req, res, ctx) => {
-            return res(ctx.delay(), ctx.status(200), ctx.json(mockProjectIModels));
-        }), msw_1.rest.get(`${imodels_client_management_1.Constants.api.baseUrl}/${mockIModelId1}`, async (_req, res, ctx) => {
-            return res(ctx.delay(), ctx.status(200), ctx.json(mockIModelsResponse[0]));
-        }), msw_1.rest.get(`${imodels_client_management_1.Constants.api.baseUrl}/${mockIModelId2}`, async (_req, res, ctx) => {
-            return res(ctx.delay(), ctx.status(200), ctx.json(mockIModelsResponse[1]));
-        }), msw_1.rest.get(`${ReportsConfigUiProvider_1.REPORTS_CONFIG_BASE_URL}/insights/reporting/datasources/imodels/${mockProjectIModels.iModels[0].id}/mappings`, async (_req, res, ctx) => {
-            return res(ctx.delay(), ctx.status(200), ctx.json(mockMappingsResponse));
-        }), msw_1.rest.post(`${ReportsConfigUiProvider_1.REPORTS_CONFIG_BASE_URL}/insights/reporting/reports/${mockReportId}/datasources/imodelMappings`, async (_req, res, ctx) => {
-            return res(ctx.delay(), ctx.status(200), ctx.json(mockReportMappings));
-        }), ...iModelHandlers);
-        const { user } = test_utils_1.render(react_1.default.createElement(ReportMappings_1.ReportMappings, { report: mockReport, goBack: jest.fn() }));
+        const mockMappings = mockMappingsFactory(mockReportMappings);
+        mockIModelsClient.setup(async (x) => x.getSingle(moq.It.isObjectWith({ iModelId: mockIModelId1 })))
+            .returns(async () => mockIModelsResponse[0].iModel);
+        mockIModelsClient.setup(async (x) => x.getSingle(moq.It.isObjectWith({ iModelId: mockIModelId2 })))
+            .returns(async () => mockIModelsResponse[1].iModel);
+        mockGetMapping.mockReturnValueOnce(mockMappings[0].mapping).mockReturnValueOnce(mockMappings[1].mapping);
+        mockGetReportMappings.mockReturnValueOnce(mockReportMappings.mappings);
+        const { user } = test_utils_1.render(react_1.default.createElement(ReportMappings_1.ReportMappings, { report: mockReport, bulkExtractor: mockBulkExtractor.object, goBack: jest.fn() }));
         await test_utils_1.waitForElementToBeRemoved(() => test_utils_1.screen.getByText(/loading/i));
+        mockGetMappings.mockReturnValueOnce(mockMappings.map((m) => m.mapping));
         const addMappingButton = test_utils_1.screen.getByRole("button", {
             name: /addmapping/i,
         });
         await user.click(addMappingButton);
-        await test_utils_1.waitForElementToBeRemoved(() => test_utils_1.screen.getByTestId(/rcw-action-loading-spinner/i));
-        // Add modal dialog should appear
-        const modal = test_utils_1.screen.getByRole("dialog");
-        expect(test_utils_1.screen.getByRole("dialog")).toBeInTheDocument();
-        const withinModal = test_utils_1.within(modal);
-        expect(withinModal.getByText(/addmappings/i)).toBeInTheDocument();
-        const addButton = withinModal.getByRole("button", {
-            name: /add/i,
-        });
-        // Add button should be disabled
-        expect(addButton).toBeDisabled();
-        // Already mapped mappings are disabled
-        for (let i = 0; i < mockMappings.length - 1; i++) {
-            const row = test_utils_1.screen.getByRole("row", {
-                name: new RegExp(`${mockMappings[i].mapping.mappingName} ${mockMappings[i].mapping.description}`, "i"),
-            });
-            const checkbox = test_utils_1.within(row).getByRole("checkbox");
-            expect(checkbox).toBeDisabled();
-        }
-        // Click on checkbox on new mapping
-        const unmappedRow = test_utils_1.screen.getByRole("row", {
-            name: new RegExp(`${mockMappings[mockMappings.length - 1].mapping.mappingName} ${mockMappings[mockMappings.length - 1].mapping.description}`, "i"),
+        const addMappingsModal = await test_utils_1.screen.findByTestId("add-mappings-modal");
+        expect(addMappingsModal).toBeInTheDocument();
+        await test_utils_1.act(async () => {
+            await returnFn();
         });
-        const enabledCheckbox = test_utils_1.within(unmappedRow).getByRole("checkbox");
-        await user.click(enabledCheckbox);
-        await user.click(addButton);
-        // Modal should go away
-        await test_utils_1.waitForElementToBeRemoved(() => test_utils_1.screen.getByTestId(/rcw-action-loading-spinner/i));
-        await test_utils_1.waitForElementToBeRemoved(() => test_utils_1.screen.getByRole("dialog"));
+        const horizontalTiles = test_utils_1.screen.getAllByTestId("horizontal-tile");
+        assertHorizontalTiles(horizontalTiles, mockMappings);
+        expect(mockGetReportMappings).toBeCalledTimes(2);
     });
     it("odata feed url", async () => {
-        const mockReportMappings = mockReportMappingsFactory();
-        const [_, iModelHandlers] = mockMappingsFactory(mockReportMappings);
-        server.use(msw_1.rest.get(`${ReportsConfigUiProvider_1.REPORTS_CONFIG_BASE_URL}/insights/reporting/reports/${mockReportId}/datasources/imodelMappings`, async (_req, res, ctx) => {
-            return res(ctx.delay(), ctx.status(200), ctx.json(mockReportMappings));
-        }), msw_1.rest.get(`${imodels_client_management_1.Constants.api.baseUrl}/${mockIModelId1}`, async (_req, res, ctx) => {
-            return res(ctx.delay(), ctx.status(200), ctx.json(mockIModelsResponse[0]));
-        }), msw_1.rest.get(`${imodels_client_management_1.Constants.api.baseUrl}/${mockIModelId2}`, async (_req, res, ctx) => {
-            return res(ctx.delay(), ctx.status(200), ctx.json(mockIModelsResponse[1]));
-        }), ...iModelHandlers);
-        const { user } = test_utils_1.render(react_1.default.createElement(ReportMappings_1.ReportMappings, { report: mockReport, goBack: jest.fn() }));
+        const { user } = test_utils_1.render(react_1.default.createElement(ReportMappings_1.ReportMappings, { report: mockReport, bulkExtractor: mockBulkExtractor.object, goBack: jest.fn() }));
         await test_utils_1.waitForElementToBeRemoved(() => test_utils_1.screen.getByText(/loading/i));
         const urlTextbox = test_utils_1.screen.getByRole("textbox", {
             name: /odatafeedurl/i,
@@ -429,111 +354,34 @@ describe("Report Mappings View", () => {
         await user.click(copyButton);
         expect(test_utils_1.screen.getByText(/copiedtoclipboard/i)).toBeInTheDocument();
     });
-    it("full extraction", async () => {
+    it("update all datasets", async () => {
         const mockReportMappings = mockReportMappingsFactory();
-        const [_, iModelHandlers] = mockMappingsFactory(mockReportMappings);
-        const delay = Extraction_1.REFRESH_DELAY + 1000;
-        // Faking timers currently makes all promise based queries from RTL become unpredictable.
-        // https://github.com/testing-library/dom-testing-library/issues/988
-        // Should come back to this later.
-        // Consequently, this test will be a bit slower.
-        // jest.useFakeTimers()
-        Element.prototype.scrollIntoView = jest.fn();
-        const mockIModel = mockIModelsResponse[0].iModel;
-        const mockRunId = faker_1.default.datatype.uuid();
-        const mockExtractionResponse = {
-            run: {
-                id: mockRunId,
-                _links: {
-                    status: {
-                        href: "",
-                    },
-                },
-            },
-        };
-        let mockStatusResponse = {
-            status: {
-                state: insights_client_1.ExtractorState.Queued,
-                reason: "",
-                containsIssues: false,
-                _links: {
-                    logs: {
-                        href: "",
-                    },
-                },
-            },
-        };
-        server.use(msw_1.rest.get(`${ReportsConfigUiProvider_1.REPORTS_CONFIG_BASE_URL}/insights/reporting/reports/${mockReportId}/datasources/imodelMappings`, async (_req, res, ctx) => {
-            return res(ctx.delay(), ctx.status(200), ctx.json(mockReportMappings));
-        }), msw_1.rest.get(`${imodels_client_management_1.Constants.api.baseUrl}/${mockIModelId1}`, async (_req, res, ctx) => {
-            return res(ctx.delay(), ctx.status(200), ctx.json(mockIModelsResponse[0]));
-        }), msw_1.rest.get(`${imodels_client_management_1.Constants.api.baseUrl}/${mockIModelId2}`, async (_req, res, ctx) => {
-            return res(ctx.delay(), ctx.status(200), ctx.json(mockIModelsResponse[1]));
-        }), ...iModelHandlers, msw_1.rest.post(`${ReportsConfigUiProvider_1.REPORTS_CONFIG_BASE_URL}/insights/reporting/datasources/imodels/${mockIModel.id}/extraction/run`, async (_req, res, ctx) => {
-            return res(ctx.delay(800), ctx.status(200), ctx.json(mockExtractionResponse));
-        }), msw_1.rest.get(`${ReportsConfigUiProvider_1.REPORTS_CONFIG_BASE_URL}/insights/reporting/datasources/extraction/status/${mockRunId}`, async (_req, res, ctx) => {
-            return res(ctx.delay(), ctx.status(200), ctx.json(mockStatusResponse));
-        }));
-        test_utils_1.render(react_1.default.createElement(ReportMappings_1.ReportMappings, { report: mockReport, goBack: jest.fn() }));
-        // https://github.com/testing-library/user-event/issues/833
-        const user = user_event_1.default.setup({ delay: null });
+        const mockMappings = mockMappingsFactory(mockReportMappings);
+        mockIModelsClient.setup(async (x) => x.getSingle(moq.It.isObjectWith({ iModelId: mockIModelId1 })))
+            .returns(async () => mockIModelsResponse[0].iModel);
+        mockIModelsClient.setup(async (x) => x.getSingle(moq.It.isObjectWith({ 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 } = test_utils_1.render(react_1.default.createElement(ReportMappings_1.ReportMappings, { report: mockReport, bulkExtractor: mockBulkExtractor.object, goBack: jest.fn() }));
         await test_utils_1.waitForElementToBeRemoved(() => test_utils_1.screen.getByText(/loading/i));
-        const comboBox = test_utils_1.screen.getByRole("combobox", {
-            name: /updatedataset/i,
-        });
-        await user.type(comboBox, mockIModel.displayName);
-        const option = test_utils_1.screen.getByRole("menuitem", {
-            name: mockIModel.displayName,
-        });
-        await user.click(option);
-        // Combobox should have correct status
-        const extractionComponent = test_utils_1.screen.getByTestId("extraction-combo-box");
-        expect(test_utils_1.within(extractionComponent).getByDisplayValue(mockIModel.displayName)).toBeInTheDocument();
-        // Should be two in the document. One in the status and the other in the list.
-        // TODO Assert that it is in the correct HorizontalTile
-        const startingStates = await test_utils_1.screen.findAllByTitle(/starting/i);
-        expect(startingStates).toHaveLength(2);
-        const loadingStates = await test_utils_1.screen.findAllByTitle(/loading/i);
-        expect(loadingStates).toHaveLength(2);
-        // act(() => {
-        //   jest.advanceTimersByTime(2000)
-        // });
-        const queuedStates = await test_utils_1.screen.findAllByTitle(/queued/i, undefined, {
-            timeout: delay,
-        });
-        expect(queuedStates).toHaveLength(2);
-        mockStatusResponse = {
-            status: {
-                state: insights_client_1.ExtractorState.Running,
-                reason: "",
-                containsIssues: false,
-                _links: {
-                    logs: {
-                        href: "",
-                    },
-                },
-            },
-        };
-        const runningStates = await test_utils_1.screen.findAllByTitle(/running/i, undefined, {
-            timeout: delay,
-        });
-        expect(runningStates).toHaveLength(2);
-        mockStatusResponse = {
-            status: {
-                state: insights_client_1.ExtractorState.Succeeded,
-                reason: "",
-                containsIssues: false,
-                _links: {
-                    logs: {
-                        href: "",
-                    },
-                },
-            },
-        };
-        const succeededStates = await test_utils_1.screen.findAllByTitle(/success/i, undefined, {
-            timeout: delay,
-        });
-        expect(succeededStates).toHaveLength(2);
+        const extractAllButton = test_utils_1.screen.getByText(/UpdateAllDatasets/i);
+        expect(extractAllButton).toBeInTheDocument();
+        await user.click(extractAllButton);
+        mockBulkExtractor.verify(async (x) => x.runIModelExtractions(iModels), moq.Times.once());
     });
+    const assertHorizontalTiles = (horizontalTiles, mockMappings) => {
+        var _a, _b;
+        expect(horizontalTiles).toHaveLength(mockMappings.length);
+        for (const [index, horizontalTile] of horizontalTiles.entries()) {
+            const reportMappingTile = test_utils_1.within(horizontalTile);
+            const mockiModel = mockIModelsResponse.find((iModel) => iModel.iModel.id === mockMappings[index].mapping._links.imodel.href);
+            expect(reportMappingTile.getByText(mockMappings[index].mapping.mappingName)).toBeInTheDocument();
+            expect(reportMappingTile.getByTitle((_a = mockMappings[index].mapping.description) !== null && _a !== void 0 ? _a : "")).toBeInTheDocument();
+            expect(reportMappingTile.getByText((_b = mockiModel === null || mockiModel === void 0 ? void 0 : mockiModel.iModel.displayName) !== null && _b !== void 0 ? _b : "")).toBeInTheDocument();
+        }
+    };
 });
 //# sourceMappingURL=ReportMappings.test.js.map