diff options
Diffstat (limited to 'tests/unittest.py')
| -rw-r--r-- | tests/unittest.py | 10 |
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() |