diff options
| author | Erik Johnston <erik@matrix.org> | 2019-10-23 17:25:54 +0100 |
|---|---|---|
| committer | Erik Johnston <erik@matrix.org> | 2019-10-30 14:46:54 +0000 |
| commit | 69f0054ce675bd9d35104c39af9fae9a908b7f33 (patch) | |
| tree | 40dc037ff3bd919937cc5cb2deba317d3df40421 /tests/test_state.py | |
| parent | Add StateGroupStorage interface (diff) | |
| download | synapse-69f0054ce675bd9d35104c39af9fae9a908b7f33.tar.xz | |
Port to use state storage
Diffstat (limited to 'tests/test_state.py')
| -rw-r--r-- | tests/test_state.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tests/test_state.py b/tests/test_state.py index 610ec9fb46..38246555bd 100644 --- a/tests/test_state.py +++ b/tests/test_state.py @@ -158,10 +158,12 @@ class Graph(object): class StateTestCase(unittest.TestCase): def setUp(self): self.store = StateGroupStore() + storage = Mock(main=self.store, state=self.store) hs = Mock( spec_set=[ "config", "get_datastore", + "get_storage", "get_auth", "get_state_handler", "get_clock", @@ -174,6 +176,7 @@ class StateTestCase(unittest.TestCase): hs.get_clock.return_value = MockClock() hs.get_auth.return_value = Auth(hs) hs.get_state_resolution_handler = lambda: StateResolutionHandler(hs) + hs.get_storage.return_value = storage self.state = StateHandler(hs) self.event_id = 0 |