10 промтов для написания unit-тестов и интеграционных тестов: от новичка до эксперта

Напиши unit-тест на pytest для функции `calculate_discount(price: float, discount_percent: float) -> float`, которая возвращает цену со скидкой. Учти граничные случаи: discount_percent = 0, discount_percent = 100, price = 0. Используй фикстуры.

import pytest

@pytest.fixture
def sample_price():
    return 100.0

def test_calculate_discount_normal(sample_price):
    assert calculate_discount(sample_price, 10) == 90.0

def test_calculate_discount_zero_percent(sample_price):
    assert calculate_discount(sample_price, 0) == 100.0

def test_calculate_discount_full(sample_price):
    assert calculate_discount(sample_price, 100) == 0.0

def test_calculate_discount_zero_price():
    assert calculate_discount(0, 50) == 0.0

def test_calculate_discount_negative():
    with pytest.raises(ValueError):
        calculate_discount(100, -10)

Напиши интеграционный тест на pytest для функции, которая сохраняет пользователя в базу данных PostgreSQL и затем извлекает его. Используй SQLAlchemy и тестовую базу данных, создаваемую через testcontainers.

import pytest
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from testcontainers.postgres import PostgresContainer
from app.models import User, Base

@pytest.fixture(scope="module")
def db_session():
    with PostgresContainer("postgres:15") as postgres:
        engine = create_engine(postgres.get_connection_url())
        Base.metadata.create_all(engine)
        Session = sessionmaker(bind=engine)
        yield Session()

def test_create_and_retrieve_user(db_session):
    user = User(name="Alice", email="alice@example.com")
    db_session.add(user)
    db_session.commit()

    retrieved = db_session.query(User).filter_by(email="alice@example.com").first()
    assert retrieved.name == "Alice"

Создай интеграционный тест на pytest с использованием TestClient из FastAPI для эндпоинта POST /api/items. Тест должен проверить: успешное создание, возврат 201, валидацию тела запроса (пропущено обязательное поле), и ответ 422.

import pytest
from fastapi.testclient import TestClient
from app.main import app

client = TestClient(app)

def test_create_item_success():
    response = client.post("/api/items", json={"name": "Laptop", "price": 1200})
    assert response.status_code == 201
    assert response.json()["name"] == "Laptop"

def test_create_item_missing_field():
    response = client.post("/api/items", json={"name": "Laptop"})
    assert response.status_code == 422

Напиши тесты на Jest для функции `validateEmail(email: string): boolean`, которая проверяет email по простому regex. Включи тесты для: валидного email, невалидного email, пустой строки, null.

const validateEmail = require('./validateEmail');

test('valid email returns true', () => {
  expect(validateEmail('test@example.com')).toBe(true);
});

test('invalid email returns false', () => {
  expect(validateEmail('test@')).toBe(false);
});

test('empty string returns false', () => {
  expect(validateEmail('')).toBe(false);
});

test('null returns false', () => {
  expect(validateEmail(null)).toBe(false);
});

Напиши асинхронный тест на pytest для функции `async fetch_data(url: str) -> dict`, которая использует aiohttp. Используй unittest.mock или pytest-asyncio для мока HTTP-запроса.

import pytest
from unittest.mock import AsyncMock, patch
import aiohttp

@pytest.mark.asyncio
async def test_fetch_data_success():
    mock_response = AsyncMock()
    mock_response.json = AsyncMock(return_value={"key": "value"})
    mock_response.status = 200

    with patch("aiohttp.ClientSession.get", return_value=mock_response):
        result = await fetch_data("http://example.com")
        assert result == {"key": "value"}

Напиши unit-тест на pytest с использованием mocker из pytest-mock для функции `send_email(user_email, message)`, которая вызывает внешний SMTP-сервер. Проверь, что функция вызвана с правильными аргументами.

def test_send_email_calls_smtp(mocker):
    mock_smtp = mocker.patch('smtplib.SMTP')
    send_email('user@example.com', 'Hello')
    mock_smtp.assert_called_once()
    mock_smtp.return_value.sendmail.assert_called_once()

Используй @pytest.mark.parametrize для тестирования функции `is_prime(n: int) -> bool`. Передай 10 тестовых случаев: простые числа, составные числа, 0, 1, отрицательные.

import pytest

@pytest.mark.parametrize("n,expected", [
    (2, True),
    (3, True),
    (4, False),
    (0, False),
    (1, False),
    (-5, False),
    (17, True),
])
def test_is_prime(n, expected):
    assert is_prime(n) == expected

Напиши тест на pytest для функции `divide(a, b)`, который проверяет, что при b=0 выбрасывается ZeroDivisionError с сообщением "Division by zero is not allowed".

def test_divide_by_zero():
    with pytest.raises(ZeroDivisionError, match="Division by zero is not allowed"):
        divide(10, 0)

Напиши интеграционный тест на pytest, который вызывает реальный API погоды (например, OpenWeatherMap) с корректным ключом. Используй фикстуру для настройки ключа из переменной окружения. Добавь пропуск теста, если ключа нет.

import pytest
import os
import requests

@pytest.fixture
def api_key():
    key = os.getenv("OWM_API_KEY")
    if not key:
        pytest.skip("API key not set")
    return key

def test_get_weather(api_key):
    url = f"http://api.openweathermap.org/data/2.5/weather?q=London&appid={api_key}"
    response = requests.get(url)
    assert response.status_code == 200
    assert "weather" in response.json()

Проанализируй следующий код и напиши тесты на pytest так, чтобы покрытие было 100%:

def process_order(order):
    if order["total"] > 100:
        return "free_shipping"
    elif order["total"] > 0:
        return "paid_shipping"
    else:
        return "invalid_order"

import pytest

def test_free_shipping():
    assert process_order({"total": 150}) == "free_shipping"

def test_paid_shipping():
    assert process_order({"total": 50}) == "paid_shipping"

def test_invalid_order():
    assert process_order({"total": 0}) == "invalid_order"