This commit is contained in:
Erik Johnston 2014-11-04 16:35:38 +00:00
parent dd3711bdbd
commit 7f4c7fe4e8
1 changed files with 3 additions and 1 deletions

View File

@ -40,6 +40,7 @@ class DirectoryTestCase(unittest.TestCase):
]) ])
self.query_handlers = {} self.query_handlers = {}
def register_query_handler(query_type, handler): def register_query_handler(query_type, handler):
self.query_handlers[query_type] = handler self.query_handlers[query_type] = handler
self.mock_federation.register_query_handler = register_query_handler self.mock_federation.register_query_handler = register_query_handler
@ -50,7 +51,8 @@ class DirectoryTestCase(unittest.TestCase):
self.mock_config = Mock() self.mock_config = Mock()
self.mock_config.signing_key = [MockKey()] self.mock_config.signing_key = [MockKey()]
hs = HomeServer("test", hs = HomeServer(
"test",
db_pool=db_pool, db_pool=db_pool,
http_client=None, http_client=None,
resource_for_federation=Mock(), resource_for_federation=Mock(),