summary refs log tree commit diff
path: root/tests/handlers
diff options
context:
space:
mode:
authorDavid Baker <dave@matrix.org>2016-01-06 16:46:41 +0000
committerDavid Baker <dave@matrix.org>2016-01-06 16:46:41 +0000
commit0e48f7f2458f08341131b3b90c78b7034fe02d14 (patch)
tree0e52de81fd0882623a3353f939e12b1d06c83780 /tests/handlers
parentDelete notifications for redacted events (diff)
downloadsynapse-0e48f7f2458f08341131b3b90c78b7034fe02d14.tar.xz
fix tests
Diffstat (limited to 'tests/handlers')
-rw-r--r--tests/handlers/test_federation.py4
-rw-r--r--tests/handlers/test_room.py5
2 files changed, 9 insertions, 0 deletions
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()