first commit

test/integration/draft-mode/pages/another.tsx

@@ -0,0 +1,32 @@
+import Link from 'next/link'
+import type { GetStaticProps } from 'next'
+
+type Props = {
+  random: number
+  draftMode: string
+}
+
+export const getStaticProps: GetStaticProps<Props> = ({ draftMode }) => {
+  return {
+    props: {
+      random: Math.random(),
+      draftMode: Boolean(draftMode).toString(),
+    },
+    revalidate: 100000,
+  }
+}
+
+export default function Another(props: Props) {
+  return (
+    <>
+      <h1>Another</h1>
+      <p>
+        Draft Mode: <em id="draft">{props.draftMode}</em>
+      </p>
+      <p>
+        Random: <em id="rand">{props.random}</em>
+      </p>
+      <Link href="/">Go home</Link>
+    </>
+  )
+}

test/integration/draft-mode/pages/api/disable.ts

@@ -0,0 +1,6 @@
+import type { NextApiRequest, NextApiResponse } from 'next'
+
+export default function handler(_req: NextApiRequest, res: NextApiResponse) {
+  res.setDraftMode({ enable: false })
+  res.end('Check your cookies...')
+}

test/integration/draft-mode/pages/api/enable.ts

@@ -0,0 +1,6 @@
+import type { NextApiRequest, NextApiResponse } from 'next'
+
+export default function handler(_req: NextApiRequest, res: NextApiResponse) {
+  res.setDraftMode({ enable: true })
+  res.end('Check your cookies...')
+}

test/integration/draft-mode/pages/api/read.ts

@@ -0,0 +1,6 @@
+import type { NextApiRequest, NextApiResponse } from 'next'
+
+export default (req: NextApiRequest, res: NextApiResponse) => {
+  const { draftMode } = req
+  res.json({ draftMode })
+}

test/integration/draft-mode/pages/index.tsx

@@ -0,0 +1,40 @@
+import { useState } from 'react'
+import Link from 'next/link'
+import type { GetStaticProps } from 'next'
+
+type Props = {
+  random: number
+  draftMode: string
+}
+
+export const getStaticProps: GetStaticProps<Props> = ({ draftMode }) => {
+  return {
+    props: {
+      random: Math.random(),
+      draftMode: Boolean(draftMode).toString(),
+    },
+    revalidate: 100000,
+  }
+}
+
+export default function Home(props: Props) {
+  const [count, setCount] = useState(0)
+  return (
+    <>
+      <h1>Home</h1>
+      <p>
+        Draft Mode: <em id="draft">{props.draftMode}</em>
+      </p>
+      <button id="inc" onClick={() => setCount(count + 1)}>
+        Increment
+      </button>
+      <p>
+        Count: <span id="count">{count}</span>
+      </p>
+      <p>
+        Random: <em id="rand">{props.random}</em>
+      </p>
+      <Link href="/another">Visit another page</Link>
+    </>
+  )
+}

test/integration/draft-mode/pages/ssp.tsx

@@ -0,0 +1,36 @@
+import Link from 'next/link'
+import type { GetServerSideProps } from 'next'
+
+type Props = {
+  random: number
+  draftMode: string
+}
+
+export const getServerSideProps: GetServerSideProps<Props> = async ({
+  res,
+  draftMode,
+}) => {
+  // test override header
+  res.setHeader('Cache-Control', 'public, max-age=3600')
+  return {
+    props: {
+      random: Math.random(),
+      draftMode: Boolean(draftMode).toString(),
+    },
+  }
+}
+
+export default function SSP(props: Props) {
+  return (
+    <>
+      <h1>Server Side Props</h1>
+      <p>
+        Draft Mode: <em id="draft">{props.draftMode}</em>
+      </p>
+      <p>
+        Random: <em id="rand">{props.random}</em>
+      </p>
+      <Link href="/">Go home</Link>
+    </>
+  )
+}

test/integration/draft-mode/pages/to-index.tsx

@@ -0,0 +1,15 @@
+import Link from 'next/link'
+
+export function getStaticProps() {
+  return { props: {} }
+}
+
+export default function () {
+  return (
+    <main>
+      <Link href="/" id="to-index">
+        To Index
+      </Link>
+    </main>
+  )
+}