import pytest

from src.dtos.ISayHelloDto import ISayHelloDto
from src.index import root, say_hello, hello_message


@pytest.mark.asyncio
async def test_root():
    result = await root()
    assert result == {'message': 'Hello World'}


@pytest.mark.asyncio
async def test_say_hello():
    result = await say_hello("John")
    assert result == {'message': 'Hello John'}


@pytest.mark.asyncio
async def test_hello_message():
    dto = ISayHelloDto(message="Alice")
    result = await hello_message(dto)
    assert result == {'message': 'Hello Alice'}