import pytest
from aiopg.sa import SAConnection
from httpx import AsyncClient

from src.repositories.tables import messenger_handbook_country_table
from tests.samples import ACCOUNT_1, ACCOUNT_2


@pytest.fixture
def sa_tables():
    return [messenger_handbook_country_table]


@pytest.fixture(autouse=True)
async def _enter_data(connection: SAConnection):
    await connection.execute(messenger_handbook_country_table.insert().values([ACCOUNT_1, ACCOUNT_2]))


async def test_get_info_from_postgresql(test_client: AsyncClient):
    response = await test_client.get("/test_psql/1")
    response_json = response.json()

    assert (response.status_code, response_json) == (200, [ACCOUNT_1])