Selenium wait для загрузки?

Я пытаюсь проверить счастливый путь для фрагмента кода, на ответ которого требуется много времени, а затем начинается запись файла в поток вывода ответа, который вызывает диалоговое окно загрузки в браузерах.

Проблема в том, что в прошлом этот процесс завершался ошибкой, вызывая исключение после такого длительного объема работы. Есть ли в селене способ ожидания загрузки или аналогичный?

Я мог бы добавить Thread.sleep , но это было бы неточно и излишне замедлило бы тест run.

Что мне здесь делать?