diff options
| author | Erik Johnston <erik@matrix.org> | 2019-10-31 16:17:53 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2019-10-31 16:17:53 +0100 |
| commit | dfe0cd71b61c043592cc27b64725918aa5c64724 (patch) | |
| tree | ec416471ef0ef54e0f959a9618a07e6ba9ccb64c /tests/test_state.py | |
| parent | Expose some homeserver functionality to spam checkers (#6259) (diff) | |
| parent | Apply suggestions from code review (diff) | |
| download | synapse-dfe0cd71b61c043592cc27b64725918aa5c64724.tar.xz | |
Merge pull request #6294 from matrix-org/erikj/add_state_storage
Add StateGroupStorage interface
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 |