refactor: Fix Response.headers type

schemathesis · Apr 26, 2020 · 88649fe · 88649fe
1 parent b13f6c6
commit 88649fe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/schemathesis/models.py b/src/schemathesis/models.py
@@ -318,7 +318,7 @@ class Response:
 
     status_code: int = attr.ib()  # pragma: no mutate
     message: str = attr.ib()  # pragma: no mutate
-    headers: Dict[str, str] = attr.ib()  # pragma: no mutate
+    headers: Dict[str, List[str]] = attr.ib()  # pragma: no mutate
     body: str = attr.ib()  # pragma: no mutate
     encoding: str = attr.ib()  # pragma: no mutate
     http_version: str = attr.ib()  # pragma: no mutate
@@ -355,7 +355,7 @@ def from_wsgi(cls, response: WSGIResponse, elapsed: float) -> "Response":
             elapsed=elapsed,
         )
 
-    def asdict(self) -> Dict[str, Union[str, Dict[str, str]]]:
+    def asdict(self) -> Dict[str, Union[str, Dict[str, str], Dict[str, List[str]]]]:
         return {
             "status": {"code": str(self.status_code), "message": self.message},
             "headers": self.headers,