Fix case in `wait_for_background_updates` where `self.store` does not exist (#11331)
Pull the DataStore from the HomeServer instance, which always exists.
This commit is contained in:
parent
26b5d2320f
commit
d6fb96e056
|
@ -0,0 +1 @@
|
|||
A test helper (`wait_for_background_updates`) no longer depends on classes defining a `store` property.
|
|
@ -331,16 +331,13 @@ class HomeserverTestCase(TestCase):
|
|||
time.sleep(0.01)
|
||||
|
||||
def wait_for_background_updates(self) -> None:
|
||||
"""Block until all background database updates have completed.
|
||||
|
||||
Note that callers must ensure there's a store property created on the
|
||||
testcase.
|
||||
"""
|
||||
"""Block until all background database updates have completed."""
|
||||
store = self.hs.get_datastore()
|
||||
while not self.get_success(
|
||||
self.store.db_pool.updates.has_completed_background_updates()
|
||||
store.db_pool.updates.has_completed_background_updates()
|
||||
):
|
||||
self.get_success(
|
||||
self.store.db_pool.updates.do_next_background_update(False), by=0.1
|
||||
store.db_pool.updates.do_next_background_update(False), by=0.1
|
||||
)
|
||||
|
||||
def make_homeserver(self, reactor, clock):
|
||||
|
|
Loading…
Reference in New Issue