From 0f0690b1b65acd0087156f105efaea89b7be82eb Mon Sep 17 00:00:00 2001 From: Georgiy Zatserklianyi Date: Fri, 10 Nov 2023 12:16:16 +0200 Subject: [PATCH] cookies_tests added new Echo2 endpoint --- tests/mockserver.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/tests/mockserver.py b/tests/mockserver.py index 647b0682e..cea79530c 100644 --- a/tests/mockserver.py +++ b/tests/mockserver.py @@ -200,6 +200,18 @@ class Echo(LeafResource): render_POST = render_GET +class Echo2(LeafResource): + def render_GET(self, request): + req = json.loads(request.content.read()) + + request.setResponseCode(req.get("status", 200)) + for k, v in req.get("headers").items(): + request.setHeader(bytes(k, encoding="utf8"), bytes(v, encoding="utf8")) + return bytes(req.get("body", ""), encoding="utf8") + + render_POST = render_GET + + class RedirectTo(LeafResource): def render(self, request): goto = getarg(request, b"goto", b"/") @@ -249,6 +261,7 @@ class Root(resource.Resource): self.putChild(b"drop", Drop()) self.putChild(b"raw", Raw()) self.putChild(b"echo", Echo()) + self.putChild(b"echo2", Echo2()) self.putChild(b"payload", PayloadResource()) self.putChild( b"xpayload",