Fixed TestClient error when response headers missing (#1200)

* Fixed abersheeran/asgi-ratelimit#14 * lint it * Black it Co-authored-by: euri10 <[email protected]>
dudleyhunt86 · Jun 21, 2021 · c1267f6 · c1267f6
1 parent bc963f7
commit c1267f6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/starlette/testclient.py b/starlette/testclient.py
@@ -214,7 +214,8 @@ async def send(message: Message) -> None:
                 raw_kwargs["status"] = message["status"]
                 raw_kwargs["reason"] = _get_reason_phrase(message["status"])
                 raw_kwargs["headers"] = [
-                    (key.decode(), value.decode()) for key, value in message["headers"]
+                    (key.decode(), value.decode())
+                    for key, value in message.get("headers", [])
                 ]
                 raw_kwargs["preload_content"] = False
                 raw_kwargs["original_response"] = _MockOriginalResponse(