From d100ac8c825d08640634850b22d3bd013e0bcfd1 Mon Sep 17 00:00:00 2001 From: Erik Johnston Date: Wed, 20 Aug 2014 15:10:36 +0100 Subject: [PATCH] Fix test. get_joined_hosts_for_room get's called multiple times --- tests/handlers/test_room.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/tests/handlers/test_room.py b/tests/handlers/test_room.py index bfdde6135..be68f1769 100644 --- a/tests/handlers/test_room.py +++ b/tests/handlers/test_room.py @@ -140,9 +140,11 @@ class RoomMemberHandlerTestCase(unittest.TestCase): joined = ["red", "green"] self.state_handler.handle_new_event.return_value = defer.succeed(True) - self.datastore.get_joined_hosts_for_room.return_value = ( - defer.succeed(joined) - ) + + def get_joined(*args): + return defer.succeed(joined) + + self.datastore.get_joined_hosts_for_room.side_effect = get_joined store_id = "store_id_fooo" self.datastore.persist_event.return_value = defer.succeed(store_id)