def wait_until_drained(self): self.log.info("draining disk write queue") master = self.input.servers[0] bucket = self.param("bucket", "default") ready = RebalanceHelper.wait_for_persistence(master, bucket) self.assertTrue(ready, "not all items persisted. see logs") self.log.info("disk write queue has been drained") return time.time()