import pytest
from httpx import AsyncClient

from .repository import Repository as Repo


@pytest.mark.asyncio
async def test_menu_crud_empty(client: AsyncClient) -> None:
    """Тестирование функций меню"""
    code, rspn = await Repo.Menu.read_all(client)
    assert code == 200
    assert rspn == []


@pytest.mark.asyncio
async def test_menu_crud_add(client: AsyncClient) -> None:
    """Тестирование функций меню"""
    data = {'title': 'Menu', 'description': None}
    code, rspn = await Repo.Menu.write(client, data)
    assert code == 201
    assert rspn['title'] == 'Menu'
    assert rspn['description'] is None
    await Repo.Menu.delete(client, rspn)


@pytest.mark.asyncio
async def test_menu_crud_get(client: AsyncClient) -> None:
    """Тестирование функций меню"""
    data = {'title': 'Menu', 'description': None}
    code, rspn = await Repo.Menu.write(client, data)
    code, menu = await Repo.Menu.get(client, {'id': rspn.get('id')})
    assert code == 200
    assert menu['title'] == rspn['title']
    await Repo.Menu.delete(client, menu)


@pytest.mark.asyncio
async def test_menu_crud_update(client: AsyncClient) -> None:
    """Тестирование функций меню"""
    data = {'title': 'Menu', 'description': None}
    code, rspn = await Repo.Menu.write(client, data)

    upd_data = {
        'id': rspn.get('id'),
        'title': 'upd Menu',
        'description': '',
    }
    code, upd_rspn = await Repo.Menu.update(client, upd_data)
    assert code == 200
    assert upd_rspn['title'] == 'upd Menu'
    await Repo.Menu.delete(client, upd_rspn)


@pytest.mark.asyncio
async def test_menu_crud_delete(client: AsyncClient) -> None:
    """Тестирование функций меню"""
    data = {'title': 'Menu', 'description': None}
    code, rspn = await Repo.Menu.write(client, data)

    code = await Repo.Menu.delete(client, rspn)
    assert code == 200

    code, rspn = await Repo.Menu.get(client, {'id': rspn.get('id')})
    assert code == 404


@pytest.mark.asyncio
async def test_menu_crud_get_all(client: AsyncClient) -> None:
    """Тестирование функций меню"""
    code, rspn = await Repo.Menu.read_all(client)
    assert code == 200
    assert rspn == []

    data = {'title': 'Menu', 'description': None}
    code, rspn = await Repo.Menu.write(client, data)

    code, upd_rspn = await Repo.Menu.read_all(client)
    assert code == 200
    assert upd_rspn == [rspn]
    await Repo.Menu.delete(client, rspn)


@pytest.mark.asyncio
async def test_submenus_get_all(client) -> None:
    # Создаем меню и проверяем ответ
    menu = {'title': 'Menu', 'description': 'main menu'}
    code, rspn = await Repo.Menu.write(client, menu)
    assert code == 201
    menu.update(rspn)

    # Проверяем наличие подменю
    code, rspn = await Repo.Submenu.read_all(client, menu)
    assert code == 200
    assert rspn == []

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

    # Проверяем наличие подменю
    code, upd_rspn = await Repo.Submenu.read_all(client, menu)
    assert code == 200
    assert upd_rspn == [rspn]

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


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

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

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


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

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

    # Обновляем подменю и проверяем
    submenu['title'] = 'updated_submenu'
    code, rspn = await Repo.Submenu.update(client, menu, submenu)
    assert code == 200
    assert submenu['title'] == rspn['title']
    submenu.update(rspn)

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


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

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

    # Удаляем подменю
    code = await Repo.Submenu.delete(client, menu, submenu)
    assert code == 200

    # Проверяем удаленное подменю
    code, rspn = await Repo.Submenu.get(client, menu, submenu)
    assert code == 404

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


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

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

    # Проверяем все блюда в подменю
    code, rspn = await Repo.Dish.read_all(client, menu, submenu)
    assert code == 200
    assert rspn == []

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

    code, upd_rspn = await Repo.Dish.read_all(client, menu, submenu)

    assert code == 200

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


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

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

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

    # Получаем блюдо
    code, rspn = await Repo.Dish.get(client, menu, submenu, dish)
    assert code == 200
    assert rspn['title'] == dish['title']

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


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

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

    # Добавляем блюдо
    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)

    # Обновляем блюдо и проверяем
    dish['title'] = 'updated_dish'
    code, rspn = await Repo.Dish.update(client, menu, submenu, dish)
    assert code == 200
    assert dish['title'] == rspn['title']
    dish.update(rspn)

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


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

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

    # Добавляем блюдо
    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)

    # Удаляем подменю
    code = await Repo.Dish.delete(client, menu, submenu, dish)
    assert code == 200

    # Проверяем удаленное блюдо
    code, rspn = await Repo.Dish.get(client, menu, submenu, dish)
    assert code == 404

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