Автоматизация тестирования веб-приложений с помощью Selenium и Python

Продвинутые примеры использования Selenium

Пример 1: Автоматизация многостраничной формы

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

driver = webdriver.Chrome()
driver.get('https://example.com/form')

# Страница 1: заполнение полей
driver.find_element(By.NAME, 'firstName').send_keys('Иван')
driver.find_element(By.NAME, 'email').send_keys('ivan@mail.com')
driver.find_element(By.CLASS_NAME, 'next-btn').click()

# Страница 2: ожидание загрузки и выбор опции
wait = WebDriverWait(driver, 10)
option = wait.until(EC.element_to_be_clickable((By.XPATH, '//input[@value="option2"]')))
option.click()

submit = driver.find_element(By.ID, 'submit')
submit.click()

# Проверка успешного завершения
success_msg = WebDriverWait(driver, 5).until(
    EC.presence_of_element_located((By.CLASS_NAME, 'success'))
)
print('Форма отправлена:', success_msg.text)

driver.quit()

Пояснение: Используются явные ожидания для каждого шага, чтобы избежать ошибок при переходах между страницами.

Форма отправлена: Спасибо за регистрацию!

Пример 2: Извлечение данных из таблицы в pandas DataFrame

import pandas as pd
from selenium import webdriver
from selenium.webdriver.common.by import By

driver = webdriver.Chrome()
driver.get('https://quotes.toscrape.com/table')

# Находим таблицу
table = driver.find_element(By.TAG_NAME, 'table')
rows = table.find_elements(By.TAG_NAME, 'tr')

data = []
for row in rows:
    cols = row.find_elements(By.TAG_NAME, 'td')
    if cols:
        data.append([col.text for col in cols])

df = pd.DataFrame(data)
print(df.head())
driver.quit()

Пояснение: Собираем все строки таблицы, извлекаем текст из ячеек. Результат конвертируется в DataFrame.

0  1  2
0  Имя     Возраст  Город
1  Анна   25       Москва
2  Борис  30       СПб

Пример 3: Работа с несколькими вкладками

from selenium import webdriver
from selenium.webdriver.common.by import By

driver = webdriver.Chrome()
driver.get('https://example.com')

# Открываем новую вкладку через JS
driver.execute_script('window.open("_blank");')

# Переключаемся на последнюю вкладку
driver.switch_to.window(driver.window_handles[-1])
driver.get('https://google.com')
print('Текущая вкладка:', driver.title)

# Вернуться на первую вкладку
driver.switch_to.window(driver.window_handles[0])
print('Первая вкладка:', driver.title)

driver.quit()

Пояснение: window_handles - список всех открытых вкладок. Переключение по индексу или по дескриптору.

Текущая вкладка: Google
Первая вкладка: Example Domain

Пример 4: Сохранение и использование кук для сессии

import pickle
from selenium import webdriver

driver = webdriver.Chrome()
driver.get('https://example.com/login')

# Выполняем вход (заполняем логин/пароль и жмём кнопку) ...
# После успешного входа сохраняем куки
cookies = driver.get_cookies()
with open('cookies.pkl', 'wb') as f:
    pickle.dump(cookies, f)

# В новом сеансе загружаем куки
driver2 = webdriver.Chrome()
driver2.get('https://example.com')  # обязательно сначала открыть домен
with open('cookies.pkl', 'rb') as f:
    loaded_cookies = pickle.load(f)
for cookie in loaded_cookies:
    driver2.add_cookie(cookie)

driver2.get('https://example.com/dashboard')
# Теперь пользователь авторизован
driver2.quit()

Пояснение: Куки сохраняются в файл, затем для новой сессии добавляются перед переходом на защищённую страницу.

(Нет вывода, но после загрузки кук страница дашборда открывается авторизованной)

Пример 5: Headless режим с настройками ChromeOptions

from selenium import webdriver

options = webdriver.ChromeOptions()
options.add_argument('--headless')
options.add_argument('--disable-gpu')
options.add_argument('--window-size=1920x1080')

driver = webdriver.Chrome(options=options)
driver.get('https://example.com')

print(driver.title)
driver.save_screenshot('headless_screenshot.png')
driver.quit()

Пояснение: Headless-режим запускает браузер без графического интерфейса. Полезно для серверов и CI.

Example Domain