async def test_recv_event_anyio2_end_of_stream(): class MockH11Connection: def __init__(self): self.data = None def next_event(self): if self.data == b"": return h11.PAUSED else: return h11.NEED_DATA def receive_data(self, data): self.data = data class MockSock: def receive(self): raise EndOfStream req = RequestProcessor(None, "get", "toot-toot", None) req.sock = MockSock() h11_connection = MockH11Connection() event = await req._recv_event(h11_connection) assert event is h11.PAUSED assert h11_connection.data == b""
def _catch_response(monkeypatch, headers, data): req = RequestProcessor(None, 'get', "toot-toot", None) events = [ h11._events.Response(status_code=200, headers=headers), h11._events.Data(data=data), h11._events.EndOfMessage(), ] async def _recv_event(hconn): return events.pop(0) monkeypatch.setattr(req, '_recv_event', _recv_event) monkeypatch.setattr(req, 'host', 'lol') cr = req._catch_response(None) try: cr.send(None) except StopIteration as e: response = e.value return response
def _catch_response(monkeypatch, headers, data, http_version=b"1.1"): req = RequestProcessor(None, "get", "toot-toot", None) events = [ h11._events.Response(status_code=200, headers=headers, http_version=http_version), h11._events.Data(data=data), h11._events.EndOfMessage(), ] async def _recv_event(hconn): return events.pop(0) monkeypatch.setattr(req, "_recv_event", _recv_event) monkeypatch.setattr(req, "host", "lol") cr = req._catch_response(None) try: cr.send(None) except StopIteration as e: response = e.value return response
def test_dict_to_query(data, query_str): assert RequestProcessor._dict_to_query(data) == query_str