feat(store): more tests

Closes ngrx#2363
evgenyfedorenko · May 10, 2020 · 4bcddb4 · 4bcddb4
1 parent 09d57ad
commit 4bcddb4
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/modules/store/testing/spec/mock_store.spec.ts b/modules/store/testing/spec/mock_store.spec.ts
@@ -1,5 +1,5 @@
 import { TestBed, ComponentFixture } from '@angular/core/testing';
-import { skip, take, tap } from 'rxjs/operators';
+import { skip, take } from 'rxjs/operators';
 import { MockStore, provideMockStore } from '@ngrx/store/testing';
 import {
   Store,
@@ -10,10 +10,12 @@ import {
   createFeatureSelector,
 } from '@ngrx/store';
 import { INCREMENT } from '../../spec/fixtures/counter';
-import { Component, OnInit } from '@angular/core';
+import { Component } from '@angular/core';
 import { Observable } from 'rxjs';
 import { By } from '@angular/platform-browser';
 
+import * as store from '@ngrx/store';
+
 interface TestAppSchema {
   counter1: number;
   counter2: number;
@@ -40,6 +42,8 @@ describe('Mock Store', () => {
   );
 
   beforeEach(() => {
+    spyOn(store, 'setNgrxMockEnvironment').and.callThrough();
+
     TestBed.configureTestingModule({
       providers: [
         provideMockStore({
@@ -63,10 +67,15 @@ describe('Mock Store', () => {
     mockStore.resetSelectors();
   });
 
+  it('should set NgrxMockEnvironment', () => {
+    expect(store.setNgrxMockEnvironment).toHaveBeenCalledWith(true);
+  });
+
   it('should provide the same instance with Store and MockStore', () => {
     const fromStore = TestBed.inject(Store);
     const fromMockStore = TestBed.inject(MockStore);
     expect(fromStore).toBe(fromMockStore);
+    expect(store.setNgrxMockEnvironment).toHaveBeenCalled();
   });
 
   it('should set the initial state to a mocked one', (done: any) => {