Вы можете использовать WebDriverWait
Он будет ждать, пока будет выполнено определенное условие. Ниже код будет ждать, пока элемент не станет видимым:
WebElement element = driver.findElement(By.id("some-valid-element-id"));
WebDriverWait wait = new WebDriverWait(driver, 10);
wait.until(ExpectedConditions.visibilityOf(element));