diff --git a/homeassistant/util/aiohttp.py b/homeassistant/util/aiohttp.py index aa1aea1abc3..765a9994a67 100644 --- a/homeassistant/util/aiohttp.py +++ b/homeassistant/util/aiohttp.py @@ -44,13 +44,13 @@ class MockRequest: self.method = method self.url = url self.status = status - self.headers: CIMultiDict[str] = CIMultiDict(headers or {}) + self.headers: CIMultiDict[str, str] = CIMultiDict(headers or {}) self.query_string = query_string or "" self._content = content self.mock_source = mock_source @property - def query(self) -> MultiDict[str]: + def query(self) -> MultiDict[str, str]: """Return a dictionary with the query variables.""" return MultiDict(parse_qsl(self.query_string, keep_blank_values=True)) @@ -68,7 +68,7 @@ class MockRequest: """Return the body as JSON.""" return json.loads(self._text) - async def post(self) -> MultiDict[str]: + async def post(self) -> MultiDict[str, str]: """Return POST parameters.""" return MultiDict(parse_qsl(self._text, keep_blank_values=True))