Split preview output at the first blank line

1 file changed, 76 insertions, 0 deletions

diff --git a/tests/serve-split-check.py b/tests/serve-split-check.pynew file mode 100644index 0000000..63c8f97--- /dev/null+++ b/tests/serve-split-check.py@@ -0,0 +1,76 @@+"""Check that tools/serve.py splits a CGI response at the right place.++A CGI response ends its header block at the first blank line. cgit writes its+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.+"""++import importlib.util+import sys+++def load_serve(root):+    path = root + "/tools/serve.py"+    spec = importlib.util.spec_from_file_location("serve", path)+    module = importlib.util.module_from_spec(spec)+    spec.loader.exec_module(module)+    return module+++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"",+        ),+    )++    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+++sys.exit(main())
diff --git a/tests/serve-split-check.py b/tests/serve-split-check.py
new file mode 100644
index 0000000..63c8f97
--- /dev/null
+++ b/tests/serve-split-check.py
@@ -0,0 +1,76 @@
+"""Check that tools/serve.py splits a CGI response at the right place.
+
+A CGI response ends its header block at the first blank line. cgit writes its
+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.
+"""
+
+import importlib.util
+import sys
+
+
+def load_serve(root):
+    path = root + "/tools/serve.py"
+    spec = importlib.util.spec_from_file_location("serve", path)
+    module = importlib.util.module_from_spec(spec)
+    spec.loader.exec_module(module)
+    return module
+
+
+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"",
+        ),
+    )
+
+    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
+
+
+sys.exit(main())