Fix federation test, since we now hit store.get_room
This commit is contained in:
parent
beb0a179bd
commit
5c4c591c61
|
@ -1,4 +1,3 @@
|
||||||
# -*- 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):
|
||||||
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):
|
||||||
|
|
||||||
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)
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue