tests/test_server.py - main - dancig/TattooImageGenerator

tests/test_server.py
144 строки8 KB
dancig
feat: add image background removing request
07 июн 2025, 02:00
07 июн 2025, 02:0096af0ef
import pytest
from Server import app
from Server import imageGen


@pytest.fixture()
def testapp():

    yield app


@pytest.fixture()
def client(testapp):
    return testapp.test_client()


# Запрос GET с адресом '/123'
# Модуль генерации изображений не занят
# Должен возвращаться ответ с кодом 404
def test_process_request_1(testapp, client):
    imageGen.isBusy = False
    response = client.get("/123")
    assert response.status_code == 404


# Запрос GET с адресом '/'
# Модуль генерации изображений не занят
# Должен возвращаться ответ с начальной страницей и кодом 200
def test_process_request_2(client):
    imageGen.isBusy = False
    response = client.get("/")
    assert response.status_code == 200
    assert "<h1>Генератор изображений татуировок</h1>" in response.data.decode()


# Запрос POST с адресом '/'
# Модуль генерации изображений не занят
# Должен возвращаться ответ с начальной страницей, ошибкой некорректоного ввода и кодом 200
def test_process_request_3(client):
    imageGen.isBusy = False
    response = client.post("/")
    assert response.status_code == 200
    assert "Некорректный ввод." in response.data.decode()


# Запрос POST с адресом '/' и данными запроса {'prompt'= пустая строка}
# Модуль генерации изображений не занят
# Должен возвращаться ответ с начальной страницей, ошибкой некорректоного ввода и кодом 200
def test_process_request_4(client):
    imageGen.isBusy = False
    response = client.post("/", data={'prompt': ""})
    assert response.status_code == 200
    assert "Некорректный ввод." in response.data.decode()


# Запрос POST с адресом '/' и данными запроса {'prompt'= строка с 501 символом}
# Модуль генерации изображений не занят
# Должен возвращаться ответ с начальной страницей, ошибкой некорректоного ввода и кодом 200
def test_process_request_5(client):
    imageGen.isBusy = False
    response = client.post("/", data={'prompt': ("aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
                                                 "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
                                                 "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
                                                 "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
                                                 "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
                                                 "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
                                                 "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
                                                 "aaaaaaaaaaaaaaaaaa")})
    assert response.status_code == 200
    assert "Некорректный ввод." in response.data.decode()


# Запрос POST с адресом '/' и данными запроса {'prompt'= "image"}
# Модуль генерации изображений занят
# Должен возвращаться ответ с начальной страницей, просьбой попробовать позже и кодом 200
def test_process_request_6(client, testapp):
    imageGen.isBusy = True
    response = client.post("/", data={'prompt': "image"})
    assert response.status_code == 200
    assert "Попробуйте позже." in response.data.decode()


# Запрос POST с адресом '/' и данными запроса {'prompt'= "image of a cat"}
# Модуль генерации изображений не занят
# Должен возвращаться ответ со страницей после генерации изображения и кодом 200
def test_process_request_7(client):
    imageGen.isBusy = False
    response = client.post("/", data={'prompt': "image of a cat"})
    assert response.status_code == 200
    assert "СгенерированноеИзображение" in response.data.decode()


# Запрос POST с адресом '/' и данными запроса {'prompt'= "Изображение кота"}
# Модуль генерации изображений не занят
# Должен возвращаться ответ со страницей после генерации изображения и кодом 200
def test_process_request_8(client):
    imageGen.isBusy = False
    response = client.post("/", data={'prompt': "Изображение кота"})
    assert response.status_code == 200
    assert "СгенерированноеИзображение" in response.data.decode()


# Запрос POST с адресом '/' и данными запроса {'prompt'= строка с 500 символов}
# Модуль генерации изображений не занят
# Должен возвращаться ответ со страницей после генерации изображения и кодом 200
def test_process_request_9(client):
    imageGen.isBusy = False
    response = client.post("/", data={'prompt': ("aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
                                                 "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
                                                 "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
                                                 "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
                                                 "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
                                                 "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
                                                 "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
                                                 "aaaaaaaaaaaaaaaaa")})
    assert response.status_code == 200
    assert "СгенерированноеИзображение" in response.data.decode()


# Запрос POST с адресом '/' и данными запроса {'prompt'= "Изображение числа 8"}
# Модуль генерации изображений не занят
# Должен возвращаться ответ со страницей после генерации изображения и кодом 200
def test_process_request_10(client):
    imageGen.isBusy = False
    response = client.post("/", data={'prompt': "Изображение числа 8"})
    assert response.status_code == 200
    assert "СгенерированноеИзображение" in response.data.decode()


def test_process_request_11(client):
    response = client.post("/", json={'path': 'http://127.0.0.1:5000/static/images/2025_06_06_161638.png',
                                      'threshold': '40',
                                      'erode': 'false',
                                      'dilate': 'true'})
    assert response.status_code == 200
    assert response.json['image'] == 'static\\images\\2025_06_06_161638_no_bg.png'


def test_process_request_12(client):
    response = client.post("/", json={'path': 'http://127.0.0.1:5000/static/images/image.png',
                                      'threshold': '40',
                                      'erode': 'false',
                                      'dilate': 'true'})
    assert response.status_code == 404