Fixup tests

This commit is contained in:
Erik Johnston 2019-12-06 16:02:50 +00:00
parent 75f87450d8
commit 852f80d8a6
6 changed files with 20 additions and 23 deletions

View file

@ -15,8 +15,6 @@
from twisted.internet import defer
from synapse.storage.data_stores.main.user_directory import UserDirectoryStore
from tests import unittest
from tests.utils import setup_test_homeserver
@ -29,7 +27,7 @@ class UserDirectoryStoreTestCase(unittest.TestCase):
@defer.inlineCallbacks
def setUp(self):
self.hs = yield setup_test_homeserver(self.addCleanup)
self.store = UserDirectoryStore(self.hs.get_db_conn(), self.hs)
self.store = self.hs.get_datastore()
# alice and bob are both in !room_id. bobby is not but shares
# a homeserver with alice.