server: adjust indentation, add 404 to "custom server" test case

Krinkle · Krinkle · commit 511900262f9d · 2025-04-20T21:45:17.000+01:00
Follows-up 4f73f34.
diff --git a/src/server.js b/src/server.js
@@ -432,21 +432,23 @@ class ControlServer {
     //   transparent and correct.
     // * Ignore <heading> and <head-thing>.
     // * Support <head x=y...>, including with tabs or newlines before ">".
-    html = util.replaceOnce(html, [
-      /<head(?:\s[^>]*)?>/i,
-      /<html(?:\s[^>]*)?>/i,
-      /<!doctype[^>]*>/i,
-      /^/
-    ],
-    (m) => m + headInjectHtml
+    html = util.replaceOnce(html,
+      [
+        /<head(?:\s[^>]*)?>/i,
+        /<html(?:\s[^>]*)?>/i,
+        /<!doctype[^>]*>/i,
+        /^/
+      ],
+      (m) => m + headInjectHtml
     );
 
-    html = util.replaceOnce(html, [
-      /<\/body>(?![\s\S]*<\/body>)/i,
-      /<\/html>(?![\s\S]*<\/html>)/i,
-      /$/
-    ],
-    (m) => '<script>(' + util.fnToStr(qtapClientBody, qtapTapUrl) + ')();</script>' + m
+    html = util.replaceOnce(html,
+      [
+        /<\/body>(?![\s\S]*<\/body>)/i,
+        /<\/html>(?![\s\S]*<\/html>)/i,
+        /$/
+      ],
+      (m) => '<script>(' + util.fnToStr(qtapClientBody, qtapTapUrl) + ')();</script>' + m
     );
 
     return {
diff --git a/test/qtap.test.js b/test/qtap.test.js
@@ -506,11 +506,16 @@ QUnit.module('qtap', function (hooks) {
     assert.timeout(40_000);
 
     const server = http.createServer((req, resp) => {
-      resp.writeHead(200);
-      resp.end(
-        '# console: Origin server URL is ' + req.url + '\n'
-          + 'TAP version 13\nok 1 Foo\nok 2 Bar\n1..2\n'
-      );
+      if (req.url.startsWith('/test/example.html')) {
+        resp.writeHead(200);
+        resp.end(
+          '# console: Origin server URL is ' + req.url + '\n'
+            + 'TAP version 13\nok 1 Foo\nok 2 Bar\n1..2\n'
+        );
+      } else {
+        resp.writeHead(404);
+        resp.end('Not Found\n');
+      }
     });
     server.listen();
     const port = await new Promise((resolve) => {
