diff --git a/tests/storage/test_background_update.py b/tests/storage/test_background_update.py
index 3f5bfa09d4..67ea640902 100644
--- a/tests/storage/test_background_update.py
+++ b/tests/storage/test_background_update.py
@@ -457,8 +457,8 @@ class BackgroundUpdateValidateConstraintTestCase(unittest.HomeserverTestCase):
);
"""
self.get_success(
- self.store.db_pool.execute(
- "test_not_null_constraint", lambda _: None, table_sql
+ self.store.db_pool.runInteraction(
+ "test_not_null_constraint", lambda txn: txn.execute(table_sql)
)
)
@@ -466,8 +466,8 @@ class BackgroundUpdateValidateConstraintTestCase(unittest.HomeserverTestCase):
# using SQLite.
index_sql = "CREATE INDEX test_index ON test_constraint(a)"
self.get_success(
- self.store.db_pool.execute(
- "test_not_null_constraint", lambda _: None, index_sql
+ self.store.db_pool.runInteraction(
+ "test_not_null_constraint", lambda txn: txn.execute(index_sql)
)
)
@@ -574,13 +574,13 @@ class BackgroundUpdateValidateConstraintTestCase(unittest.HomeserverTestCase):
);
"""
self.get_success(
- self.store.db_pool.execute(
- "test_foreign_key_constraint", lambda _: None, base_sql
+ self.store.db_pool.runInteraction(
+ "test_foreign_key_constraint", lambda txn: txn.execute(base_sql)
)
)
self.get_success(
- self.store.db_pool.execute(
- "test_foreign_key_constraint", lambda _: None, table_sql
+ self.store.db_pool.runInteraction(
+ "test_foreign_key_constraint", lambda txn: txn.execute(table_sql)
)
)
diff --git a/tests/storage/test_profile.py b/tests/storage/test_profile.py
index 95f99f4130..6afb5403bd 100644
--- a/tests/storage/test_profile.py
+++ b/tests/storage/test_profile.py
@@ -120,7 +120,7 @@ class ProfileStoreTestCase(unittest.HomeserverTestCase):
res = self.get_success(
self.store.db_pool.execute(
- "", None, "SELECT full_user_id from profiles ORDER BY full_user_id"
+ "", "SELECT full_user_id from profiles ORDER BY full_user_id"
)
)
self.assertEqual(len(res), len(expected_values))
diff --git a/tests/storage/test_user_filters.py b/tests/storage/test_user_filters.py
index d4637d9d1e..2da6a018e8 100644
--- a/tests/storage/test_user_filters.py
+++ b/tests/storage/test_user_filters.py
@@ -87,7 +87,7 @@ class UserFiltersStoreTestCase(unittest.HomeserverTestCase):
res = self.get_success(
self.store.db_pool.execute(
- "", None, "SELECT full_user_id from user_filters ORDER BY full_user_id"
+ "", "SELECT full_user_id from user_filters ORDER BY full_user_id"
)
)
self.assertEqual(len(res), len(expected_values))
|