Selenium и Python: создание стабильных автотестов от А до Я

from selenium import webdriver
from webdriver_manager.chrome import ChromeDriverManager

driver = webdriver.Chrome(ChromeDriverManager().install())

from selenium import webdriver

driver = webdriver.Chrome()
driver.get('https://www.python.org')
print(driver.title)
driver.quit()

python -m venv selenium_env
source selenium_env/bin/activate # для Linux/Mac
selenium_env\Scripts\activate # для Windows

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

driver = webdriver.Chrome()
driver.get('https://www.python.org')

# Поиск элемента по ID
search_box = driver.find_element(By.ID, 'id-search-field')

# Поиск элемента по CSS-селектору
submit_button = driver.find_element(By.CSS_SELECTOR, 'button.search-button')

# Ввод текста в поле
search_box.send_keys('selenium python')

# Клик по кнопке
submit_button.click()

# Очистка текстового поля
search_box.clear()

# Проверка отображения элемента
is_displayed = search_box.is_displayed()

# Получение атрибута элемента
placeholder = search_box.get_attribute('placeholder')

# Найти все ссылки на странице
all_links = driver.find_elements(By.TAG_NAME, 'a')

# Перебор найденных элементов
for link in all_links:
print(link.get_attribute('href'))

from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

# Неявное ожидание
driver.implicitly_wait(10) # секунд

# Явное ожидание
element = WebDriverWait(driver, 10).until(
EC.presence_of_element_located((By.ID, 'results'))
)

# Открытие страницы
driver.get('https://www.python.org')

# Получение текущего URL
current_url = driver.current_url

# Получение заголовка страницы
title = driver.title

# Получение исходного кода страницы
page_source = driver.page_source

# Переход назад/вперед в истории браузера
driver.back()
driver.forward()

# Обновление страницы
driver.refresh()

# Открытие новой вкладки
driver.execute_script("window.open('https://www.selenium.dev', '_blank')")

# Получение списка идентификаторов окон
window_handles = driver.window_handles

# Переключение на новую вкладку (обычно последнюю в списке)
driver.switch_to.window(window_handles[-1])

# Переключение обратно на первую вкладку
driver.switch_to.window(window_handles[0])

# Закрытие текущей вкладки
driver.close()

# Закрытие всех вкладок и завершение сеанса
driver.quit()

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

# Получение текста алерта
alert_text = alert.text

# Принятие алерта (нажатие OK)
alert.accept()

# Отклонение алерта (нажатие Cancel)
alert.dismiss()

# Ввод текста в алерт (если поддерживается)
alert.send_keys('Текст для ввода')

# Добавление cookie
driver.add_cookie({"name": "session_id", "value": "12345"})

# Получение всех cookies
all_cookies = driver.get_cookies()

# Получение конкретной cookie
session_cookie = driver.get_cookie("session_id")

# Удаление cookie
driver.delete_cookie("session_id")

# Удаление всех cookies
driver.delete_all_cookies()

# Выполнение простого скрипта
driver.execute_script("return document.title")

# Прокрутка до элемента
element = driver.find_element(By.ID, "my-element")
driver.execute_script("arguments[0].scrollIntoView(true);", element)

# Скрытие элемента
driver.execute_script("arguments[0].style.display='none';", element)

# Установка размеров окна
driver.set_window_size(1920, 1080)

# Установка позиции окна
driver.set_window_position(0, 0)

# Разворачивание окна на весь экран
driver.maximize_window()

# Сворачивание окна
driver.minimize_window()

import pytest
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

@pytest.fixture
def browser():
driver = webdriver.Chrome()
driver.implicitly_wait(10)
yield driver
driver.quit()

def test_python_search(browser):
# Открываем страницу
browser.get("https://www.python.org")

# Находим поле поиска и вводим запрос
search_field = browser.find_element(By.ID, "id-search-field")
search_field.clear()
search_field.send_keys("pycon")

# Нажимаем кнопку поиска
browser.find_element(By.ID, "submit").click()

# Проверяем, что результаты поиска отображаются
WebDriverWait(browser, 10).until(
EC.presence_of_element_located((By.CLASS_NAME, "list-recent-events"))
)

# Проверяем, что "pycon" присутствует в результатах
results = browser.find_elements(By.CSS_SELECTOR, ".list-recent-events li")
assert len(results) > 0, "Результаты поиска не найдены"

# Проверяем текст в результатах
page_text = browser.page_source.lower()
assert "pycon" in page_text, "Запрос 'pycon' не найден в результатах"

# file: pages.py
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

class BasePage:
def __init__(self, driver):
self.driver = driver

def find_element(self, locator, time=10):
return WebDriverWait(self.driver, time).until(
EC.presence_of_element_located(locator),
message=f"Can't find element by locator {locator}"
)

class MainPage(BasePage):
def __init__(self, driver):
super().__init__(driver)
self.url = "https://www.python.org"

def open(self):
self.driver.get(self.url)

def search_for(self, query):
search_field = self.find_element((By.ID, "id-search-field"))
search_field.clear()
search_field.send_keys(query)
submit_button = self.find_element((By.ID, "submit"))
submit_button.click()

class SearchResultsPage(BasePage):
def get_results(self):
results_list = self.find_element((By.CLASS_NAME, "list-recent-events"))
items = results_list.find_elements(By.TAG_NAME, "li")
return items

def has_results(self):
return len(self.get_results()) > 0

def contains_text(self, text):
page_text = self.driver.page_source.lower()
return text.lower() in page_text

# file: test_search.py
import pytest
from selenium import webdriver
from pages import MainPage, SearchResultsPage

@pytest.fixture
def browser():
driver = webdriver.Chrome()
yield driver
driver.quit()

def test_python_search_with_pom(browser):
# Инициализация страниц
main_page = MainPage(browser)
results_page = SearchResultsPage(browser)

# Выполнение теста
main_page.open()
main_page.search_for("pycon")

# Проверки
assert results_page.has_results(), "Результаты поиска не найдены"
assert results_page.contains_text("pycon"), "Запрос 'pycon' не найден в результатах"

import pytest

@pytest.mark.parametrize("query,expected_text", [
("pycon", "pycon"),
("django", "web framework"),
("flask", "micro framework")
])
def test_python_search_parametrized(browser, query, expected_text):
main_page = MainPage(browser)
results_page = SearchResultsPage(browser)

main_page.open()
main_page.search_for(query)

assert results_page.has_results(), f"Результаты для запроса '{query}' не найдены"
assert results_page.contains_text(expected_text), f"Текст '{expected_text}' не найден в результатах"

def test_function(browser):
try:
# Тестовая логика
browser.get("https://www.example.com")
# ...
except Exception as e:
# Создание скриншота с временной меткой
import time
timestamp = int(time.time())
screenshot_path = f"error_{timestamp}.png"
browser.save_screenshot(screenshot_path)
print(f"Screenshot saved at {screenshot_path}")
raise # Передаем исключение дальше

from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.common.exceptions import StaleElementReferenceException

# Переопределяем функцию для повторных попыток при устаревшей ссылке на элемент
def click_with_retry(driver, locator, max_attempts=3):
for attempt in range(max_attempts):
try:
element = WebDriverWait(driver, 10).until(
EC.element_to_be_clickable(locator)
)
element.click()
return True
except StaleElementReferenceException:
if attempt == max_attempts – 1:
raise
print(f"StaleElementReferenceException, retrying... (attempt {attempt+1}/{max_attempts})")

# Использование
click_with_retry(driver, (By.ID, "dynamic-button"))

import os

# Получаем абсолютный путь к файлу
file_path = os.path.abspath("path/to/file.txt")

# Находим элемент загрузки и передаем путь
file_input = driver.find_element(By.ID, "fileUpload")
file_input.send_keys(file_path)

# Для JavaScript-алертов
try:
WebDriverWait(driver, 5).until(EC.alert_is_present())
alert = driver.switch_to.alert
alert.accept() # или alert.dismiss()
except:
print("No alert present")

# Для HTML-модальных окон
modal = WebDriverWait(driver, 10).until(
EC.visibility_of_element_located((By.ID, "modal-dialog"))
)
close_button = modal.find_element(By.CLASS_NAME, "close-btn")
close_button.click()

# Переключение на iframe по индексу
driver.switch_to.frame(0)

# Переключение на iframe по ID или имени
driver.switch_to.frame("iframe-id")

# Переключение на iframe по элементу
iframe = driver.find_element(By.TAG_NAME, "iframe")
driver.switch_to.frame(iframe)

# Возврат к основному содержимому
driver.switch_to.default_content()

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

chrome_options = Options()

# Эмуляция мобильного устройства
mobile_emulation = {
"deviceName": "iPhone X"
}
chrome_options.add_experimental_option("mobileEmulation", mobile_emulation)

driver = webdriver.Chrome(options=chrome_options)

# Или просто установка определенного разрешения
driver.set_window_size(375, 812) # iPhone X размеры