Fix tests

responsible-ai-collaborative · Jan 27, 2025 · e14ec88 · e14ec88
1 parent a65ecf5
commit e14ec88
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/site/gatsby-site/server/tests/entities.spec.ts b/site/gatsby-site/server/tests/entities.spec.ts
@@ -1,6 +1,6 @@
 import { expect, jest, it } from '@jest/globals';
 import { ApolloServer } from "@apollo/server";
-import { makeRequest, seedFixture, startTestServer } from "./utils";
+import { makeRequest, mockSession, seedFixture, startTestServer } from "./utils";
 import * as context from '../context';
 
 describe(`Entities`, () => {
@@ -73,7 +73,7 @@ describe(`Entities`, () => {
     });
 
 
-    jest.spyOn(context, 'verifyToken').mockResolvedValue({ sub: "123" })
+    mockSession("123");
 
     const response = await makeRequest(url, mutationData);
 

diff --git a/site/gatsby-site/server/tests/notifications.spec.ts b/site/gatsby-site/server/tests/notifications.spec.ts
@@ -1629,8 +1629,7 @@ describe(`Notifications`, () => {
             }
         });
 
-
-        jest.spyOn(context, 'verifyToken').mockResolvedValue({ sub: "123" })
+        mockSession('123');
 
         // No recipients
         jest.spyOn(common, 'getUserAdminData').mockResolvedValue(null);