adding test case for integration supplied context, to also show up on…

… 404 return from component
withastro · Jan 9, 2025 · 83c3bdd · 83c3bdd
1 parent e3f40ea
commit 83c3bdd
Show file tree

Hide file tree

Showing 6 changed files with 115 additions and 0 deletions.
diff --git a/packages/astro/test/custom-404-locals.test.js b/packages/astro/test/custom-404-locals.test.js
@@ -0,0 +1,59 @@
+import assert from 'node:assert/strict';
+import { after, before, describe, it } from 'node:test';
+import * as cheerio from 'cheerio';
+import { loadFixture } from './test-utils.js';
+
+describe('Custom 404 locals', () => {
+	let fixture;
+
+	before(async () => {
+		fixture = await loadFixture({
+			root: './fixtures/custom-404-locals/',
+			site: 'http://example.com',
+		});
+	});
+
+	describe('dev', () => {
+		let devServer;
+		let $;
+
+		before(async () => {
+			devServer = await fixture.startDevServer();
+		});
+
+		after(async () => {
+			await devServer.stop();
+		});
+
+		it('renders /', async () => {
+			const html = await fixture.fetch('/').then((res) => res.text());
+			$ = cheerio.load(html);
+
+			assert.equal($('h1').text(), 'Home');
+		});
+
+		it('renders 404 for /a', async () => {
+			const res = await fixture.fetch('/a');
+			assert.equal(res.status, 404);
+
+			const html = await res.text();
+			$ = cheerio.load(html);
+
+			assert.equal($('h1').text(), 'Page not found');
+			assert.equal($('p.message').text(), 'This 404 is a dynamic HTML file.');
+			assert.equal($('p.runtime').text(), 'locals');
+		});
+
+		it('renders 404 for /404-return', async () => {
+			const res = await fixture.fetch('/404-return');
+			assert.equal(res.status, 404);
+
+			const html = await res.text();
+			$ = cheerio.load(html);
+
+			assert.equal($('h1').text(), 'Page not found');
+			assert.equal($('p.message').text(), 'This 404 is a dynamic HTML file.');
+			assert.equal($('p.runtime').text(), 'locals');
+		});
+	});
+});
diff --git a/packages/astro/test/fixtures/custom-404-locals/astro.config.mjs b/packages/astro/test/fixtures/custom-404-locals/astro.config.mjs
@@ -0,0 +1,19 @@
+import { defineConfig } from 'astro/config';
+
+// https://astro.build/config
+export default defineConfig({
+	integrations: [{
+		name: 'locals-integration',
+		hooks: {
+			'astro:server:setup': ({ server }) => {
+				server.middlewares.use(async function middleware(req, res, next) {
+						const clientLocalsSymbol = Symbol.for('astro.locals');
+						Reflect.set(req, clientLocalsSymbol, {
+							runtime: 'locals'
+					});
+					next();
+				});
+			},
+		},
+	}]
+});
diff --git a/packages/astro/test/fixtures/custom-404-locals/package.json b/packages/astro/test/fixtures/custom-404-locals/package.json
@@ -0,0 +1,8 @@
+{
+  "name": "@test/custom-404-locals",
+  "version": "0.0.0",
+  "private": true,
+  "dependencies": {
+    "astro": "workspace:*"
+  }
+}
diff --git a/packages/astro/test/fixtures/custom-404-locals/src/pages/404-return.astro b/packages/astro/test/fixtures/custom-404-locals/src/pages/404-return.astro
@@ -0,0 +1,5 @@
+---
+return new Response(null, { status: 404 });
+---
+
+I should never load
diff --git a/packages/astro/test/fixtures/custom-404-locals/src/pages/404.astro b/packages/astro/test/fixtures/custom-404-locals/src/pages/404.astro
@@ -0,0 +1,13 @@
+---
+const runtime = Astro.locals.runtime
+---
+<html lang="en">
+<head>
+  <title>Not Found - Custom 404</title>
+</head>
+<body>
+  <h1>Page not found</h1>
+	<p class="message">This 404 is a dynamic HTML file.</p>
+  <p class="runtime">{runtime}</p>
+</body>
+</html>
diff --git a/packages/astro/test/fixtures/custom-404-locals/src/pages/index.astro b/packages/astro/test/fixtures/custom-404-locals/src/pages/index.astro
@@ -0,0 +1,11 @@
+---
+---
+
+<html lang="en">
+<head>
+  <title>Custom 404</title>
+</head>
+<body>
+  <h1>Home</h1>
+</body>
+</html>