Add missing type hints for tests.unittest. (#13397)

2022-07-27 13:18:41 -04:00
parent 502f075e96
commit 922b771337
6 changed files with 66 additions and 52 deletions
--- a/tests/server.py
+++ b/tests/server.py
@@ -25,6 +25,7 @@ from typing import (
    Callable,
    Dict,
    Iterable,
+    List,
    MutableMapping,
    Optional,
    Tuple,
@@ -121,7 +122,15 @@ class FakeChannel:

    @property
    def json_body(self) -> JsonDict:
-        return json.loads(self.text_body)
+        body = json.loads(self.text_body)
+        assert isinstance(body, dict)
+        return body
+
+    @property
+    def json_list(self) -> List[JsonDict]:
+        body = json.loads(self.text_body)
+        assert isinstance(body, list)
+        return body

    @property
    def text_body(self) -> str: