fix storybook query mocking

Joystream · Nov 29, 2023 · 0de0c5b · 0de0c5b
1 parent 99164bb
commit 0de0c5b
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 20 deletions.
diff --git a/packages/ui/src/app/pages/Validators/ValidatorList.stories.tsx b/packages/ui/src/app/pages/Validators/ValidatorList.stories.tsx
@@ -153,20 +153,14 @@ export default {
             },
           },
         },
-        queryNode: [
-          {
-            query: GetMembersWithDetailsDocument,
-            resolver: ({ variables } = {}) => {
-              if (variables?.where) {
-                return {
-                  loading: false,
-                  data: { memberships: [member('alice'), member('bob'), member('charlie'), member('dave')] },
-                }
-              }
-              return { loading: false, data: undefined }
+        gql: {
+          queries: [
+            {
+              query: GetMembersWithDetailsDocument,
+              data: { memberships: [member('alice'), member('bob'), member('charlie'), member('dave')] },
             },
-          },
-        ],
+          ],
+        },
       }
     },
   },

diff --git a/packages/ui/src/validators/hooks/useValidatorsList.tsx b/packages/ui/src/validators/hooks/useValidatorsList.tsx
@@ -55,13 +55,9 @@ export const useValidatorsList = () => {
   }
 
   const getValidatorsInfo = (api: Api, validatorsWithMembership: ValidatorMembership[]) => {
-    return of(validatorsWithMembership).pipe(
-      switchMap((validatorsWithMembership) => {
-        const validatorAddresses = validatorsWithMembership.map(({ stashAccount }) => stashAccount)
-        const validatorInfoObservables = validatorAddresses.map((address) => getValidatorInfo(address, api))
-        return combineLatest(validatorInfoObservables)
-      })
-    )
+    const validatorAddresses = validatorsWithMembership.map(({ stashAccount }) => stashAccount)
+    const validatorInfoObservables = validatorAddresses.map((address) => getValidatorInfo(address, api))
+    return combineLatest(validatorInfoObservables)
   }
 
   const allValidators = useFirstObservableValue(