Fix test. get_joined_hosts_for_room get's called multiple times

2014-08-20 15:10:36 +01:00 · 2014-08-20 15:10:36 +01:00 · d100ac8c82
parent 5c4c591c61
commit d100ac8c82
1 changed files with 5 additions and 3 deletions
--- 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)