chore: fixes from review feedback

ngrx · Oct 5, 2019 · a1f987d · a1f987d
1 parent 603e09e
commit a1f987d
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/modules/store/testing/src/mock_store.ts b/modules/store/testing/src/mock_store.ts
@@ -36,7 +36,7 @@ export class MockStore<T> extends Store<T> {
   >();
 
   public scannedActions$: Observable<Action>;
-  private lastState: T | undefined;
+  private lastState: T;
 
   constructor(
     private state$: MockState<T>,
@@ -47,8 +47,7 @@ export class MockStore<T> extends Store<T> {
   ) {
     super(state$, actionsObserver, reducerManager);
     this.resetSelectors();
-    this.state$.next(this.initialState);
-    this.lastState = this.initialState;
+    this.setState(this.initialState);
     this.scannedActions$ = actionsObserver.asObservable();
     if (mockSelectors) {
       mockSelectors.forEach(mockSelector => {