fix test

vercel · Mar 30, 2022 · 6d40e6e · 6d40e6e
1 parent 00f408b
commit 6d40e6e
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/test/integration/react-streaming-and-server-components/test/index.test.js b/test/integration/react-streaming-and-server-components/test/index.test.js
@@ -241,11 +241,17 @@ const cssSuite = {
 }
 
 const documentSuite = {
-  runTests: (context) => {
-    it('should error when custom _document has getInitialProps method', async () => {
-      const res = await fetchViaHTTP(context.appPort, '/')
-      expect(res.status).toBe(500)
-    })
+  runTests: (context, env) => {
+    if (env === 'dev') {
+      it('should error when custom _document has getInitialProps method', async () => {
+        const res = await fetchViaHTTP(context.appPort, '/')
+        expect(res.status).toBe(500)
+      })
+    } else {
+      it('should failed building', async () => {
+        expect(context.code).toBe(1)
+      })
+    }
   },
   beforeAll: () => documentPage.write(documentWithGip),
   afterAll: () => documentPage.delete(),
@@ -270,7 +276,7 @@ function runSuite(suiteName, env, options) {
       options.beforeAll?.()
       if (env === 'prod') {
         context.appPort = await findPort()
-        await nextBuild(context.appDir)
+        context.code = (await nextBuild(context.appDir)).code
         context.server = await nextStart(context.appDir, context.appPort)
       }
       if (env === 'dev') {