Skip to content
Snippets Groups Projects
Commit 5c4c591c authored by Erik Johnston's avatar Erik Johnston
Browse files

Fix federation test, since we now hit store.get_room

parent beb0a179
No related branches found
No related tags found
No related merge requests found
# -*- coding: utf-8 -*-
# Copyright 2014 matrix.org # Copyright 2014 matrix.org
# #
# Licensed under the Apache License, Version 2.0 (the "License"); # Licensed under the Apache License, Version 2.0 (the "License");
...@@ -41,6 +40,7 @@ class FederationTestCase(unittest.TestCase): ...@@ -41,6 +40,7 @@ class FederationTestCase(unittest.TestCase):
datastore=NonCallableMock(spec_set=[ datastore=NonCallableMock(spec_set=[
"persist_event", "persist_event",
"store_room", "store_room",
"get_room",
]), ]),
resource_for_federation=NonCallableMock(), resource_for_federation=NonCallableMock(),
http_client=NonCallableMock(spec_set=[]), http_client=NonCallableMock(spec_set=[]),
...@@ -69,6 +69,7 @@ class FederationTestCase(unittest.TestCase): ...@@ -69,6 +69,7 @@ class FederationTestCase(unittest.TestCase):
store_id = "ASD" store_id = "ASD"
self.datastore.persist_event.return_value = defer.succeed(store_id) self.datastore.persist_event.return_value = defer.succeed(store_id)
self.datastore.get_room.return_value = defer.succeed(True)
yield self.handlers.federation_handler.on_receive(event, False, False) yield self.handlers.federation_handler.on_receive(event, False, False)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment