qlik-oss/sn-table

/* eslint react/require-default-props: 0 */
import { ThemeProvider } from "@mui/material/styles";
import { stardust } from "@nebula.js/stardust";
import type { ExtendedTheme } from "@qlik/nebula-table-utils/lib/hooks/use-extended-theme/types";
import React from "react";
import { EMPTY_TABLE_DATA, TableContextProvider } from "../table/context";
import muiSetup from "../table/mui-setup";
import { FocusedCellCoord } from "../table/types";
import {
  ApplyColumnWidths,
  ChangeSortOrder,
  ExtendedSelectionAPI,
  TableData,
  TableLayout,
  ViewService,
} from "../types";
import { generateLayout } from "./generate-test-data";

interface ProviderProps {
  app?: EngineAPI.IApp;
  children?: JSX.Element;
  selectionsAPI?: ExtendedSelectionAPI;
  tableData?: TableData;
  cellCoordMock?: FocusedCellCoord;
  layout?: TableLayout;
  model?: EngineAPI.IGenericObject;
  translator?: stardust.Translator;
  interactions?: stardust.Interactions;
  theme?: ExtendedTheme;
  keyboard?: stardust.Keyboard;
  direction?: "ltr" | "rtl";
  rootElement?: HTMLElement;
  embed?: stardust.Embed;
  changeSortOrder?: ChangeSortOrder;
  applyColumnWidths?: ApplyColumnWidths;
  initialDataPages?: EngineAPI.INxDataPage[];
  rect?: stardust.Rect;
  viewService?: ViewService;
  isNewHeadCellMenuEnabled?: boolean;
}

type HookWrapperProps = { children: JSX.Element };

const TestWithProviders = ({
  children,
  app = {
    getField: () => Promise.resolve({}),
    createSessionObject: () => Promise.resolve({}),
  } as unknown as EngineAPI.IApp,
  layout = generateLayout(1, 1, 5),
  interactions = {
    active: true,
    passive: true,
    select: true,
  },
  selectionsAPI = {
    isModal: () => false,
    on: () => undefined,
    removeListener: () => undefined,
  } as unknown as ExtendedSelectionAPI,
  model = {
    getHyperCubeData: () => Promise.resolve(),
  } as unknown as EngineAPI.IGenericObject,
  keyboard = {} as stardust.Keyboard,
  translator = { get: (s: string) => s } as unknown as stardust.Translator,
  theme = {
    getColorPickerColor: () => undefined,
    name: () => undefined,
    getStyle: () => undefined,
    background: { isDark: false },
  } as unknown as ExtendedTheme,
  tableData = EMPTY_TABLE_DATA,
  cellCoordMock = undefined,
  direction = "ltr",
  rootElement = {} as HTMLElement,
  embed = {} as stardust.Embed,
  changeSortOrder = async () => {},
  applyColumnWidths = () => {},
  rect = { width: 0, height: 0, top: 0, left: 0 },
  initialDataPages = undefined,
  viewService = { qTop: 0, qHeight: 1, scrollLeft: 0, estimatedRowHeight: 25 },
  isNewHeadCellMenuEnabled = false,
}: ProviderProps) => (
  <ThemeProvider theme={muiSetup(direction)}>
    <TableContextProvider
      app={app}
      selectionsAPI={selectionsAPI}
      layout={layout}
      translator={translator}
      interactions={interactions}
      theme={theme}
      keyboard={keyboard}
      model={model}
      tableData={tableData}
      cellCoordMock={cellCoordMock}
      rootElement={rootElement}
      embed={embed}
      changeSortOrder={changeSortOrder}
      applyColumnWidths={applyColumnWidths}
      rect={rect}
      initialDataPages={initialDataPages}
      viewService={viewService}
      isNewHeadCellMenuEnabled={isNewHeadCellMenuEnabled}
    >
      {children as JSX.Element}
    </TableContextProvider>
  </ThemeProvider>
);

export const wrapper = ({ children }: HookWrapperProps) => <TestWithProviders>{children}</TestWithProviders>;

export default TestWithProviders;