/*
 * Copyright 2019 Red Hat, Inc. and/or its affiliates.
 *
 * Licensed 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 { Middleware } from 'redux';
import createMockStore, { MockStoreCreator } from 'redux-mock-store';
import thunk, { ThunkDispatch } from 'redux-thunk';
import WebSocketClient from '../websocket/WebSocketClient';
import { UpdateRouteAction } from './route/types';
import { AppState } from './types';
import { WebSocketAction } from './websocket/types';

jest.mock('../websocket/WebSocketClient');

export const mockStore = (state: AppState) => {
  const client = new WebSocketClient('');
  const middlewares: Middleware[] = [thunk.withExtraArgument(client)];
  type DispatchExts = ThunkDispatch<AppState, WebSocketClient, WebSocketAction | UpdateRouteAction>;
  const mockStoreCreator: MockStoreCreator<AppState, DispatchExts> = (
    createMockStore<AppState, DispatchExts>(middlewares)
  );
  return { store: mockStoreCreator(state), client };
};
