118 lines
3.3 KiB
Python
118 lines
3.3 KiB
Python
|
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',
|
||
|
|
}
|
||
|
|
code, rspn = await Repo.Menu.write(client, menu)
|
||
|
|
menu.update(rspn)
|
||
|
|
|
||
|
|
# Удалим ненужные ключи, тк в модели они не используются
|
||
|
|
del menu['submenus_count']
|
||
|
|
del menu['dishes_count']
|
||
|
|
menu.__setattr__('submenus', list())
|
||
|
|
|
||
|
|
# Проверяем 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 = {
|
||
|
|
'title': 'Menu',
|
||
|
|
'description': 'main menu',
|
||
|
|
}
|
||
|
|
code, rspn = await Repo.Menu.write(client, menu)
|
||
|
|
menu.update(rspn)
|
||
|
|
|
||
|
|
del menu['submenus_count']
|
||
|
|
del menu['dishes_count']
|
||
|
|
menu.__setattr__('submenus', list())
|
||
|
|
|
||
|
|
# Создаем и проверяем подменю
|
||
|
|
submenu = {
|
||
|
|
'title': 'Submenu',
|
||
|
|
'description': 'submenu',
|
||
|
|
'parent_menu': menu['id'],
|
||
|
|
}
|
||
|
|
code, rspn = await Repo.Submenu.write(client, menu, submenu)
|
||
|
|
submenu.update(rspn)
|
||
|
|
submenu.__setattr__('dishes', list())
|
||
|
|
del submenu['dishes_count']
|
||
|
|
del submenu['parent_menu']
|
||
|
|
|
||
|
|
menu.__setattr__('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 = {
|
||
|
|
'title': 'Menu',
|
||
|
|
'description': 'main menu',
|
||
|
|
}
|
||
|
|
code, rspn = await Repo.Menu.write(client, menu)
|
||
|
|
menu.update(rspn)
|
||
|
|
|
||
|
|
del menu['submenus_count']
|
||
|
|
del menu['dishes_count']
|
||
|
|
menu.__setattr__('submenus', list())
|
||
|
|
|
||
|
|
# Создаем и проверяем подменю
|
||
|
|
submenu = {
|
||
|
|
'title': 'Submenu',
|
||
|
|
'description': 'submenu',
|
||
|
|
'parent_menu': menu['id'],
|
||
|
|
}
|
||
|
|
code, rspn = await Repo.Submenu.write(client, menu, submenu)
|
||
|
|
submenu.update(rspn)
|
||
|
|
submenu.__setattr__('dishes', list())
|
||
|
|
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.__setattr__('dishes', dish)
|
||
|
|
menu.__setattr__('submenus', submenu)
|
||
|
|
|
||
|
|
code, rspn = await Repo.Summary.read_summary(client)
|
||
|
|
assert code == 200
|
||
|
|
assert rspn == [menu]
|
||
|
|
|
||
|
|
await Repo.Menu.delete(client, menu)
|