import { Store } from '@ngrx/store';
import { Observable } from 'rxjs/Rx';
import { AppStateInterface } from '../models/index';

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

export default initMockStore;