test: add tests for getRequestURL

unjs · May 15, 2023 · d510483 · d510483
1 parent b7126b8
commit d510483
Showing 1 changed file with 41 additions and 0 deletions.
diff --git a/test/utils.test.ts b/test/utils.test.ts
@@ -10,6 +10,7 @@ import {
   eventHandler,
   getMethod,
   getQuery,
+  getRequestURL,
 } from "../src";
 
 describe("", () => {
@@ -93,6 +94,46 @@ describe("", () => {
     });
   });
 
+  describe("getRequestURL", () => {
+    const tests = [
+      { path: "/foo", url: "http://127.0.0.1/foo" },
+      { path: "/test", host: "example.com", url: "http://example.com/test" },
+      {
+        path: "/test",
+        headers: [["x-forwarded-proto", "https"]],
+        url: "https://127.0.0.1:80/test",
+      },
+      {
+        path: "/test",
+        headers: [["x-forwarded-host", "example.com"]],
+        url: "http://example.com/test",
+      },
+    ];
+    for (const test of tests) {
+      it("getRequestURL: " + JSON.stringify(test), async () => {
+        app.use(
+          "/",
+          eventHandler((event) => {
+            const url = getRequestURL(event);
+            // @ts-ignore
+            url.port = 80;
+            return url;
+          })
+        );
+        const req = request.get(test.path);
+        if (test.host) {
+          req.set("Host", test.host);
+        }
+        if (test.headers) {
+          for (const header of test.headers) {
+            req.set(header[0], header[1]);
+          }
+        }
+        expect((await req).text).toBe(JSON.stringify(test.url));
+      });
+    }
+  });
+
   describe("assertMethod", () => {
     it("only allow head and post", async () => {
       app.use(