mirror of
https://github.com/usestrix/strix.git
synced 2026-10-16 04:28:23 +00:00
cd415e7d4d
build_raw_request only auto-adds Content-Length for a non-empty body, so replacing the body with an empty string produced a request with no Content-Length. Set it to 0 explicitly in apply_modifications (scoped to the body-replacement path, so body-less requests like GET are unaffected). Adds a test.
75 lines
2.3 KiB
Python
75 lines
2.3 KiB
Python
"""Tests for raw-request building in the Caido proxy helper."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from strix.tools.proxy.caido_api import apply_modifications, build_raw_request
|
|
|
|
|
|
def _content_length(raw: bytes) -> str | None:
|
|
for line in raw.decode("utf-8").split("\r\n"):
|
|
if line.lower().startswith("content-length:"):
|
|
return line.split(":", 1)[1].strip()
|
|
return None
|
|
|
|
|
|
def test_body_replacement_recomputes_content_length() -> None:
|
|
components = {
|
|
"method": "POST",
|
|
"headers": {"Host": "x.test", "Content-Length": "3"},
|
|
"body": "old",
|
|
}
|
|
new_body = "this is the new and much longer body!"
|
|
result = apply_modifications(components, {"body": new_body}, "http://x.test/submit")
|
|
_conn, raw = build_raw_request(
|
|
method=result["method"],
|
|
url=result["url"],
|
|
headers=result["headers"],
|
|
body=result["body"],
|
|
)
|
|
assert _content_length(raw) == str(len(new_body.encode("utf-8")))
|
|
|
|
|
|
def test_explicit_content_length_override_is_preserved() -> None:
|
|
components = {
|
|
"method": "POST",
|
|
"headers": {"Host": "x.test", "Content-Length": "3"},
|
|
"body": "old",
|
|
}
|
|
result = apply_modifications(
|
|
components,
|
|
{"body": "new body", "headers": {"Content-Length": "999"}},
|
|
"http://x.test/",
|
|
)
|
|
_conn, raw = build_raw_request(
|
|
method=result["method"],
|
|
url=result["url"],
|
|
headers=result["headers"],
|
|
body=result["body"],
|
|
)
|
|
assert _content_length(raw) == "999"
|
|
|
|
|
|
def test_no_body_modification_keeps_content_length() -> None:
|
|
components = {
|
|
"method": "POST",
|
|
"headers": {"Host": "x.test", "Content-Length": "3"},
|
|
"body": "old",
|
|
}
|
|
result = apply_modifications(components, {"headers": {"X-Test": "1"}}, "http://x.test/")
|
|
assert result["headers"].get("Content-Length") == "3"
|
|
|
|
|
|
def test_empty_body_replacement_sets_content_length_zero() -> None:
|
|
components = {
|
|
"method": "POST",
|
|
"headers": {"Host": "x.test", "Content-Length": "3"},
|
|
"body": "old",
|
|
}
|
|
result = apply_modifications(components, {"body": ""}, "http://x.test/")
|
|
_conn, raw = build_raw_request(
|
|
method=result["method"],
|
|
url=result["url"],
|
|
headers=result["headers"],
|
|
body=result["body"],
|
|
)
|
|
assert _content_length(raw) == "0"
|