"""Tests for Apple Music model."""

from participant.models.service import apple_music


def test_artist_search(mocker):
    """Test artist search."""
    apple_music_mock = mocker.patch(
        'participant.models.service.apple_music._apple_music_client.search_artist'
    )
    apple_music_search_response_mock = apple_music_mock.return_value

    query, limit, offset, localization = 'Danger Mouse', 10, 0, 'fr'
    result = apple_music.search_artist(query, limit, offset, localization)

    assert result == apple_music_search_response_mock
    apple_music_mock.assert_called_once_with(
        query=query, limit=limit, offset=offset, localization=localization
    )


def test_get_artist_by_id(mocker):
    """Test get artist by ID."""
    apple_music_mock = mocker.patch(
        'participant.models.service.apple_music._apple_music_client.get_artist_by_id'
    )
    apple_music_search_response_mock = apple_music_mock.return_value

    artist_id = 'ARTIST ID'
    result = apple_music.get_artist_by_id(artist_id)

    assert result == apple_music_search_response_mock
    apple_music_mock.assert_called_once_with(artist_id)