summary refs log tree commit diff
path: root/tests/test_state.py
diff options
context:
space:
mode:
authorErik Johnston <erik@matrix.org>2014-08-15 17:42:21 +0100
committerErik Johnston <erik@matrix.org>2014-08-15 17:42:21 +0100
commitf5fca6f78789e1f9711924fd4dad29c8c12b692f (patch)
treec1afe59e8702670a2a58088383ca828a24e14d60 /tests/test_state.py
parentMerge branch 'master' of github.com:matrix-org/synapse into sql_refactor (diff)
downloadsynapse-f5fca6f78789e1f9711924fd4dad29c8c12b692f.tar.xz
Fix some of the tests to reflect changes in the storage layer.
Diffstat (limited to 'tests/test_state.py')
-rw-r--r--tests/test_state.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/test_state.py b/tests/test_state.py
index aaf873a856..e64d15a3a2 100644
--- a/tests/test_state.py
+++ b/tests/test_state.py
@@ -36,7 +36,7 @@ class StateTestCase(unittest.TestCase):
             "get_unresolved_state_tree",
             "update_current_state",
             "get_latest_pdus_in_context",
-            "get_current_state",
+            "get_current_state_pdu",
             "get_pdu",
         ])
         self.replication = Mock(spec=["get_pdu"])
@@ -247,7 +247,7 @@ class StateTestCase(unittest.TestCase):
         pdus = [tup]
 
         self.persistence.get_latest_pdus_in_context.return_value = pdus
-        self.persistence.get_current_state.return_value = state_pdu
+        self.persistence.get_current_state_pdu.return_value = state_pdu
 
         yield self.state.handle_new_event(event)