fix(local-traits): fix broken test

gabrielguerrero · Mar 30, 2022 · a453684 · a453684
1 parent 14a2dd7
commit a453684
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/libs/ngrx-traits/testing/src/provide-mock-local-traits.ts b/libs/ngrx-traits/testing/src/provide-mock-local-traits.ts
@@ -21,7 +21,7 @@ export function provideMockLocalTraits<
 }: {
   traitFactory: Type<T>;
   selectors?: {
-    [key in keyof T['localSelectors']]?: ReturnType<T['localSelectors'][key]>;
+    [key in keyof T['selectors']]?: ReturnType<T['selectors'][key]>;
   };
 }): Provider[] {
   return [
@@ -42,11 +42,8 @@ export function provideMockLocalTraits<
         if (selectors) {
           Object.keys(selectors).forEach(
             (key) =>
-              trait.localSelectors?.[key] &&
-              mockStore.overrideSelector(
-                trait.localSelectors[key],
-                selectors?.[key]
-              )
+              trait.selectors?.[key] &&
+              mockStore.overrideSelector(trait.selectors[key], selectors?.[key])
           );
         }
         return trait;