"""Tests for top_sound_recordings_schema module."""

from decimal import Decimal

from analytics.schemas import top_sound_recordings


def test_top_sound_recordings_schema():
    """Test that top_sound_recordings_schema converts to JSON object."""
    mock_model_response_message = {
        "items": [
            {
                "isrc": "USA561066580",
                "streams": 500000,
                "growth_percentage": Decimal("0.500000"),
            },
            {
                "isrc": "USJMZ1800003",
                "streams": 400000,
                "growth_percentage": Decimal("16.300000"),
            },
        ]
    }

    expected_result = {
        "items": [
            {"isrc": "USA561066580", "growth_percentage": 0.500000, "streams": 500000},
            {"isrc": "USJMZ1800003", "growth_percentage": 16.300000, "streams": 400000},
        ]
    }

    result = top_sound_recordings.TopSoundRecordingsSchema().dump(
        mock_model_response_message
    )
    assert result == expected_result