test: fix assertions

framesjs · Mar 26, 2024 · d2d6e00 · d2d6e00
1 parent b1e9261
commit d2d6e00
Showing 1 changed file with 61 additions and 7 deletions.
diff --git a/packages/frames.js/src/next/getCurrentUrl.test.ts b/packages/frames.js/src/next/getCurrentUrl.test.ts
@@ -1,15 +1,26 @@
+/* eslint-disable turbo/no-undeclared-env-vars */
 import { getCurrentUrl } from "./getCurrentUrl";
 
 describe("getCurrentUrl", () => {
+  beforeEach(() => {
+    // @ts-expect-error
+    process.env.NODE_ENV = "test";
+  });
+
+  afterEach(() => {
+    delete process.env.VERCEL_URL;
+    delete process.env.APP_URL;
+  });
+
   it("works with relative url property", () => {
     expect(
       getCurrentUrl({
         url: "/test",
         headers: {
           host: "localhost",
         },
-      } as unknown as Request)
-    ).toEqual(new URL("http://localhost/test"));
+      } as unknown as Request)?.toString()
+    ).toEqual("http://localhost/test");
   });
 
   it("takes value from process.env.VERCEL_URL if available", () => {
@@ -21,13 +32,56 @@ describe("getCurrentUrl", () => {
         headers: {
           host: "localhost",
         },
-      } as unknown as Request)
-    ).toEqual(new URL("http://test.com/"));
+      } as unknown as Request)?.toString()
+    ).toEqual("http://test.com/test");
+  });
+
+  it("takes value from process.env.VERCEL_URL and uses https if NODE_ENV=production if available", () => {
+    process.env.VERCEL_URL = "test.com";
+    // @ts-expect-error
+    process.env.NODE_ENV = "production";
+
+    expect(
+      getCurrentUrl({
+        url: "/test",
+        headers: {
+          host: "localhost",
+        },
+      } as unknown as Request)?.toString()
+    ).toEqual("https://test.com/test");
+  });
+
+  it("takes value from process.env.APP_URL if available", () => {
+    process.env.APP_URL = "app.com";
+
+    expect(
+      getCurrentUrl({
+        url: "/test",
+        headers: {
+          host: "localhost",
+        },
+      } as unknown as Request)?.toString()
+    ).toEqual("http://app.com/test");
+  });
+
+  it("takes value from process.env.APP_URL and uses https if NODE_ENV=production if available", () => {
+    process.env.APP_URL = "test.com";
+    // @ts-expect-error
+    process.env.NODE_ENV = "production";
+
+    expect(
+      getCurrentUrl({
+        url: "/test",
+        headers: {
+          host: "localhost",
+        },
+      } as unknown as Request)?.toString()
+    ).toEqual("https://test.com/test");
   });
 
   it("supports proper url in Request object", () => {
-    expect(getCurrentUrl(new Request("http://localhost/test"))).toEqual(
-      new URL("http://localhost/test")
-    );
+    expect(
+      getCurrentUrl(new Request("http://localhost/test"))?.toString()
+    ).toEqual("http://localhost/test");
   });
 });