diff --git a/tests/handlers/test_federation.py b/tests/handlers/test_federation.py
index 6acc4ebadc..029c094115 100644
--- a/tests/handlers/test_federation.py
+++ b/tests/handlers/test_federation.py
@@ -53,6 +53,8 @@ class FederationTestCase(unittest.TestCase):
"bulk_get_push_rules",
"get_current_state",
"set_push_actions_for_event_and_users",
+ "is_guest",
+ "get_state_for_events",
]),
resource_for_federation=NonCallableMock(),
http_client=NonCallableMock(spec_set=[]),
@@ -73,6 +75,8 @@ class FederationTestCase(unittest.TestCase):
self.handlers.federation_handler = FederationHandler(self.hs)
+ self.datastore.get_state_for_events.return_value = {"$a:b": {}}
+
@defer.inlineCallbacks
def test_msg(self):
pdu = FrozenEvent({
diff --git a/tests/handlers/test_room.py b/tests/handlers/test_room.py
index ff2b597124..b1c8e61522 100644
--- a/tests/handlers/test_room.py
+++ b/tests/handlers/test_room.py
@@ -47,6 +47,8 @@ class RoomMemberHandlerTestCase(unittest.TestCase):
"bulk_get_push_rules",
"get_current_state",
"set_push_actions_for_event_and_users",
+ "get_state_for_events",
+ "is_guest",
]),
resource_for_federation=NonCallableMock(),
http_client=NonCallableMock(spec_set=[]),
@@ -116,6 +118,7 @@ class RoomMemberHandlerTestCase(unittest.TestCase):
defer.succeed([])
)
self.datastore.get_current_state.return_value = {}
+ self.datastore.get_state_for_events = lambda event_ids,types: {x: {} for x in event_ids}
def annotate(_):
ctx = Mock()
@@ -198,6 +201,7 @@ class RoomMemberHandlerTestCase(unittest.TestCase):
defer.succeed([])
)
self.datastore.get_current_state.return_value = {}
+ self.datastore.get_state_for_events = lambda event_ids,types: {x: {} for x in event_ids}
def annotate(_):
ctx = Mock()
@@ -274,6 +278,7 @@ class RoomMemberHandlerTestCase(unittest.TestCase):
defer.succeed([])
)
self.datastore.get_current_state.return_value = {}
+ self.datastore.get_state_for_events = lambda event_ids,types: {x: {} for x in event_ids}
def annotate(_):
ctx = Mock()
|
