summary refs log tree commit diff
path: root/tests/storage/test_directory.py
diff options
context:
space:
mode:
authorAmber Brown <hawkowl@atleastfornow.net>2018-09-04 02:21:48 +1000
committerGitHub <noreply@github.com>2018-09-04 02:21:48 +1000
commit77055dba92af80cdfdffc30fd3084ecd24902c2e (patch)
tree1d0a9c0afdecdeae970d98c77ba4c2ff8f4f39be /tests/storage/test_directory.py
parentMerge pull request #3737 from matrix-org/erikj/remove_redundant_state_func (diff)
downloadsynapse-77055dba92af80cdfdffc30fd3084ecd24902c2e.tar.xz
Fix tests on postgresql (#3740)
Diffstat (limited to 'tests/storage/test_directory.py')
-rw-r--r--tests/storage/test_directory.py3
1 files changed, 1 insertions, 2 deletions
diff --git a/tests/storage/test_directory.py b/tests/storage/test_directory.py
index b4510c1c8d..4e128e1047 100644
--- a/tests/storage/test_directory.py
+++ b/tests/storage/test_directory.py
@@ -16,7 +16,6 @@
 
 from twisted.internet import defer
 
-from synapse.storage.directory import DirectoryStore
 from synapse.types import RoomAlias, RoomID
 
 from tests import unittest
@@ -28,7 +27,7 @@ class DirectoryStoreTestCase(unittest.TestCase):
     def setUp(self):
         hs = yield setup_test_homeserver(self.addCleanup)
 
-        self.store = DirectoryStore(None, hs)
+        self.store = hs.get_datastore()
 
         self.room = RoomID.from_string("!abcde:test")
         self.alias = RoomAlias.from_string("#my-room:test")