1 files changed, 6 insertions, 4 deletions
diff --git a/tests/unittest.py b/tests/unittest.py
index eea0903f05..165aafc574 100644
--- a/tests/unittest.py
+++ b/tests/unittest.py
@@ -331,16 +331,17 @@ class HomeserverTestCase(TestCase):
             time.sleep(0.01)
 
     def wait_for_background_updates(self) -> None:
-        """Block until all background database updates have completed.
+        """
+        Block until all background database updates have completed.
 
-        Note that callers must ensure there's a store property created on the
+        Note that callers must ensure that's a store property created on the
         testcase.
         """
         while not self.get_success(
             self.store.db_pool.updates.has_completed_background_updates()
         ):
             self.get_success(
-                self.store.db_pool.updates.do_next_background_update(False), by=0.1
+                self.store.db_pool.updates.do_next_background_update(100), by=0.1
             )
 
     def make_homeserver(self, reactor, clock):
@@ -499,7 +500,8 @@ class HomeserverTestCase(TestCase):
 
         async def run_bg_updates():
             with LoggingContext("run_bg_updates"):
-                self.get_success(stor.db_pool.updates.run_background_updates(False))
+                while not await stor.db_pool.updates.has_completed_background_updates():
+                    await stor.db_pool.updates.do_next_background_update(1)
 
         hs = setup_test_homeserver(self.addCleanup, *args, **kwargs)
         stor = hs.get_datastore()