Spaces:
Sleeping
Sleeping
File size: 3,336 Bytes
3885f96 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 |
from fastapi.testclient import TestClient
import pytest
from app.main import app
# --- Подготовка к тестам ---
# Создаем "виртуального клиента", который будет отправлять запросы к приложению.
client = TestClient(app)
# --- Тест-кейс №1 ---
def test_read_main_page():
"""
Тест проверяет, что главная страница ("/") успешно загружается.
"""
# 1. Действие:
# Отправляем GET-запрос на главную страницу.
response = client.get("/")
# 2. Проверка (Assert):
# Утверждаем, что сервер ответил кодом 200 (OK).
assert response.status_code == 200
# 3. Проверка содержимого:
assert "Генератор объявлений" in response.text
# --- Тест-кейс №2 ---
def test_submit_image_successfully(mocker):
"""
Тест проверяет успешную отправку изображения и постановку задачи в очередь.
'mocker' - это специальный аргумент от pytest-mock для создания "моков".
"""
# 1. Подготовка "мока":
# Мы "подменяем" настоящую, тяжелую ML-функцию на заглушку.
mocker.patch(
'app.main.generate_ad_from_image',
return_value={"status": "mocked_ok"}
)
# 2. Действие:
# Имитируем отправку файла, как это делает браузер.
with open("tests/assets/test_image.png", "rb") as f:
response = client.post(
"/generate-ad",
data={"style": "brief"},
files={"image": ("test_image.png", f, "image/png")}
)
# 3. Проверки (Asserts):
# Утверждаем, что сервер ответил кодом 202 (Accepted),
# что означает "задача принята в обработку".
assert response.status_code == 202
# Утверждаем, что в JSON-ответе есть ключ "task_id".
data = response.json()
assert "task_id" in data
# --- Тест-кейс №3 ---
def test_submit_wrong_file_type():
"""
Тест проверяет, что сервер правильно отклоняет файлы, не являющиеся изображениями.
"""
# 1. Действие:
# Открываем текстовый файл и отправляем его на сервер.
with open("tests/assets/fake_image.txt", "rb") as f:
response = client.post(
"/generate-ad",
data={"style": "brief"},
files={"image": ("fake_image.txt", f, "text/plain")}
)
# 2. Проверки (Asserts):
# Утверждаем, что сервер ответил кодом 400 (Bad Request).
assert response.status_code == 400
# Утверждаем, что в JSON-ответе содержится правильное сообщение об ошибке.
data = response.json()
assert "detail" in data
assert data["detail"] == "Недопустимый тип файла." |