Правильное условие для закрытия Pop-up и переход к следующим шагам в тесте?

webdriver
, ,
#1

Есть Pop-up который всплывает при входе в учетную запись, если заполнить в нем поля он перестает в дальнейшем появляться, собственно вопрос как написать проверку если он всплыл то закрыть его и перейти к следующим шагам, если его нет пропускать шаг с его закрытием.

Пока выходит только пропускать шаг его закрытия если его нет, как написать условие если он есть закрыть его и перейти к следующим шагам.

    def tracker(self):
        try:
            pop_up = self.element_is_present(self.locators.CLOSE_EDIT_PROFILE)
            pop_up.click()
        except:
            self.element_is_clickable(self.locators.ADD_BUTTON).click()
            self.element_is_visible(self.locators.ADD_TRACKER).click()
            self.element_is_visible(self.locators.CREATE_NEW_TRACKER).click()
            self.element_is_visible(self.locators.ENTER_TRACKER_NAME).send_keys('TEST TRACKER')
            self.element_is_visible(self.locators.CREATE_TRACKER_BUTTON).click()
#2

Насколько знаю, тут нет красивого решения. Обычно делает проверка, if (el.isDisplayed()) и закрытие. Только перед проверкой нужно влепить sleep() на время через которое появляется ваш попап.

#3

Решил задачу таким способом:

    def close_popup_if_exist(self, locator):
        try:
            pop_up = wait(self.driver, 5).until(EC.presence_of_element_located(locator))
            pop_up.click()
        finally:
            return False

Ну а дальше вызываю эту функцию в if else