VIA-629 AJ/EO Add testing headers functionality for open campaigns only in prod environment

Author: elenaoanea-nhs

src/utils/date.test.ts

@@ -122,39 +122,77 @@ describe("utils-date", () => {
   });
 
   describe("getNow", () => {
+    const ENVIRONMENT = process.env;
     const fakeDateInSystem = "2000-01-01T01:01:01Z";
     const fakeDateInHeader = "1212-12-12T12:12:12Z";
 
     beforeEach(() => {
-      jest.useFakeTimers();
-      jest.setSystemTime(new Date(fakeDateInSystem));
-    });
-
-    it("should return current date by default", async () => {
-      expect(await getNow()).toEqual(new Date(fakeDateInSystem));
-    });
-
-    it("should return date set in the header, when it is valid", async () => {
-      const mockHeaders = {
-        get: jest.fn(() => {
-          return fakeDateInHeader;
-        }),
-      };
-      (headers as jest.Mock).mockResolvedValue(mockHeaders);
-
-      expect(await getNow()).toEqual(new Date(fakeDateInHeader));
+      jest.resetModules();
+      process.env = { ...ENVIRONMENT };
     });
 
-    it("should return current date when date set in the header is malformed", async () => {
-      const fakeDateInHeaderInvalid = "invalid-date";
-      const mockHeaders = {
-        get: jest.fn(() => {
-          return fakeDateInHeaderInvalid;
-        }),
-      };
-      (headers as jest.Mock).mockResolvedValue(mockHeaders);
-
-      expect(await getNow()).toEqual(new Date(fakeDateInSystem));
+    afterAll(() => {
+      process.env = ENVIRONMENT;
+    });
+
+    describe("getNow  in non-production environments", () => {
+      beforeEach(() => {
+        jest.useFakeTimers();
+        jest.setSystemTime(new Date(fakeDateInSystem));
+      });
+
+      it("should return current date by default", async () => {
+        expect(await getNow()).toEqual(new Date(fakeDateInSystem));
+      });
+
+      it("should return date set in the header, when it is valid", async () => {
+        const mockHeaders = {
+          get: jest.fn(() => {
+            return fakeDateInHeader;
+          }),
+        };
+        (headers as jest.Mock).mockResolvedValue(mockHeaders);
+
+        expect(await getNow()).toEqual(new Date(fakeDateInHeader));
+      });
+
+      it("should return current date when date set in the header is malformed", async () => {
+        const fakeDateInHeaderInvalid = "invalid-date";
+        const mockHeaders = {
+          get: jest.fn(() => {
+            return fakeDateInHeaderInvalid;
+          }),
+        };
+        (headers as jest.Mock).mockResolvedValue(mockHeaders);
+
+        expect(await getNow()).toEqual(new Date(fakeDateInSystem));
+      });
+    });
+
+    describe("getNow in production environment", () => {
+      beforeEach(() => {
+        jest.useFakeTimers();
+        jest.setSystemTime(new Date(fakeDateInSystem));
+      });
+
+      it("should return current date by default in production environment", async () => {
+        process.env.DEPLOY_ENVIRONMENT = "prod";
+        expect(await getNow()).toEqual(new Date(fakeDateInSystem));
+      });
+
+      it("should not check headers when environment is production", async () => {
+        process.env.DEPLOY_ENVIRONMENT = "prod";
+
+        const mockHeaders = {
+          get: jest.fn(() => {
+            return fakeDateInHeader;
+          }),
+        };
+        (headers as jest.Mock).mockResolvedValue(mockHeaders);
+
+        expect(await getNow()).toEqual(new Date(fakeDateInSystem));
+        expect(headers).not.toHaveBeenCalled();
+      });
     });
   });
 });

src/utils/date.ts

@@ -40,11 +40,15 @@ const calculateAge = (date: string): Age => {
 };
 
 const getNow = async (): Promise<Date> => {
-  try {
-    const headersList = await headers();
-    return UtcDateTimeFromStringSchema.safeParse(headersList.get("x-e2e-datetime")).data ?? new Date();
-  } catch {
+  if (process.env.DEPLOY_ENVIRONMENT === "prod") {
     return new Date();
+  } else {
+    try {
+      const headersList = await headers();
+      return UtcDateTimeFromStringSchema.safeParse(headersList.get("x-e2e-datetime")).data ?? new Date();
+    } catch {
+      return new Date();
+    }
   }
 };