Автоматизация тестирования на Python: практические решения и варианты

Расширенные примеры автоматизации тестирования на Python

Пример 1: Параметризованные тесты с разными типами данных

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

import pytest

def divide(a, b):
    if b == 0:
        raise ValueError('Division by zero')
    return a / b

@pytest.mark.parametrize('a,b,expected', [
    (10, 2, 5.0),
    (7, 3, 7/3),
    (0, 1, 0.0),
    (-4, 2, -2.0),
    (1, 3, 1/3)
])
def test_divide_normal(a, b, expected):
    assert divide(a, b) == expected

@pytest.mark.parametrize('a,b', [
    (5, 0),
    (0, 0),
    (100, 0)
])
def test_divide_by_zero(a, b):
    with pytest.raises(ValueError, match='Division by zero'):
        divide(a, b)

Результат выполнения (фрагмент вывода pytest):

test_parametrize.py ........                          [100%]
8 passed in 0.04s

Каждый тест выполняется отдельно, при падении одного из них остальные продолжают работу.

Пример 2: Тестирование REST API с помощью pytest и requests

Для проверки внешних HTTP-сервисов удобно использовать библиотеку requests. Создадим фикстуру для базового URL и протестируем несколько эндпоинтов.

import pytest
import requests

@pytest.fixture
def base_url():
    return 'https://jsonplaceholder.typicode.com'

def test_get_posts(base_url):
    response = requests.get(f'{base_url}/posts/1')
    assert response.status_code == 200
    data = response.json()
    assert 'userId' in data
    assert 'title' in data

def test_create_post(base_url):
    payload = {'title': 'foo', 'body': 'bar', 'userId': 1}
    response = requests.post(f'{base_url}/posts', json=payload)
    assert response.status_code == 201
    created = response.json()
    assert created['title'] == 'foo'
    assert created['body'] == 'bar'
    assert created['userId'] == 1

def test_error_404(base_url):
    response = requests.get(f'{base_url}/posts/9999')
    assert response.status_code == 404

Вывод при успешном прохождении:

test_api.py ...                                   [100%]
3 passed

Если API недоступен, тесты упадут с ошибкой соединения. Для изоляции рекомендуется использовать моки (библиотека responses или unittest.mock).

Пример 3: Тестирование веб-интерфейса с Selenium WebDriver

Автоматизация браузера позволяет проверять UI. Установите selenium и драйвер (например, ChromeDriver).

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

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

def test_search_google(browser):
    browser.get('https://www.google.com')
    search_box = browser.find_element(By.NAME, 'q')
    search_box.send_keys('Python автоматизация тестирования')
    search_box.submit()
    assert 'Python' in browser.title

Результат (успех):

test_selenium.py .                                [100%]
1 passed

Типичные проблемы: версия драйвера не соответствует версии браузера, отсутствие драйвера в PATH. Решение – использование WebDriver Manager (pip install webdriver-manager).

Пример 4: Использование mock для изоляции зависимостей

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

from unittest.mock import patch
import pytest

# Функция, которая получает данные из внешнего API
def fetch_user_name(user_id):
    import requests
    response = requests.get(f'https://api.example.com/users/{user_id}')
    return response.json()['name']

# Тест с моком
def test_fetch_user_name():
    with patch('requests.get') as mock_get:
        mock_get.return_value.status_code = 200
        mock_get.return_value.json.return_value = {'name': 'Иван'}
        result = fetch_user_name(1)
        assert result == 'Иван'
        mock_get.assert_called_once_with('https://api.example.com/users/1')

Вывод:

test_mock.py .                                    [100%]
1 passed

Mock позволяет тестировать код без реальных сетевых вызовов, что ускоряет тесты и делает их детерминированными.

Пример 5: Генерация отчётов Allure для наглядности

Allure Framework формирует детальные отчёты с историей и вложениями. Установите allure-pytest и allure-commandline.

pip install allure-pytest

Добавим шаги в тест с помощью декораторов:

import allure
import pytest

@allure.step('Сложение двух чисел')
def add(a, b):
    return a + b

@allure.feature('Арифметика')
@allure.story('Сложение')
def test_add_allure():
    result = add(3, 4)
    allure.attach(str(result), name='Результат', attachment_type=allure.attachment_type.TEXT)
    assert result == 7

Запуск с генерацией отчёта:

pytest --alluredir=./allure-results test_allure.py
allure serve ./allure-results

Откроется веб-страница с визуализацией пройденных тестов, шагов и вложений.

Эти примеры демонстрируют гибкость Python в автоматизации различных видов тестирования. Используя комбинации pytest, mock, Selenium и других библиотек, можно покрыть практически любой сценарий проверки качества.