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

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

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