add mock for @tsndr/cloudflare-worker-jwt and generateUniqueToken, fix test cases

Author: prakashchoudhary07

__mocks__/@tsndr/cloudflare-worker-jwt.js

@@ -1,14 +1,8 @@
-"use strict";
-import jsonwebtoken from "jsonwebtoken";
-
-const cloudflareWorkerJwt = jest.createMockFromModule("jsonwebtoken");
-
-function sign() {
-  console.log("ASD");
-  return "asd";
-}
-
-// @ts-ignore
-cloudflareWorkerJwt.sign = sign;
-
-module.exports = cloudflareWorkerJwt;
+// __mocks__/@tsndr/cloudflare-worker-jwt.js
+const mockJwt = {
+  sign: jest.fn().mockImplementation(() => {
+    return "SIGNED_JWT";
+  }),
+};
+
+export default mockJwt;

__mocks__/@tsndr/cloudflare-worker-jwt.ts

@@ -5,7 +5,7 @@
   "scripts": {
     "start": "wrangler dev",
     "test": "echo 'running tests 🧪' && jest --coverage && echo '✅ All tests passed'",
-    "test-watch": "jest --watch",
+    "test-watch": "jest . --watch",
     "login": "wrangler login",
     "check": "wrangler whoami",
     "deploy": "wrangler deploy",

package.json

@@ -3,7 +3,7 @@ import { RETRY_COMMAND, VERIFICATION_STRING } from "../constants/responses";
 import { env } from "../typeDefinitions/default.types";
 import { discordEphemeralResponse } from "../utils/discordEphemeralResponse";
 import { generateUniqueToken } from "../utils/generateUniqueToken";
-import  {sendUserDiscordData} from "../utils/sendUserDiscordData";
+import { sendUserDiscordData } from "../utils/sendUserDiscordData";
 
 export async function verifyCommand(
   userId: number,
@@ -13,7 +13,8 @@ export async function verifyCommand(
   env: env
 ) {
   const token = await generateUniqueToken();
-  const response = await sendUserDiscordData(    token,
+  const response = await sendUserDiscordData(
+    token,
     userId,
     userAvatarHash,
     userName,

src/controllers/verifyCommand.ts

@@ -1,4 +1,3 @@
-const crypto = require("crypto");
 export const generateUniqueToken = async () => {
   const uuidToken = crypto.randomUUID();
   const randomNumber = Math.floor(Math.random() * 1000000);

src/utils/generateUniqueToken.ts

@@ -1,89 +1,105 @@
-import * as response from "../../../src/constants/responses";
-import JSONResponse from "../../../src/utils/JsonResponse";
-import { verifyCommand } from "../../../src/controllers/verifyCommand";
-import { guildEnv } from "../../fixtures/fixture";
+import {
+  RETRY_COMMAND,
+  VERIFICATION_STRING,
+} from "../../../src/constants/responses";
 import config from "../../../config/config";
 
-describe("verifyCommand", () => {
-  test("should return INTERNAL_SERVER_ERROR when response is not ok", async () => {
-    
-    jest.mock("crypto", () => {
-      return {
-        randomUUID: jest.fn(()=>'shreya'),
-        subtle: { digest: jest.fn(() => "123") },
-      };
-    });
+const mockDateNow = 1626512345678;
+const UNIQUE_TOKEN = "UNIQUE_TOKEN";
+const env = {
+  BOT_PUBLIC_KEY: "BOT_PUBLIC_KEY",
+  DISCORD_GUILD_ID: "DISCORD_GUILD_ID",
+  DISCORD_TOKEN: "SIGNED_JWT",
+};
 
+describe("verifyCommand", () => {
+  beforeEach(() => {
+    jest.mock("@tsndr/cloudflare-worker-jwt");
+    jest.spyOn(Date, "now").mockReturnValue(mockDateNow);
     jest.mock("../../../src/utils/generateUniqueToken", () => ({
-      generateUniqueToken: () => Promise.resolve("jashdkjahskajhd"),
+      generateUniqueToken: () => Promise.resolve(UNIQUE_TOKEN),
     }));
+  });
 
-    // const mockResponse = response.INTERNAL_SERVER_ERROR;
-    // jest
-    //   .spyOn(global, "fetch")
-    //   .mockImplementation(() =>
-    //     Promise.resolve(new JSONResponse(mockResponse))
-    //   );
-
-    const env = {
-      BOT_PUBLIC_KEY: "xyz",
-      DISCORD_GUILD_ID: "123",
-      DISCORD_TOKEN: "abc",
-    };
+  afterEach(() => {
+    jest.spyOn(Date, "now").mockRestore();
+  });
 
+  test("should return JSON response when response is ok", async () => {
     const data = {
-      token: 1233434,
-      userId: "sjkhdkjashdksjh",
-      userAvatarHash: "test user",
-      userName: "sndbhsbgdj",
-      env: env,
+      type: "discord",
+      token: UNIQUE_TOKEN,
+      attributes: {
+        discordId: 1,
+        userAvatar: "https://cdn.discordapp.com/avatars/1/userAvatarHash.jpg",
+        userName: "userName",
+        discriminator: "discriminator",
+        expiry: mockDateNow + 1000 * 60 * 2,
+      },
     };
 
+    jest.spyOn(global, "fetch").mockResolvedValueOnce({
+      ok: true,
+      status: 200,
+      json: jest.fn().mockResolvedValueOnce(data),
+    } as unknown as Response);
+
+    const { verifyCommand } = await import(
+      "../../../src/controllers/verifyCommand"
+    );
+
     const result = await verifyCommand(
-      1233434,
-      "sjkhdkjashdksjh",
-      "test user",
-      "sndbhsbgdj",
+      1,
+      "userAvatarHash",
+      "userName",
+      "discriminator",
       env
     );
 
-    // expect(result.data.content).toEqual(response.RETRY_COMMAND);
     expect(global.fetch).toHaveBeenCalledWith(
-      `https://api.realdevsquad.com/external-accounts`,
+      `http://localhost:3000/external-accounts`,
       {
         method: "POST",
         headers: {
           "Content-Type": "application/json",
-          Authorization: `Bot ${guildEnv.DISCORD_TOKEN}`,
+          Authorization: `Bearer ${env.DISCORD_TOKEN}`,
         },
         body: JSON.stringify(data),
       }
     );
-  });
-
-  test("should return JSON response when response is ok", async () => {
-    const mockResponse = {};
+    const resultText = await result.text();
+    const resultData = JSON.parse(resultText);
 
-    // jest
-    //   .spyOn(global, "fetch")
-    //   .mockImplementation(() =>
-    //     Promise.resolve(new JSONResponse(mockResponse))
-    //   );
+    const verificationSiteURL = config(env).VERIFICATION_SITE_URL;
+    const message =
+      `${verificationSiteURL}/discord?token=${UNIQUE_TOKEN}\n` +
+      VERIFICATION_STRING;
 
-    const env = {
-      BOT_PUBLIC_KEY: "xyz",
-      DISCORD_GUILD_ID: "123",
-      DISCORD_TOKEN: "abc",
-    };
+    expect(resultData.data.content).toEqual(message);
+  });
 
+  test("should return INTERNAL_SERVER_ERROR when response is not ok", async () => {
     const data = {
-      token: 1233434,
-      userId: "sjkhdkjashdksjh",
-      userAvatarHash: "test user",
-      userName: "sndbhsbgdj",
-      env: env,
+      type: "discord",
+      token: UNIQUE_TOKEN,
+      attributes: {
+        discordId: 1,
+        userAvatar: "https://cdn.discordapp.com/avatars/1/userAvatarHash.jpg",
+        userName: "userName",
+        discriminator: "discriminator",
+        expiry: mockDateNow + 1000 * 60 * 2,
+      },
     };
 
+    jest.spyOn(global, "fetch").mockResolvedValueOnce({
+      ok: true,
+      status: 400, // ERROR STATUS
+      json: jest.fn().mockResolvedValueOnce(data),
+    } as unknown as Response);
+
+    const { verifyCommand } = await import(
+      "../../../src/controllers/verifyCommand"
+    );
     const result = await verifyCommand(
       1233434,
       "sjkhdkjashdksjh",
@@ -92,21 +108,20 @@ describe("verifyCommand", () => {
       env
     );
 
-    const verificationSiteURL = config(env).VERIFICATION_SITE_URL;
-    const message =
-      `${verificationSiteURL}/discord?token=${guildEnv.DISCORD_TOKEN}\n` +
-      response.VERIFICATION_STRING;
-
     expect(global.fetch).toHaveBeenCalledWith(
-      `https://api.realdevsquad.com/external-accounts`,
+      `http://localhost:3000/external-accounts`,
       {
         method: "POST",
         headers: {
           "Content-Type": "application/json",
-          Authorization: `Bot ${guildEnv.DISCORD_TOKEN}`,
+          Authorization: `Bearer ${env.DISCORD_TOKEN}`,
         },
         body: JSON.stringify(data),
       }
     );
+    const resultText = await result.text();
+    const resultData = JSON.parse(resultText);
+
+    expect(resultData.data.content).toEqual(RETRY_COMMAND);
   });
 });