Автоматизация веб-интерфейсов: Selenium и Python

Как использовать кастомные ожидания с условием?

Иногда встроенных условий недостаточно. Можно написать свой класс, реализующий expected_condition.

from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.common.by import By

class element_has_text(object):
    def __init__(self, locator, text):
        self.locator = locator
        self.text = text

    def __call__(self, driver):
        element = driver.find_element(*self.locator)
        if self.text in element.text:
            return element
        else:
            return False

wait = WebDriverWait(driver, 10)
element = wait.until(element_has_text((By.ID, "status"), "Завершено"))
print(element.text)

Завершено

Такое ожидание будет ждать, пока в тексте элемента не появится нужная подстрока.

Как управлять куками браузера?

Куки можно добавлять, удалять и получать для настройки сессии.

# Получить все куки
cookies = driver.get_cookies()

# Добавить куку
driver.add_cookie({"name": "session_id", "value": "abc123", "domain": ".example.com"})

# Удалить куку по имени
driver.delete_cookie("session_id")

# Очистить все куки
driver.delete_all_cookies()

# Проверить, что кука установлена
driver.refresh()  # обновление страницы для применения куки
print(driver.get_cookie("session_id"))

{'name': 'session_id', 'value': 'abc123', ...}

Как работать с алертами и модальными окнами?

Для работы с JavaScript алертами, подтверждениями и промптами используется switch_to.alert.

# Нажатие кнопки, вызывающей alert
driver.find_element(By.ID, "alert-btn").click()

# Переключение на alert
alert = driver.switch_to.alert

# Получить текст алерта
print(alert.text)

# Принять (OK)
alert.accept()

# Для confirm: alert.dismiss() – отмена
# Для prompt: alert.send_keys("текст") – затем accept()

Если игнорировать алерт, дальнейшие действия вызовут UnexpectedAlertPresentException.

Как использовать ActionChains для сложных взаимодействий?

ActionChains позволяет выполнять последовательности действий: перетаскивание, hover, двойной клик, контекстное меню.

from selenium.webdriver import ActionChains

action = ActionChains(driver)

# Перетаскивание элемента source на target
action.drag_and_drop(source_element, target_element).perform()

# Наведение мыши на элемент
action.move_to_element(element).perform()

# Двойной клик
action.double_click(element).perform()

# Комбинация клавиш (Ctrl+A)
action.key_down(Keys.CONTROL).send_keys('a').key_up(Keys.CONTROL).perform()

# Клик и удержание
action.click_and_hold(element).move_by_offset(50, 0).release().perform()

Цепочки действий могут быть не выполнены, если элемент не видим или перекрыт.

Как работать с Shadow DOM?

Элементы, инкапсулированные в Shadow DOM, недоступны через обычные селекторы. Необходимо получить корневой элемент Shadow Root через JavaScript.

# Псевдо-селектор /deep/ и >>> устарели, используем execute_script
host_element = driver.find_element(By.CSS_SELECTOR, "my-component")
shadow_root = driver.execute_script("return arguments[0].shadowRoot", host_element)
# Теперь shadow_root можно использовать как обычный элемент (но только с Selenium 4)
# Для поиска внутри Shadow DOM:
inner_element = shadow_root.find_element(By.CSS_SELECTOR, ".inner")
print(inner_element.text)

В Selenium 4 поиск внутри shadow root доступен через метод find_element, но в ранних версиях нужен JavaScript.

Как запустить тесты параллельно с Selenium Grid?

Selenium Grid позволяет распределить тесты между несколькими узлами (nodes) и браузерами. На машине запускается хаб, а тесты подключаются к нему через Remote WebDriver.

from selenium import webdriver
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities

driver = webdriver.Remote(
    command_executor='http://localhost:4444/wd/hub',
    desired_capabilities=DesiredCapabilities.CHROME
)
driver.get("https://example.com")
print(driver.title)
driver.quit()

Параллельное выполнение обычно настраивается через тестовый фреймворк (pytest, unittest) с использованием плагинов (pytest-xdist). Каждый тест получает свой driver, созданный через Remote.

Example Domain

Как использовать браузерный профиль для сохранения сессии?

При тестировании может потребоваться сохранить куки и локальное хранилище между запусками. Для Chrome используется user-data-dir.

from selenium import webdriver
from selenium.webdriver.chrome.options import Options

options = Options()
options.add_argument("--user-data-dir=/path/to/profile")
options.add_argument("--profile-directory=Default")
driver = webdriver.Chrome(options=options)
driver.get("https://example.com")
# Все данные сессии сохранятся в profile

В Firefox используется FirefoxProfile.

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