Add some type annotations

matrix-org · Dec 20, 2021 · f11fc08 · f11fc08
1 parent af654b9
commit f11fc08
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/tests/rest/client/utils.py b/tests/rest/client/utils.py
@@ -149,12 +149,12 @@ def invite(self, room=None, src=None, targ=None, expect_code=200, tok=None):
 
     def join(
         self,
-        room=None,
-        user=None,
-        expect_code=200,
-        tok=None,
+        room: str,
+        user: Optional[str] = None,
+        expect_code: int = 200,
+        tok: Optional[str] = None,
         appservice_user_id: Optional[str] = None,
-    ):
+    ) -> None:
         self.change_membership(
             room=room,
             src=user,
@@ -207,8 +207,8 @@ def leave(self, room=None, user=None, expect_code=200, tok=None):
     def change_membership(
         self,
         room: str,
-        src: str,
-        targ: str,
+        src: Optional[str],
+        targ: Optional[str],
         membership: str,
         extra_data: Optional[dict] = None,
         tok: Optional[str] = None,