superset-frontend/spec/helpers/testing-library.tsx

/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */
import '@testing-library/jest-dom/extend-expect';
import React, { ReactNode, ReactElement } from 'react';
import { render, RenderOptions } from '@testing-library/react';
import { ThemeProvider, supersetTheme } from '@superset-ui/core';
import { Provider } from 'react-redux';
import { combineReducers, createStore, applyMiddleware, compose } from 'redux';
import thunk from 'redux-thunk';
import reducerIndex from 'spec/helpers/reducerIndex';

type Options = Omit<RenderOptions, 'queries'> & {
  useRedux?: boolean;
  initialState?: {};
  reducers?: {};
};

function createWrapper(options?: Options) {
  const { useRedux, initialState, reducers } = options || {};

  if (useRedux) {
    const store = createStore(
      combineReducers(reducers || reducerIndex),
      initialState || {},
      compose(applyMiddleware(thunk)),
    );

    return ({ children }: { children?: ReactNode }) => (
      <Provider store={store}>
        <ThemeProvider theme={supersetTheme}>{children}</ThemeProvider>
      </Provider>
    );
  }

  return ({ children }: { children?: ReactNode }) => (
    <ThemeProvider theme={supersetTheme}>{children}</ThemeProvider>
  );
}

const customRender = (ui: ReactElement, options?: Options) =>
  render(ui, { wrapper: createWrapper(options), ...options });

export function sleep(time: number) {
  return new Promise(resolve => {
    setTimeout(resolve, time);
  });
}

export * from '@testing-library/react';
export { customRender as render };