import { Store } from '@ngrx/store';
import {
    of,
} from 'rxjs';

export const initMockStore = <T>(jest) => (state: T) => {
    const mockStore = {
        addReducer: jest.fn(),
        dispatch: jest.fn(),
        select: jest.fn((selector: (stateSelector) => {} | string) => {
            if (typeof(selector) === 'string') {
                return of(state[selector as string]);
            }
            return of(selector(state));
        }),
    };
    return mockStore as Store<T>;
};