summary refs log tree commit diff
path: root/tests/storage
diff options
context:
space:
mode:
authorPatrick Cloke <clokep@users.noreply.github.com>2020-08-07 12:17:17 -0400
committerGitHub <noreply@github.com>2020-08-07 12:17:17 -0400
commitf3fe6961b211d898aa347771df598c531fbca90c (patch)
tree2389d40f2742f84264b3688edff377774dbfc495 /tests/storage
parentClarify that undoing a shutdown might not be possible (#8010) (diff)
downloadsynapse-f3fe6961b211d898aa347771df598c531fbca90c.tar.xz
Convert additional database stores to async/await (#8045)
Diffstat (limited to 'tests/storage')
-rw-r--r--tests/storage/test_user_directory.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/storage/test_user_directory.py b/tests/storage/test_user_directory.py
index 6a545d2eb0..ecfafe68a9 100644
--- a/tests/storage/test_user_directory.py
+++ b/tests/storage/test_user_directory.py
@@ -40,7 +40,7 @@ class UserDirectoryStoreTestCase(unittest.TestCase):
     def test_search_user_dir(self):
         # normally when alice searches the directory she should just find
         # bob because bobby doesn't share a room with her.
-        r = yield self.store.search_user_dir(ALICE, "bob", 10)
+        r = yield defer.ensureDeferred(self.store.search_user_dir(ALICE, "bob", 10))
         self.assertFalse(r["limited"])
         self.assertEqual(1, len(r["results"]))
         self.assertDictEqual(
@@ -51,7 +51,7 @@ class UserDirectoryStoreTestCase(unittest.TestCase):
     def test_search_user_dir_all_users(self):
         self.hs.config.user_directory_search_all_users = True
         try:
-            r = yield self.store.search_user_dir(ALICE, "bob", 10)
+            r = yield defer.ensureDeferred(self.store.search_user_dir(ALICE, "bob", 10))
             self.assertFalse(r["limited"])
             self.assertEqual(2, len(r["results"]))
             self.assertDictEqual(