summary refs log tree commit diff
diff options
context:
space:
mode:
authorErik Johnston <erik@matrix.org>2015-02-09 14:23:57 +0000
committerErik Johnston <erik@matrix.org>2015-02-09 14:23:57 +0000
commit164f6b9256782a5040c1a770c0758a31cf88e6e3 (patch)
treedbd54ec82d195ea542d61a9c24f64afc544101d6
parentMerge branch 'develop' of github.com:matrix-org/synapse into state-chache (diff)
downloadsynapse-164f6b9256782a5040c1a770c0758a31cf88e6e3.tar.xz
Fix tests
-rw-r--r--tests/test_state.py7
1 files changed, 6 insertions, 1 deletions
diff --git a/tests/test_state.py b/tests/test_state.py
index 019e794aa2..fea25f7021 100644
--- a/tests/test_state.py
+++ b/tests/test_state.py
@@ -21,6 +21,8 @@ from synapse.api.auth import Auth
 from synapse.api.constants import EventTypes, Membership
 from synapse.state import StateHandler
 
+from .utils import MockClock
+
 from mock import Mock
 
 
@@ -138,10 +140,13 @@ class StateTestCase(unittest.TestCase):
                 "add_event_hashes",
             ]
         )
-        hs = Mock(spec=["get_datastore", "get_auth", "get_state_handler"])
+        hs = Mock(spec=[
+            "get_datastore", "get_auth", "get_state_handler", "get_clock",
+        ])
         hs.get_datastore.return_value = self.store
         hs.get_state_handler.return_value = None
         hs.get_auth.return_value = Auth(hs)
+        hs.get_clock.return_value = MockClock()
 
         self.state = StateHandler(hs)
         self.event_id = 0