1 file changed, 46 insertions, 50 deletions
diff --git a/tests/serve-split-check.py b/tests/serve-split-check.py
index 3a74803..8a41b5b 100644
--- a/tests/serve-split-check.py
+++ b/tests/serve-split-check.py
@@ -5,7 +5,7 @@ own headers with bare LF endings, so a body carrying a CRLF blank line, which
 any commit message written on DOS does, must not be mistaken for that end.
 Getting it wrong takes the document head into the header block, and the page
 that reaches the browser then has no stylesheet and no chrome. Run with the
-repository root as the only argument.
+repository root and the name of one case.
 """
 
 import importlib.util
@@ -24,57 +24,53 @@ def load_serve(root):
     return module
 
 
+# Each case names the raw output, then the status, headers and body it should
+# split into.
+CASES = {
+    "crlf-in-body": (
+        b"Content-Type: text/html\n\n<!DOCTYPE html>\nmsg\r\n\r\ntail",
+        200,
+        [("Content-Type", "text/html")],
+        b"<!DOCTYPE html>\nmsg\r\n\r\ntail",
+    ),
+    "lf-in-body": (
+        b"Content-Type: text/plain\r\nX: y\r\n\r\nline\n\nline2",
+        200,
+        [("Content-Type", "text/plain"), ("X", "y")],
+        b"line\n\nline2",
+    ),
+    "status": (
+        b"Status: 404 Not Found\nContent-Type: text/html\n\nnope",
+        404,
+        [("Content-Type", "text/html")],
+        b"nope",
+    ),
+    "empty-body": (
+        b"Status: 302 Found\nLocation: /x\n\n",
+        302,
+        [("Location", "/x")],
+        b"",
+    ),
+    "no-blank-line": (
+        b"Content-Type: text/html\n",
+        200,
+        [("Content-Type", "text/html")],
+        b"",
+    ),
+}
+
+
 def main():
-    split = load_serve(sys.argv[1]).split_cgi_output
-    cases = (
-        (
-            "bare LF headers, CRLF blank line inside the body",
-            b"Content-Type: text/html\n\n<!DOCTYPE html>\nmsg\r\n\r\ntail",
-            200,
-            [("Content-Type", "text/html")],
-            b"<!DOCTYPE html>\nmsg\r\n\r\ntail",
-        ),
-        (
-            "CRLF headers, LF blank line inside the body",
-            b"Content-Type: text/plain\r\nX: y\r\n\r\nline\n\nline2",
-            200,
-            [("Content-Type", "text/plain"), ("X", "y")],
-            b"line\n\nline2",
-        ),
-        (
-            "a status line is taken off the headers",
-            b"Status: 404 Not Found\nContent-Type: text/html\n\nnope",
-            404,
-            [("Content-Type", "text/html")],
-            b"nope",
-        ),
-        (
-            "an empty body survives, as a redirect leaves one",
-            b"Status: 302 Found\nLocation: /x\n\n",
-            302,
-            [("Location", "/x")],
-            b"",
-        ),
-        (
-            "output with no blank line at all is all headers",
-            b"Content-Type: text/html\n",
-            200,
-            [("Content-Type", "text/html")],
-            b"",
-        ),
+    root, name = sys.argv[1:]
+    raw, status, headers, body = CASES[name]
+    got = load_serve(root).split_cgi_output(raw)
+    if (got.status, got.headers, got.body) == (status, headers, body):
+        return 0
+    sys.stderr.write(
+        "expected %r\ngot      %r\n"
+        % ((status, headers, body), (got.status, got.headers, got.body))
     )
-
-    failed = 0
-    for name, raw, status, headers, body in cases:
-        got = split(raw)
-        if (got.status, got.headers, got.body) == (status, headers, body):
-            continue
-        failed += 1
-        sys.stderr.write(
-            "%s\n  expected %r\n  got      %r\n"
-            % (name, (status, headers, body), (got.status, got.headers, got.body))
-        )
-    return 1 if failed else 0
+    return 1
 
 
 sys.exit(main())