import pytest
from models import NewModel
from models_second import Config
from settings import TORTOISE_ORM
from tortoise import Tortoise
from tortoise.exceptions import OperationalError

try:
    # This error does not translate to tortoise's OperationalError
    from psycopg.errors import UndefinedColumn
except ImportError:
    errors = (OperationalError,)
else:
    errors = (OperationalError, UndefinedColumn)


@pytest.fixture(scope="session")
def anyio_backend() -> str:
    return "asyncio"


@pytest.fixture(autouse=True)
async def init_connections():
    await Tortoise.init(TORTOISE_ORM)
    try:
        yield
    finally:
        await Tortoise.close_connections()


@pytest.mark.anyio
async def test_init_db():
    m1 = await NewModel.filter(name="")
    assert isinstance(m1, list)
    m2 = await Config.filter(key="")
    assert isinstance(m2, list)
    await NewModel.create(name="")
    await Config.create(key="", label="", value={})


@pytest.mark.anyio
async def test_fake_field_1():
    assert "field_1" in NewModel._meta.fields_map
    assert "field_1" in Config._meta.fields_map
    with pytest.raises(errors):
        await NewModel.create(name="", field_1=1)
    with pytest.raises(errors):
        await Config.create(key="", label="", value={}, field_1=1)

    obj1 = NewModel(name="", field_1=1)
    with pytest.raises(errors):
        await obj1.save()
    obj1 = NewModel(name="")
    with pytest.raises(errors):
        await obj1.save()
    with pytest.raises(errors):
        obj1 = await NewModel.first()
    obj1 = await NewModel.all().first().values("id", "name")
    assert obj1 and obj1["id"]

    obj2 = Config(key="", label="", value={}, field_1=1)
    with pytest.raises(errors):
        await obj2.save()
    obj2 = Config(key="", label="", value={})
    with pytest.raises(errors):
        await obj2.save()
    with pytest.raises(errors):
        obj2 = await Config.first()
    obj2 = await Config.all().first().values("id", "key")
    assert obj2 and obj2["id"]


@pytest.mark.anyio
async def test_fake_field_2():
    assert "field_2" in NewModel._meta.fields_map
    assert "field_2" in Config._meta.fields_map
    with pytest.raises(errors):
        await NewModel.create(name="")
    with pytest.raises(errors):
        await Config.create(key="", label="", value={})