Skip to content

Commit e540880

Browse files
committed
test(ffe): frame mocked CDN responses
1 parent 3a530b2 commit e540880

2 files changed

Lines changed: 5 additions & 1 deletion

File tree

tests/test_the_test/test_mock_ffe_agentless_backend.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@ def test_mock_ffe_agentless_backend_serves_fixture_and_tracks_metadata(worker_id
3434
timeout=5,
3535
)
3636
response.raise_for_status()
37+
assert response.headers["Content-Length"] == str(len(response.content))
3738

3839
payload = response.json()
3940
assert payload["data"]["type"] == UFC_RESPONSE_TYPE

utils/docker_fixtures/_mock_ffe_agentless_backend.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -194,6 +194,7 @@ def _handle_config(self) -> None:
194194
self.send_response(status_code)
195195
for key, value in headers.items():
196196
self.send_header(key, value)
197+
self.send_header("Content-Length", str(len(body)))
197198
self.end_headers()
198199
if body:
199200
self.wfile.write(body)
@@ -229,10 +230,12 @@ def _handle_responses_control(self) -> None:
229230

230231
def _write_json(self, status_code: HTTPStatus, payload: dict[str, Any] | MockFFEAgentlessBackendStatus) -> None:
231232
with contextlib.suppress(BrokenPipeError, ConnectionResetError):
233+
body = json.dumps(payload).encode("utf-8")
232234
self.send_response(status_code)
233235
self.send_header("Content-Type", "application/json")
236+
self.send_header("Content-Length", str(len(body)))
234237
self.end_headers()
235-
self.wfile.write(json.dumps(payload).encode("utf-8"))
238+
self.wfile.write(body)
236239

237240

238241
def _has_auth(headers: Mapping[str, str]) -> bool:

0 commit comments

Comments
 (0)