add unit test for normalizeLocationHeader

Author: sommeeeer

packages/tests-unit/tests/core/routing/util.test.ts

@@ -17,6 +17,7 @@ import {
   getUrlParts,
   invalidateCDNOnRequest,
   isExternal,
+  normalizeLocationHeader,
   revalidateIfRequired,
   unescapeRegex,
 } from "@opennextjs/aws/core/routing/util.js";
@@ -876,3 +877,45 @@ describe("constructNextUrl", () => {
     expect(result).toBe("http://localhost/base/path");
   });
 });
+
+describe("normalizeLocationHeader", () => {
+  it("should normalize relative location header", () => {
+    const result = normalizeLocationHeader(
+      "http://localhost:3000/path",
+      "http://localhost:3000",
+    );
+    expect(result).toBe("/path");
+  });
+
+  it("should not change absolute location header", () => {
+    const result = normalizeLocationHeader(
+      "https://opennext.js.org/aws",
+      "http://localhost:3000",
+    );
+    expect(result).toBe("https://opennext.js.org/aws");
+  });
+
+  it("should normalize relative location with query parameters", () => {
+    const result = normalizeLocationHeader(
+      "http://localhost:3000/path?query=1",
+      "http://localhost:3000",
+    );
+    expect(result).toBe("/path?query=1");
+  });
+
+  it("should normalize and encode special characters in location header", () => {
+    const result = normalizeLocationHeader(
+      "http://localhost:3000/æøå?query=æøå!&",
+      "http://localhost:3000",
+    );
+    expect(result).toBe("/%C3%A6%C3%B8%C3%A5?query=%C3%A6%C3%B8%C3%A5%21");
+  });
+
+  it("should normalize and respect already encoded characthers in location header", () => {
+    const result = normalizeLocationHeader(
+      "http://localhost:3000/path?query=test%2F%2F",
+      "http://localhost:3000",
+    );
+    expect(result).toBe("/path?query=test%2F%2F");
+  });
+});