feat(store): cleanup selector after a test

ngrx · Sep 24, 2019 · 2964e2b · 2964e2b
1 parent 88ebdde
commit 2964e2b
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/modules/store/testing/src/mock_store.ts b/modules/store/testing/src/mock_store.ts
@@ -1,4 +1,5 @@
 import { Inject, Injectable } from '@angular/core';
+import { TestBed } from '@angular/core/testing';
 import { Observable, BehaviorSubject } from 'rxjs';
 import {
   Action,
@@ -14,6 +15,17 @@ import { MockState } from './mock_state';
 import { MockSelector } from './mock_selector';
 import { MOCK_SELECTORS } from './tokens';
 
+if (typeof afterEach === 'function') {
+  afterEach(() => {
+    try {
+      const store = TestBed.get(Store) as MockStore<any>;
+      if (store && 'resetSelectors' in store) {
+        store.resetSelectors();
+      }
+    } catch {}
+  });
+}
+
 @Injectable()
 export class MockStore<T> extends Store<T> {
   static selectors = new Map<