ЕМНИП Selenium прЫнудительно закроет браузер довольно быстро по таймауту. а чтобы следать кнопку достаточно превратить тест в (скрытно) графическое приложение - на выбор зала - SWT или Swing или JavaFx - оно будет ничем не отличаться от консольного кроме вороха дополнительных зависимостей - вам пример ? i have both SWT и JAvaFx apps and Selenium tests in my git repo - обычно редко кто скрещиывет ежа с ужом...
при этом тест остановится не в произвольной точке а там где будет
(new CustomDialog(shell)).open();
