Refactor Project and User middleware tests; update method names for c…

…larity and improve mock implementations
OneUptime · Sep 26, 2024 · a10a83a · a10a83a
1 parent bf01322
commit a10a83a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/Common/Tests/Server/Middleware/ProjectAuthorization.test.ts b/Common/Tests/Server/Middleware/ProjectAuthorization.test.ts
@@ -18,6 +18,7 @@ import { describe, expect, afterEach, jest } from "@jest/globals";
 import getJestMockFunction from "Common/Tests/MockType";
 import { getJestSpyOn } from "Common/Tests/Spy";
 import { TestDatabaseMock } from "../TestingUtils/__mocks__/TestDatabase.mock";
+import APIKeyAccessPermission from "../../../Server/Utils/APIKey/AccessPermission";
 
 jest.mock("../../../Server/Services/ApiKeyService");
 jest.mock("../../../Server/Services/AccessTokenService");
@@ -250,7 +251,7 @@ describe("ProjectMiddleware", () => {
     test("should call Response.sendErrorResponse when apiKeyModel is not null but getApiTenantAccessPermission returned null", async () => {
       jest.spyOn(ApiKeyService, "findOneBy").mockResolvedValue(mockedApiModel);
       const spyGetApiTenantAccessPermission: jest.SpyInstance = getJestSpyOn(
-        AccessTokenService,
+        APIKeyAccessPermission,
         "getApiTenantAccessPermission",
       ).mockImplementationOnce(getJestMockFunction().mockResolvedValue(null));
 
@@ -272,7 +273,7 @@ describe("ProjectMiddleware", () => {
         {} as UserTenantAccessPermission;
       jest.spyOn(ApiKeyService, "findOneBy").mockResolvedValue(mockedApiModel);
       const spyGetApiTenantAccessPermission: jest.SpyInstance = getJestSpyOn(
-        AccessTokenService,
+        APIKeyAccessPermission,
         "getApiTenantAccessPermission",
       ).mockResolvedValue(mockedUserTenantAccessPermission);
 

diff --git a/Common/Tests/Server/Middleware/UserAuthorization.test.ts b/Common/Tests/Server/Middleware/UserAuthorization.test.ts
@@ -57,7 +57,7 @@ describe("UserMiddleware", () => {
     jest.clearAllMocks();
   });
 
-  describe("getAccessToken", () => {
+  describe("getAccessTokenFromExpressRequest", () => {
     test("should return access token when authorization token is passed in the cookie", () => {
       const req: ExpressRequest = {
         cookies: { "user-token": mockedAccessToken },
@@ -281,7 +281,7 @@ describe("UserMiddleware", () => {
         projectId,
       );
       getJestSpyOn(ProjectMiddleware, "hasApiKey").mockReturnValue(false);
-      getJestSpyOn(UserMiddleware, "getAccessToken").mockReturnValue(
+      getJestSpyOn(UserMiddleware, "getAccessTokenFromExpressRequest").mockReturnValue(
         mockedAccessToken,
       );
       getJestSpyOn(JSONWebToken, "decode").mockReturnValue(jwtTokenData);
@@ -304,7 +304,7 @@ describe("UserMiddleware", () => {
 
       const spyGetAccessToken: jest.SpyInstance = getJestSpyOn(
         UserMiddleware,
-        "getAccessToken",
+        "getAccessTokenFromExpressRequest",
       );
 
       await UserMiddleware.getUserMiddleware(req, res, next);
@@ -315,10 +315,10 @@ describe("UserMiddleware", () => {
       expect(spyGetAccessToken).not.toHaveBeenCalled();
     });
 
-    test("should call function 'next' and return, when getAccessToken returns a null value", async () => {
+    test("should call function 'next' and return, when getAccessTokenFromExpressRequest returns a null value", async () => {
       const spyGetAccessToken: jest.SpyInstance = getJestSpyOn(
         UserMiddleware,
-        "getAccessToken",
+        "getAccessTokenFromExpressRequest",
       ).mockReturnValueOnce(undefined);
 
       await UserMiddleware.getUserMiddleware(req, res, next);