feat(StoreDevtools): Pass stateSanitizer as argument instead

ngrx · Feb 12, 2018 · c6d6110 · c6d6110
1 parent c54d8bf
commit c6d6110
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 10 deletions.
diff --git a/modules/store-devtools/spec/extension.spec.ts b/modules/store-devtools/spec/extension.spec.ts
@@ -2,11 +2,7 @@ import { Action } from '@ngrx/store';
 import { of } from 'rxjs/observable/of';
 
 import { LiftedState } from '../';
-import {
-  DevtoolsExtension,
-  ReduxDevtoolsExtension,
-  ReduxDevtoolsExtensionConnection,
-} from '../src/extension';
+import { DevtoolsExtension, ReduxDevtoolsExtension } from '../src/extension';
 import { createConfig, noMonitor } from '../src/instrument';
 
 describe('DevtoolsExtension', () => {

diff --git a/modules/store-devtools/spec/store.spec.ts b/modules/store-devtools/spec/store.spec.ts
@@ -710,7 +710,7 @@ describe('Store Devtools', () => {
       const liftedState = fixture.getLiftedState();
       const sanitizedLiftedState = fixture.devtools.getSanitizedState(
         liftedState,
-        devtoolsOptions
+        devtoolsOptions.stateSanitizer
       );
       const originalAction =
         liftedState.actionsById[liftedState.nextActionId - 1];

diff --git a/modules/store-devtools/src/devtools.ts b/modules/store-devtools/src/devtools.ts
@@ -28,7 +28,11 @@ import {
   ComputedState,
 } from './reducer';
 import * as Actions from './actions';
-import { StoreDevtoolsConfig, STORE_DEVTOOLS_CONFIG } from './config';
+import {
+  StoreDevtoolsConfig,
+  STORE_DEVTOOLS_CONFIG,
+  StateSanitizer,
+} from './config';
 
 @Injectable()
 export class DevtoolsDispatcher extends ActionsSubject {}
@@ -78,7 +82,7 @@ export class StoreDevtools implements Observer<any> {
           // Extension should be sent the sanitized lifted state
           extension.notify(
             action,
-            this.getSanitizedState(reducedLiftedState, config)
+            this.getSanitizedState(reducedLiftedState, config.stateSanitizer)
           );
 
           return { state: reducedLiftedState, action };
@@ -110,8 +114,8 @@ export class StoreDevtools implements Observer<any> {
    * Restructures the lifted state passed in to prepare for sending to the
    * Redux Devtools Extension
    */
-  getSanitizedState(state: LiftedState, config: Partial<StoreDevtoolsConfig>) {
-    const sanitizedComputedStates = config.stateSanitizer
+  getSanitizedState(state: LiftedState, stateSanitizer?: StateSanitizer) {
+    const sanitizedComputedStates = stateSanitizer
       ? state.computedStates.map((entry: ComputedState) => ({
           state: entry.sanitizedState,
           error: entry.error,