Rebased and updated tests.

serverless-nextjs · Aug 27, 2020 · e54d915 · e54d915
1 parent 5726c23
commit e54d915
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 11 deletions.
diff --git a/...libs/lambda-at-edge/tests/default-handler/default-build-manifest-with-trailing-slash.json b/...libs/lambda-at-edge/tests/default-handler/default-build-manifest-with-trailing-slash.json
@@ -34,7 +34,8 @@
         "/customers/new": "pages/customers/new.js",
         "/api/getCustomers": "pages/api/getCustomers.js",
         "/_error": "pages/_error.js",
-        "/404": "pages/404.html"
+        "/404": "pages/404.html",
+        "/preview": "pages/preview.js"
       }
     },
     "html": {

diff --git a/packages/libs/lambda-at-edge/tests/default-handler/default-handler.test.ts b/packages/libs/lambda-at-edge/tests/default-handler/default-handler.test.ts
@@ -172,7 +172,7 @@ describe("Lambda@Edge", () => {
           await runRedirectTest(path, expectedRedirect);
         }
       );
-      
+
       it("handles preview mode", async () => {
         const event = createCloudFrontEvent({
           uri: "/preview",
@@ -187,15 +187,19 @@ describe("Lambda@Edge", () => {
           }
         });
 
-        mockPageRequire("pages/preview.js");
-        const result = await handler(event);
-        const response = result as CloudFrontResultResponse;
-        const decodedBody = new Buffer(
-          response.body as string,
-          "base64"
-        ).toString("utf8");
-        expect(decodedBody).toBe("pages/preview.js");
-        expect(response.status).toBe(200);
+        if (trailingSlash) {
+          await runRedirectTest("/preview", "/preview/");
+        } else {
+          mockPageRequire("pages/preview.js");
+          const result = await handler(event);
+          const response = result as CloudFrontResultResponse;
+          const decodedBody = new Buffer(
+            response.body as string,
+            "base64"
+          ).toString("utf8");
+          expect(decodedBody).toBe("pages/preview.js");
+          expect(response.status).toBe(200);
+        }
       });
     });