diff options
| author | Erik Johnston <erik@matrix.org> | 2019-10-23 12:02:36 +0100 |
|---|---|---|
| committer | Erik Johnston <erik@matrix.org> | 2019-10-23 16:15:03 +0100 |
| commit | 3ca4c7c516a349cbb9dace0ace33bb889955eda1 (patch) | |
| tree | 7cd54d5d59316ef65d5511dfb3a7a59827524363 /tests/utils.py | |
| parent | Add DataStores and Storage classes. (diff) | |
| download | synapse-3ca4c7c516a349cbb9dace0ace33bb889955eda1.tar.xz | |
Use new EventPersistenceStore
Diffstat (limited to 'tests/utils.py')
| -rw-r--r-- | tests/utils.py | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/tests/utils.py b/tests/utils.py index 0a64f75d04..2808eea933 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -326,10 +326,16 @@ def setup_test_homeserver( if homeserverToUse.__name__ == "TestHomeServer": hs.setup_master() else: + # If we have been given an explicit datastore we probably want to mock + # out the DataStores somehow too. This all feels a bit wrong, but then + # mocking the stores feels wrong too. + datastores = Mock(datastore=datastore) + hs = homeserverToUse( name, db_pool=None, datastore=datastore, + datastores=datastores, config=config, version_string="Synapse/tests", database_engine=db_engine, @@ -647,7 +653,7 @@ def create_room(hs, room_id, creator_id): creator_id (str) """ - store = hs.get_datastore() + persistence_store = hs.get_storage().persistence event_builder_factory = hs.get_event_builder_factory() event_creation_handler = hs.get_event_creation_handler() @@ -664,4 +670,4 @@ def create_room(hs, room_id, creator_id): event, context = yield event_creation_handler.create_new_client_event(builder) - yield store.persist_event(event, context) + yield persistence_store.persist_event(event, context) |