AssertJ JUnitSoftAssertions, как дождаться загрузки страницы ⇐ JAVA

[Anonymous]

Я использую JUnitSoftAssertions, и пока он мне нравится. Но иногда нажатие кнопки призыва к действию занимает больше времени, чем следующее утверждение, поэтому это приводит к сбою теста.
Например:

Код: Выделить всё

@Rule
public final JUnitSoftAssertions softly = new JUnitSoftAssertions();

@And("he should be on {page} page")
public void gotoPage(String page) {
// current page is A, click to go to page B
toPageBButton.click();
softly.assertThat(driver.getCurrentUrl()).as("Page B").isEqualTo(page);
softly.assertAll();
}

Утверждение не выполнено, поскольку getCurrentUrl() все еще находится на странице A.
Если я нахожусь в режиме отладки, оно проходит, потому что у меня есть точка останова.
Если я утверждаю, что кнопка на странице B отображается до isEqualTo(page), то она также проходит, но странно, что я утверждаю кнопка, прежде чем я подтвержу текущий URL
Есть ли способ дождаться действия по клику и утверждения? Я бы не хотел использовать Thread.sleep()

Подробнее здесь: https://stackoverflow.com/questions/792 ... ge-to-load