fastfood/tests/test_summary.py

import pytest
from httpx import AsyncClient

from .repository import Repository as Repo


@pytest.mark.asyncio
async def test_summary_with_menu(client: AsyncClient) -> None:
    # Проверяем пустое summary
    code, rspn = await Repo.Summary.read_summary(client)
    assert code == 200
    assert rspn == []

    # Создаем меню и проверяем ответ
    menu = {'title': 'Menu', 'description': 'main menu', 'submenus': []}
    code, rspn = await Repo.Menu.write(client, menu)
    menu.update(rspn)

    # Удалим ненужные ключи, тк в модели они не используются
    del menu['submenus_count']
    del menu['dishes_count']

    # Проверяем summary c меню
    code, rspn = await Repo.Summary.read_summary(client)
    assert code == 200
    assert rspn == [menu]

    # удаляем сопутствующее
    await Repo.Menu.delete(client, menu)


@pytest.mark.asyncio
async def test_summary_with_submenus(client: AsyncClient) -> None:
    # Создаем меню и проверяем ответ
    menu: dict[str, str | list | float] = {
        'title': 'Menu',
        'description': 'main menu',
        'submenus': [],
    }
    code, rspn = await Repo.Menu.write(client, menu)
    menu.update(rspn)

    del menu['submenus_count']
    del menu['dishes_count']

    # Создаем и проверяем подменю
    submenu: dict[str, str | list | float] = {
        'title': 'Submenu',
        'description': 'submenu',
        'parent_menu': menu['id'],
        'dishes': list(),
    }
    code, rspn = await Repo.Submenu.write(client, menu, submenu)
    submenu.update(rspn)
    del submenu['dishes_count']
    del submenu['parent_menu']

    menu['submenus'] = [submenu]

    # Получаем блюдо
    code, rspn = await Repo.Summary.read_summary(client)
    assert code == 200
    assert rspn == [menu]

    await Repo.Menu.delete(client, menu)


@pytest.mark.asyncio
async def test_summary_with_dishes(client: AsyncClient) -> None:
    # Создаем меню и проверяем ответ
    menu: dict[str, str | list | float] = {
        'title': 'Menu',
        'description': 'main menu',
        'submenus': [],
    }
    code, rspn = await Repo.Menu.write(client, menu)
    menu.update(rspn)

    del menu['submenus_count']
    del menu['dishes_count']

    # Создаем и проверяем подменю
    submenu: dict[str, str | list | float] = {
        'title': 'Submenu',
        'description': 'submenu',
        'parent_menu': menu['id'],
        'dishes': [],
    }
    code, rspn = await Repo.Submenu.write(client, menu, submenu)
    submenu.update(rspn)
    del submenu['dishes_count']
    del submenu['parent_menu']

    # Добавляем блюдо
    dish = {
        'title': 'dish',
        'description': 'some dish',
        'price': '12.5',
        'parent_submenu': submenu['id'],
    }
    code, rspn = await Repo.Dish.write(client, menu, submenu, dish)
    dish.update(rspn)
    del dish['parent_submenu']
    del dish['id']

    submenu['dishes'] = [dish]
    menu['submenus'] = [submenu]

    code, rspn = await Repo.Summary.read_summary(client)
    assert code == 200
    assert rspn == [menu]

    await Repo.Menu.delete(client, menu)