mirror of
https://github.com/kuhyx/screen-locker.git
synced 2026-11-11 16:53:58 +01:00
Krzysztof kuhy Rudnicki
aa3e290391
Real one-line docstrings stating what each test pins down. Files that would pass 250 lines moved whole classes into _part2 siblings (sick_panel, earned, poke_server, shutdown_reading, manual_workout_dialog). Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01WzxdWYdwRAtMg7ki6TKxuU
86 lines
3.1 KiB
Python
86 lines
3.1 KiB
Python
"""Poke HTTP routing and framing: paths, methods, Content-Length and body limits."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from http.client import HTTPConnection
|
|
import json
|
|
from typing import TYPE_CHECKING
|
|
|
|
import pytest
|
|
|
|
from screen_locker import _poke_server
|
|
from screen_locker._poke_server import MAX_BODY_BYTES, POKE_PATH
|
|
from screen_locker.tests.test_poke_server import (
|
|
_body,
|
|
_post,
|
|
_raw,
|
|
_sign,
|
|
credit_fixture,
|
|
port_fixture,
|
|
)
|
|
|
|
if TYPE_CHECKING:
|
|
from unittest.mock import MagicMock
|
|
|
|
__all__ = ["credit_fixture", "port_fixture"]
|
|
|
|
|
|
class TestRouting:
|
|
"""Path, method and Content-Length handling before any body is judged."""
|
|
|
|
def test_unknown_path_is_404(self, port: int, credit: MagicMock) -> None:
|
|
"""A POST to an unknown path is a 404 naming the path."""
|
|
body = _body()
|
|
status, reply = _post(port, body, _sign(body), path="/v2/other")
|
|
assert status == 404
|
|
assert "unknown path" in reply["reason"]
|
|
|
|
def test_get_is_405(self, port: int) -> None:
|
|
"""A GET is refused with 405 saying only POST is accepted."""
|
|
conn = HTTPConnection("127.0.0.1", port, timeout=10)
|
|
conn.request("GET", POKE_PATH)
|
|
resp = conn.getresponse()
|
|
data = json.loads(resp.read())
|
|
conn.close()
|
|
assert resp.status == 405
|
|
assert "only POST" in data["reason"]
|
|
|
|
def test_missing_content_length_is_411(self, port: int) -> None:
|
|
"""A POST without Content-Length is a 411."""
|
|
status, _ = _raw(port, ["Host: x"])
|
|
assert status == 411
|
|
|
|
@pytest.mark.parametrize("value", ["abc", "-5", ""])
|
|
def test_malformed_content_length_is_400(self, port: int, value: str) -> None:
|
|
"""A non-numeric, negative or empty Content-Length is a 400."""
|
|
status, reply = _raw(port, [f"Content-Length: {value}"])
|
|
assert status == 400
|
|
assert "bad Content-Length" in reply["reason"]
|
|
|
|
def test_oversized_body_is_413_without_reading_it(self, port: int) -> None:
|
|
"""A declared body over the cap is refused with 413 before reading it."""
|
|
status, reply = _raw(port, [f"Content-Length: {MAX_BODY_BYTES + 1}"])
|
|
assert status == 413
|
|
assert str(MAX_BODY_BYTES) in reply["reason"]
|
|
|
|
def test_body_at_the_cap_is_read_and_judged_on_its_signature(
|
|
self, port: int
|
|
) -> None:
|
|
"""A body exactly at the cap is read and then judged on content, not size."""
|
|
body = b"x" * MAX_BODY_BYTES
|
|
assert _post(port, body, _sign(body))[0] == 400
|
|
|
|
def test_truncated_body_is_400(self, port: int) -> None:
|
|
"""A body shorter than its Content-Length is a 400 reporting the byte counts."""
|
|
status, reply = _raw(port, ["Content-Length: 100"], send=b"0123456789")
|
|
assert status == 400
|
|
assert "10 of 100 bytes" in reply["reason"]
|
|
|
|
|
|
def test_reply_for_a_refusal_leaves_the_snapshot_null() -> None:
|
|
"""A refusal reply carries no shutdown or gaming-budget snapshot."""
|
|
reply = _poke_server._refused("nope")
|
|
assert reply["ok"] is False
|
|
assert (reply["shutdown"], reply["gaming_budget_minutes"]) == (None, None)
|
|
assert reply["reason"] == "nope"
|