add keepAlive tests for undici

test/integration/undici-keep-alive/next.config.js

@@ -0,0 +1,5 @@
+module.exports = {
+  experimental: {
+    useUndici: true,
+  },
+}

test/integration/undici-keep-alive/pages/api/json.js

@@ -0,0 +1,5 @@
+export default async function handler(_req, res) {
+  const fetchRes = await fetch('http://localhost:44001')
+  const props = await fetchRes.json()
+  res.json(props)
+}

test/integration/undici-keep-alive/pages/blog/[slug].js

@@ -0,0 +1,23 @@
+export default function Blog(props) {
+  return <pre id="props">{JSON.stringify(props)}</pre>
+}
+
+export async function getStaticProps({ params: { slug } }) {
+  return { props: { slug } }
+}
+
+export async function getStaticPaths() {
+  const res = await fetch('http://localhost:44001')
+  const obj = await res.json()
+  if (obj.connection === 'keep-alive') {
+    return {
+      paths: [{ params: { slug: 'first' } }],
+      fallback: false,
+    }
+  }
+
+  return {
+    paths: [],
+    fallback: false,
+  }
+}

test/integration/undici-keep-alive/pages/ssg.js

@@ -0,0 +1,9 @@
+export default function SSG(props) {
+  return <pre id="props">{JSON.stringify(props)}</pre>
+}
+
+export async function getStaticProps() {
+  const res = await fetch('http://localhost:44001')
+  const props = await res.json()
+  return { props }
+}

test/integration/undici-keep-alive/pages/ssr.js

@@ -0,0 +1,9 @@
+export default function SSR(props) {
+  return <pre id="props">{JSON.stringify(props)}</pre>
+}
+
+export async function getServerSideProps() {
+  const res = await fetch('http://localhost:44001')
+  const props = await res.json()
+  return { props }
+}

test/integration/undici-keep-alive/test/index.test.js

@@ -0,0 +1,99 @@
+/* eslint-env jest */
+
+import { join } from 'path'
+import { createServer } from 'http'
+import {
+  fetchViaHTTP,
+  nextBuild,
+  findPort,
+  nextStart,
+  launchApp,
+  killApp,
+} from 'next-test-utils'
+import webdriver from 'next-webdriver'
+
+const appDir = join(__dirname, '../')
+
+let appPort
+let app
+let mockServer
+
+describe('undici-keep-alive', () => {
+  describe('dev', () => {
+    beforeAll(async () => {
+      mockServer = createServer((req, res) => {
+        // we can test request headers by sending them
+        // back with the response
+        const { connection } = req.headers
+        res.end(JSON.stringify({ connection }))
+      })
+      mockServer.listen(44001)
+      appPort = await findPort()
+      app = await launchApp(appDir, appPort)
+    })
+    afterAll(async () => {
+      await killApp(app)
+      mockServer.close()
+    })
+
+    runTests()
+  })
+
+  describe('production', () => {
+    beforeAll(async () => {
+      mockServer = createServer((req, res) => {
+        // we can test request headers by sending them
+        // back with the response
+        const { connection } = req.headers
+        res.end(JSON.stringify({ connection }))
+      })
+      mockServer.listen(44001)
+      const { stdout, stderr } = await nextBuild(appDir, [], {
+        stdout: true,
+        stderr: true,
+      })
+      if (stdout) console.log(stdout)
+      if (stderr) console.error(stderr)
+      appPort = await findPort()
+      app = await nextStart(appDir, appPort)
+    })
+    afterAll(async () => {
+      await killApp(app)
+      mockServer.close()
+    })
+
+    runTests()
+  })
+
+  function runTests() {
+    it('should send keep-alive for json API', async () => {
+      const res = await fetchViaHTTP(appPort, '/api/json')
+      const obj = await res.json()
+      expect(obj).toEqual({ connection: 'keep-alive' })
+    })
+
+    it('should send keep-alive for getStaticProps', async () => {
+      const browser = await webdriver(appPort, '/ssg')
+      const props = await browser.elementById('props').text()
+      const obj = JSON.parse(props)
+      expect(obj).toEqual({ connection: 'keep-alive' })
+      await browser.close()
+    })
+
+    it('should send keep-alive for getStaticPaths', async () => {
+      const browser = await webdriver(appPort, '/blog/first')
+      const props = await browser.elementById('props').text()
+      const obj = JSON.parse(props)
+      expect(obj).toEqual({ slug: 'first' })
+      await browser.close()
+    })
+
+    it('should send keep-alive for getServerSideProps', async () => {
+      const browser = await webdriver(appPort, '/ssr')
+      const props = await browser.elementById('props').text()
+      const obj = JSON.parse(props)
+      expect(obj).toEqual({ connection: 'keep-alive' })
+      await browser.close()
+    })
+  }
+})