summary refs log tree commit diff
path: root/tests/storage/test_base.py
diff options
context:
space:
mode:
authorPatrick Cloke <clokep@users.noreply.github.com>2020-08-27 07:41:01 -0400
committerGitHub <noreply@github.com>2020-08-27 07:41:01 -0400
commit30426c7063a7e5567ac21cd10267651ef1935360 (patch)
treed18f000a9576bc52742301ed37793c6563914004 /tests/storage/test_base.py
parentConvert simple_update* and simple_select* to async (#8173) (diff)
downloadsynapse-30426c7063a7e5567ac21cd10267651ef1935360.tar.xz
Convert additional database methods to async (select list, search, insert_many, delete_*) (#8168)
Diffstat (limited to 'tests/storage/test_base.py')
-rw-r--r--tests/storage/test_base.py6
1 files changed, 4 insertions, 2 deletions
diff --git a/tests/storage/test_base.py b/tests/storage/test_base.py
index 64abe8cc49..40ba652248 100644
--- a/tests/storage/test_base.py
+++ b/tests/storage/test_base.py
@@ -197,8 +197,10 @@ class SQLBaseStoreTestCase(unittest.TestCase):
     def test_delete_one(self):
         self.mock_txn.rowcount = 1
 
-        yield self.datastore.db_pool.simple_delete_one(
-            table="tablename", keyvalues={"keycol": "Go away"}
+        yield defer.ensureDeferred(
+            self.datastore.db_pool.simple_delete_one(
+                table="tablename", keyvalues={"keycol": "Go away"}
+            )
         )
 
         self.mock_txn.execute.assert_called_with(