From 7e3444384b9b80e4384db84d41dd3607381ed9c9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?S=C3=A9bastien=20Chopin?= <seb@nuxtjs.com>
Date: Tue, 2 Aug 2022 19:37:50 +0200
Subject: [PATCH] fix(document-driven): Only set 404 status on SSR (#1409)

---
 playground/document-driven/content/0.index.md |  2 ++
 src/runtime/pages/document-driven.vue         | 13 +++++--------
 test/document-driven.test.ts                  |  2 +-
 3 files changed, 8 insertions(+), 9 deletions(-)

diff --git a/playground/document-driven/content/0.index.md b/playground/document-driven/content/0.index.md
index 59ac8a874..008e5beae 100644
--- a/playground/document-driven/content/0.index.md
+++ b/playground/document-driven/content/0.index.md
@@ -7,3 +7,5 @@ description: Hello, here is a description
 # Home
 
 Hello World!
+
+[404](/404)
diff --git a/src/runtime/pages/document-driven.vue b/src/runtime/pages/document-driven.vue
index 21b1a7f66..f45d9e1a7 100644
--- a/src/runtime/pages/document-driven.vue
+++ b/src/runtime/pages/document-driven.vue
@@ -1,15 +1,12 @@
 <script setup lang="ts">
-import { createError } from 'h3'
-import { useContent, useContentHead, useRoute } from '#imports'
+import { useContent, useContentHead, useRequestEvent } from '#imports'
 
 const { page } = useContent()
 
-// Page not found
-if (!page.value) {
-  throw createError({
-    statusCode: 404,
-    statusMessage: `Page not found: ${useRoute().path}`
-  })
+// Page not found, set correct status code on SSR
+if (!page.value && process.server) {
+  const event = useRequestEvent()
+  event.res.statusCode = 404
 }
 
 useContentHead(page)
diff --git a/test/document-driven.test.ts b/test/document-driven.test.ts
index ba2b53ed0..dc611d274 100644
--- a/test/document-driven.test.ts
+++ b/test/document-driven.test.ts
@@ -52,7 +52,7 @@ describe('fixtures:document-driven', async () => {
       await $fetch('/page-not-found')
     } catch (e) {
       expect(e.response.status).toBe(404)
-      expect(e.response.statusText).toBe('Page not found: /page-not-found')
+      expect(e.response.statusText).toBe('Not Found')
     }
   })
 })