From 513bebec35a60809a61143859a03f0beb597d7ad Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Primo=C5=BE=20Godec?= Date: Tue, 19 Nov 2019 17:53:52 +0100 Subject: [PATCH] Adding the event moved in send_headers --- httpx/dispatch/http2.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/httpx/dispatch/http2.py b/httpx/dispatch/http2.py index 5c6643103d..90bcb77d39 100644 --- a/httpx/dispatch/http2.py +++ b/httpx/dispatch/http2.py @@ -46,9 +46,10 @@ async def send( if not self.initialized: self.initiate_connection() - stream_id = await self.send_headers(request, timeout) - + stream_id = self.h2_state.get_next_available_stream_id() self.events[stream_id] = [] + await self.send_headers(request, stream_id, timeout) + self.timeout_flags[stream_id] = TimeoutFlag() self.window_update_received[stream_id] = self.backend.create_event() @@ -99,9 +100,9 @@ def initiate_connection(self) -> None: self.initialized = True async def send_headers( - self, request: AsyncRequest, timeout: TimeoutConfig = None - ) -> int: - stream_id = self.h2_state.get_next_available_stream_id() + self, request: AsyncRequest, stream_id: int, timeout: TimeoutConfig = None + ) -> None: + self.events[stream_id] = [] headers = [ (b":method", request.method.encode("ascii")), (b":authority", request.url.authority.encode("ascii")), @@ -119,7 +120,6 @@ async def send_headers( self.h2_state.send_headers(stream_id, headers) data_to_send = self.h2_state.data_to_send() await self.stream.write(data_to_send, timeout) - return stream_id async def send_request_data( self,