1 files changed, 278 insertions, 0 deletions
diff --git a/tests/api/test_artist.py b/tests/api/test_artist.py
new file mode 100644
index 0000000..8cb2f1a
--- /dev/null
+++ b/tests/api/test_artist.py
@@ -0,0 +1,278 @@
+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