from server.dna.constants import ARTIST, TRACK
from tests.constants import UNKNOWN_FIELD


ENTITY_ID = "58857"

RECENT_SEARCH_ARTIST_BODY = {
    "entity_id": ENTITY_ID,
    "entity_type": ARTIST,
}

RECENT_SEARCH_TRACK_BODY = {"entity_id": ENTITY_ID, "entity_type": TRACK}

INVALID_REQUEST_BODY_1 = {"entity_id": 58857, "entity_type": "test"}

INVALID_REQUEST_BODY_2 = {"entity_id": "test", "entity_type": 123}

INVALID_REQUEST_BODY_3 = {}

INVALID_REQUEST_BODY_4 = {**RECENT_SEARCH_ARTIST_BODY, "test": "test"}

INVALID_REQUEST_BODY_5 = {"entity_type": "", "entity_id": ""}

EXPECTED_RETURN_DATA_1 = {"entity_type": ["Must be one of: artist, track."], "entity_id": ["Not a valid string."]}

EXPECTED_RETURN_DATA_2 = {"entity_type": ["Not a valid string."]}

EXPECTED_RETURN_DATA_3 = {
    "entity_type": ["Missing data for required field."],
    "entity_id": ["Missing data for required field."],
}

EXPECTED_RETURN_DATA_4 = {"test": UNKNOWN_FIELD}

EXPECTED_RETURN_DATA_5 = {
    "entity_id": ["Length must be between 1 and 255."],
    "entity_type": ["Must be one of: artist, track."],
}

RECENT_SEARCH_EXPECTED_RETURN_DATA = [
    {
        "entity_type": "track",
        "entity_id": "13692288",
        "data": {
            "name": "Kaleido Scope",
            "artwork_url": "https://images-api.atlas.stream/v2/albums/by_spotify_track_id/2fdJYjj21WWSjBGtz63OXY",
            "artwork_url_alt": "https://i.scdn.co/image/9a55f5d67616387ce52d22b443d5b34ed8d5d15b",
            "artist_name": "Ugress",
            "score": 1.0,
        },
    },
    {
        "entity_type": "artist",
        "entity_id": "738253",
        "data": {
            "artwork_url": "https://images-api.atlas.stream/v2/artists/by_spotify_id/5FRM46QLXYjCJb19bYq3qu",
            "artwork_url_alt": "https://i.scdn.co/image/ab67616100005174d1e8f74ef35ba18d8009222d",
            "name": "Zach Diamond",
            "score": 1.0,
        },
    },
]

ARTIST_RECENT_SEARCH_EXPECTED_RETURN_DATA = [
    {
        "entity_type": "artist",
        "entity_id": "738253",
        "data": {
            "artwork_url": "https://images-api.atlas.stream/v2/artists/by_spotify_id/5FRM46QLXYjCJb19bYq3qu",
            "artwork_url_alt": "https://i.scdn.co/image/ab67616100005174d1e8f74ef35ba18d8009222d",
            "name": "Zach Diamond",
            "score": 1.0,
        },
    },
]

TRACK_RECENT_SEARCH_EXPECTED_RETURN_DATA = [
    {
        "entity_type": "track",
        "entity_id": "13692288",
        "data": {
            "name": "Kaleido Scope",
            "artwork_url": "https://images-api.atlas.stream/v2/albums/by_spotify_track_id/2fdJYjj21WWSjBGtz63OXY",
            "artwork_url_alt": "https://i.scdn.co/image/9a55f5d67616387ce52d22b443d5b34ed8d5d15b",
            "artist_name": "Ugress",
            "score": 1.0,
        },
    },
]


REQUEST_BODIES = [
    {
        "entity_id": "738253",
        "entity_type": "artist",
    },
    {
        "entity_id": "13692288",
        "entity_type": "track",
    },
    {
        "entity_id": "1110120",
        "entity_type": "artist",
    },
    {
        "entity_id": "1141830",
        "entity_type": "artist",
    },
    {
        "entity_id": "738253",
        "entity_type": "artist",
    },
    {
        "entity_id": "13496071",
        "entity_type": "track",
    },
    {
        "entity_id": "11160650",
        "entity_type": "track",
    },
    {
        "entity_id": "13496071",
        "entity_type": "track",
    },
    {
        "entity_id": "13692288",
        "entity_type": "track",
    },
    {
        "entity_id": "738253",
        "entity_type": "artist",
    },
    {
        "entity_id": "11160650",
        "entity_type": "track",
    },
]


UNIQUE_ENTITIES_EXPECTED_RETURN_DATA = [
    {
        "entity_type": "track",
        "data": {
            "name": "The Day Doesn't Die",
            "artwork_url": "https://images-api.atlas.stream/v2/albums/by_spotify_track_id/77SEuXiZPAUKqmwifOgSYf",
            "artwork_url_alt": "https://i.scdn.co/image/74f51993ee3d323a96e5c8101e6ef31f0710f7f4",
            "artist_name": "Classified",
            "score": 1.0,
        },
        "entity_id": "11160650",
    },
    {
        "entity_type": "artist",
        "data": {
            "artwork_url": "https://images-api.atlas.stream/v2/artists/by_spotify_id/5FRM46QLXYjCJb19bYq3qu",
            "artwork_url_alt": "https://i.scdn.co/image/ab67616100005174d1e8f74ef35ba18d8009222d",
            "name": "Zach Diamond",
            "score": 1.0,
        },
        "entity_id": "738253",
    },
    {
        "entity_type": "track",
        "data": {
            "name": "Kaleido Scope",
            "artwork_url": "https://images-api.atlas.stream/v2/albums/by_spotify_track_id/2fdJYjj21WWSjBGtz63OXY",
            "artwork_url_alt": "https://i.scdn.co/image/9a55f5d67616387ce52d22b443d5b34ed8d5d15b",
            "artist_name": "Ugress",
            "score": 1.0,
        },
        "entity_id": "13692288",
    },
    {
        "entity_type": "track",
        "data": {
            "name": "My Love",
            "artwork_url": "https://images-api.atlas.stream/v2/albums/by_spotify_track_id/0uEftEHpAjuL2V6NScCs4n",
            "artwork_url_alt": "https://i.scdn.co/image/26f2927cc3ed5951b9dd32578183d25d7a16940d",
            "artist_name": "Kovacs",
            "score": 1.0,
        },
        "entity_id": "13496071",
    },
    {
        "entity_type": "artist",
        "data": {
            "artwork_url": "https://images-api.atlas.stream/v2/artists/by_spotify_id/4xDRor7W1DuI5mk0lvE9w1",
            "artwork_url_alt": "https://i.scdn.co/image/ab67616100005174f90979564a723a920af54657",
            "name": "The Catacombs",
            "score": 1.0,
        },
        "entity_id": "1141830",
    },
    {
        "entity_type": "artist",
        "data": {
            "artwork_url": "https://images-api.atlas.stream/v2/artists/by_spotify_id/0cRKBhWUTEtR1vmA06kVKz",
            "artwork_url_alt": "https://i.scdn.co/image/ab67616100005174f62c9110e551a6d6f5b34a2b",
            "name": "NEZZA",
            "score": 1.0,
        },
        "entity_id": "1110120",
    },
]