Estou tentando verificar se a página da web está carregada concluída ou não (ou seja, verificando se todo o controle está carregado) no selênio.

Eu tentei abaixo do código:

new WebDriverWait(firefoxDriver, pageLoadTimeout).until(
          webDriver -> ((JavascriptExecutor) webDriver).executeScript("return document.readyState").equals("complete"));

mas mesmo se a página estiver carregando acima do código, não espere.

Eu sei que posso verificar se há um elemento específico para verificar se é visível / clicável etc., mas estou procurando alguma solução genérica