import pytest
from unittest.mock import AsyncMock

from server.clients.rti_api import RtiApiClient


@pytest.mark.parametrize(
    "data",
    (
            {"data": [{"isrc": "USSM19902991", "artistId": "GRAS_4867"}, {"isrc": "USSM12001261", "artistId": "GRAS_1689004"}]},
            {"data": []},
    ),
)
async def test_get_rti_milestones_v2(data, mocker, db_session, client, auth):
    rti_mock = mocker.patch.object(RtiApiClient, "_send_request", side_effect=AsyncMock(return_value=data))
    response = await client.get("/api/v1/joint/favorites/rti/milestone/v2/", headers=auth)
    assert response.status == 200
    assert rti_mock.call_count == 2
    response_data = await response.json()
    assert response_data == data