Как возвращать нужный экран при переходе с диалогового окна ?

[hulk_egor]

Есть стандартное окно диалога с кнопками - отмена(остались на экране), выйти(ушли с экрана)
хочу это окно вынести в отдельный класс и использовать один для всех, так как много где использую это окно, но тут проблема:
при выходе у меня могут быть переходы на разные экраны(
и что тогда возвращать? есть какое то общее решение? типа возвращаем наследника BasePage или что нибудь такое?

[bia]

• Можно передавать в метод нужный класс и возвращать уже его созданный объект (использовать Generics)
• Можно возвращать наследника BasePage, но будут видны только методы BasePage
• Можно ничего не возвращать, а просто дальше в коде теста брать нужный PageObject