From dfa1d54dee6fd0e2d3fe7bf089481c5f80265bf2 Mon Sep 17 00:00:00 2001
From: Theophile Sandoz <theophile.sandoz@gmail.com>
Date: Thu, 23 Nov 2023 13:50:55 +0100
Subject: [PATCH] Fix tests again

---
 .../SettingsNotificationsTab.stories.tsx      | 25 ++++++-------------
 packages/ui/src/mocks/providers/gql.tsx       |  4 +--
 2 files changed, 10 insertions(+), 19 deletions(-)

diff --git a/packages/ui/src/app/pages/Settings/SettingsNotificationsTab.stories.tsx b/packages/ui/src/app/pages/Settings/SettingsNotificationsTab.stories.tsx
index 8012c34b9e..713bdb5282 100644
--- a/packages/ui/src/app/pages/Settings/SettingsNotificationsTab.stories.tsx
+++ b/packages/ui/src/app/pages/Settings/SettingsNotificationsTab.stories.tsx
@@ -81,16 +81,14 @@ export default {
             },
             {
               query: GetBackendMeDocument,
-              data: args.isAuthorized
-                ? {
-                    me: {
-                      email: args.isEmailConfirmed ? email : null,
-                      unverifiedEmail: args.isEmailConfirmed ? null : email,
-                      receiveEmails: true,
-                      name: 'test',
-                    },
-                  }
-                : undefined,
+              data: {
+                me: {
+                  email: args.isEmailConfirmed ? email : null,
+                  unverifiedEmail: args.isEmailConfirmed ? null : email,
+                  receiveEmails: true,
+                  name: 'test',
+                },
+              },
               error: args.isAuthorized ? undefined : new Error('Unauthorized'),
             },
           ],
@@ -110,13 +108,6 @@ export default {
         },
 
         backend: {
-          notificationsSettingsMap: args.isAuthorized
-            ? {
-                [alice.id]: {
-                  accessToken: 'token',
-                },
-              }
-            : {},
           onSetMemberSettings: (...settingsArgs: any[]) => args.onSetMemberSettings(...settingsArgs),
           authToken: SIGNIN_TOKEN,
         },
diff --git a/packages/ui/src/mocks/providers/gql.tsx b/packages/ui/src/mocks/providers/gql.tsx
index 5bf6d8056b..8b6c50f3fa 100644
--- a/packages/ui/src/mocks/providers/gql.tsx
+++ b/packages/ui/src/mocks/providers/gql.tsx
@@ -100,9 +100,9 @@ export const MockGqlProvider: FC<MockGqlProps> = ({ children, queries, mutations
   const queriesMap = useMemo(
     () =>
       new Map(
-        queries?.map<[DocumentNode, Resolver]>(({ query, data, resolver }) => [
+        queries?.map<[DocumentNode, Resolver]>(({ query, data, error, resolver }) => [
           query,
-          (options) => resolver?.(options) ?? { loading: false, data },
+          (options) => resolver?.(options) ?? { loading: false, data: error ? undefined : data, error },
         ]) ?? []
       ),
     [queries]