from datetime import datetime as dt
from datetime import timezone

import pytest
from conftest import DB, Response
from hircine.db.models import Artist


@pytest.fixture
def query_artist(execute_id):
    query = """
    query artist($id: Int!) {
        artist(id: $id) {
            __typename
            ... on Artist {
                id
                name
            }
            ... on Error {
                message
            }
            ... on IDNotFoundError {
                id
            }
        }
    }
    """

    return execute_id(query)


@pytest.fixture
def query_artists(execute):
    query = """
    query artists {
        artists {
            __typename
            count
            edges {
                id
                name
            }
        }
    }
    """

    return execute(query)


@pytest.fixture
def add_artist(execute_add):
    mutation = """
    mutation addArtist($input: AddArtistInput!) {
        addArtist(input: $input) {
            __typename
            ... on AddSuccess {
                id
            }
            ... on Error {
                message
            }
            ... on InvalidParameterError {
                parameter
            }
        }
    }
    """

    return execute_add(mutation)


@pytest.fixture
def update_artists(execute_update):
    mutation = """
    mutation updateArtists($ids: [Int!]!, $input: UpdateArtistInput!) {
        updateArtists(ids: $ids, input: $input) {
            __typename
            ... on Success {
                message
            }
            ... on Error {
                message
            }
            ... on IDNotFoundError {
                id
            }
            ... on InvalidParameterError {
                parameter
            }
        }
    }
    """  # noqa: E501

    return execute_update(mutation)


@pytest.fixture
def delete_artists(execute_delete):
    mutation = """
    mutation deleteArtists($ids: [Int!]!) {
        deleteArtists(ids: $ids) {
            __typename
            ... on Success {
                message
            }
            ... on Error {
                message
            }
            ... on IDNotFoundError {
                id
            }
        }
    }
    """

    return execute_delete(mutation)


@pytest.mark.anyio
async def test_query_artist(query_artist, gen_artist):
    artist = await DB.add(next(gen_artist))

    response = Response(await query_artist(artist.id))
    response.assert_is("Artist")

    assert response.id == artist.id
    assert response.name == artist.name


@pytest.mark.anyio
async def test_query_artist_fails_not_found(query_artist):
    response = Response(await query_artist(1))
    response.assert_is("IDNotFoundError")
    assert response.id == 1
    assert response.message == "Artist ID not found: '1'"


@pytest.mark.anyio
async def test_query_artists(query_artists, gen_artist):
    artists = await DB.add_all(*gen_artist)

    response = Response(await query_artists())
    response.assert_is("ArtistFilterResult")

    assert response.count == len(artists)
    assert isinstance((response.edges), list)
    assert len(response.edges) == len(artists)

    edges = iter(response.edges)
    for artist in sorted(artists, key=lambda a: a.name):
        edge = next(edges)
        assert edge["id"] == artist.id
        assert edge["name"] == artist.name


@pytest.mark.anyio
async def test_add_artist(add_artist):
    response = Response(await add_artist({"name": "added artist"}))
    response.assert_is("AddSuccess")

    artist = await DB.get(Artist, response.id)
    assert artist is not None
    assert artist.name == "added artist"


@pytest.mark.anyio
async def test_add_artist_fails_empty_parameter(add_artist):
    response = Response(await add_artist({"name": ""}))

    response.assert_is("InvalidParameterError")
    assert response.parameter == "name"
    assert response.message == "Invalid parameter 'name': cannot be empty"


@pytest.mark.anyio
async def test_add_artist_fails_exists(add_artist, gen_artist):
    artist = await DB.add(next(gen_artist))

    response = Response(await add_artist({"name": artist.name}))
    response.assert_is("NameExistsError")
    assert response.message == "Another Artist with this name exists"


@pytest.mark.anyio
async def test_delete_artist(delete_artists, gen_artist):
    artist = await DB.add(next(gen_artist))
    id = artist.id

    response = Response(await delete_artists(id))
    response.assert_is("DeleteSuccess")

    artist = await DB.get(Artist, id)
    assert artist is None


@pytest.mark.anyio
async def test_delete_artist_not_found(delete_artists):
    response = Response(await delete_artists(1))

    response.assert_is("IDNotFoundError")
    assert response.id == 1
    assert response.message == "Artist ID not found: '1'"


@pytest.mark.anyio
async def test_update_artist(update_artists, gen_artist):
    artist = await DB.add(next(gen_artist))

    input = {"name": "updated artist"}
    response = Response(await update_artists(artist.id, input))
    response.assert_is("UpdateSuccess")

    artist = await DB.get(Artist, artist.id)
    assert artist is not None
    assert artist.name == "updated artist"


@pytest.mark.anyio
async def test_update_artist_fails_exists(update_artists, gen_artist):
    first = await DB.add(next(gen_artist))
    second = await DB.add(next(gen_artist))

    response = Response(await update_artists(second.id, {"name": first.name}))
    response.assert_is("NameExistsError")
    assert response.message == "Another Artist with this name exists"


@pytest.mark.anyio
async def test_update_artist_fails_not_found(update_artists):
    response = Response(await update_artists(1, {"name": "updated artist"}))

    response.assert_is("IDNotFoundError")
    assert response.id == 1
    assert response.message == "Artist ID not found: '1'"


@pytest.mark.anyio
async def test_update_artists_cannot_bulk_edit_name(update_artists, gen_artist):
    first = await DB.add(next(gen_artist))
    second = await DB.add(next(gen_artist))

    response = Response(await update_artists([first.id, second.id], {"name": "unique"}))
    response.assert_is("InvalidParameterError")


@pytest.mark.parametrize(
    "empty",
    [
        None,
        "",
    ],
    ids=[
        "none",
        "empty string",
    ],
)
@pytest.mark.anyio
async def test_update_artist_fails_empty_parameter(update_artists, gen_artist, empty):
    artist = await DB.add(next(gen_artist))

    response = Response(await update_artists(artist.id, {"name": empty}))

    response.assert_is("InvalidParameterError")
    assert response.parameter == "name"
    assert response.message == "Invalid parameter 'name': cannot be empty"


@pytest.mark.anyio
async def test_update_artist_changes_updated_at(update_artists):
    original_artist = Artist(name="artist")
    original_artist.updated_at = dt(2023, 1, 1, tzinfo=timezone.utc)
    original_artist = await DB.add(original_artist)

    response = Response(await update_artists(original_artist.id, {"name": "updated"}))
    response.assert_is("UpdateSuccess")

    artist = await DB.get(Artist, original_artist.id)
    assert artist.updated_at > original_artist.updated_at