Fix a test which assumes call is a namedtuple.

matrix-org · Apr 9, 2021 · 444e4a2 · 444e4a2
1 parent a7d8937
commit 444e4a2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/tests/events/test_presence_router.py b/tests/events/test_presence_router.py
@@ -313,7 +313,8 @@ def test_send_local_online_presence_to_with_module(self):
             self.hs.get_federation_transport_client().send_transaction.call_args_list
         )
         for call in calls:
-            federation_transaction = call.args[0]  # type: Transaction
+            call_args = call[0]
+            federation_transaction = call_args[0]  # type: Transaction
 
             # Get the sent EDUs in this transaction
             edus = federation_transaction.get_dict()["edus"]

diff --git a/tests/module_api/test_api.py b/tests/module_api/test_api.py
@@ -358,7 +358,8 @@ def test_send_local_online_presence_to_federation(self):
             self.hs.get_federation_transport_client().send_transaction.call_args_list
         )
         for call in calls:
-            federation_transaction = call.args[0]  # type: Transaction
+            call_args = call[0]
+            federation_transaction = call_args[0]  # type: Transaction
 
             # Get the sent EDUs in this transaction
             edus = federation_transaction.get_dict()["edus"]