import pytest
from httpx import AsyncClient

@pytest.mark.asyncio
async def test_get_yeardatas(client: AsyncClient):
    response = await client.get("/yeardata")
    assert response.status_code == 200
    assert response.json()["message"] == "Data retrieved successfully"

@pytest.mark.asyncio
async def test_create_yeardata(client: AsyncClient):
    payload = {
        "year": 2024,
        "description": "Test Year Data"
    }
    response = await client.post("/yeardata", json=payload)
    assert response.status_code == 200
    assert response.json()["message"] == "Data created successfully"