"""Tests for top_accounts_metrics schema."""

from analytics.schemas import top_accounts_metrics


def test_top_accounts_metrics():
    """Test that top_accounts_metrics converts to JSON object."""
    mock_model_response_message = {
        "items": [
            {
                "label_id": 1100,
                "subaccount_id": None,
                "label_manager_name": "Label Manager Name",
                "streams_28_days": 10,
                "streams_growth_percentage_28_day": 20,
                "track_downloads_28_days": 5,
                "track_downloads_growth_percentage_28_day": 6,
                "album_downloads_28_days": 7,
                "album_downloads_growth_percentage_28_day": 8,
            },
            {
                "label_id": 1100,
                "subaccount_id": 1101,
                "label_manager_name": "Subaccount Manager Name",
                "streams_28_days": 10,
                "streams_growth_percentage_28_day": 20,
                "track_downloads_28_days": 5,
                "track_downloads_growth_percentage_28_day": 6,
                "album_downloads_28_days": 7,
                "album_downloads_growth_percentage_28_day": 8,
            },
        ]
    }

    expected_result = {
        "items": [
            {
                "label_id": 1100,
                "subaccount_id": None,
                "label_manager_name": "Label Manager Name",
                "streams_28_days": 10,
                "streams_growth_percentage_28_day": 20,
                "track_downloads_28_days": 5,
                "track_downloads_growth_percentage_28_day": 6,
                "album_downloads_28_days": 7,
                "album_downloads_growth_percentage_28_day": 8,
            },
            {
                "label_id": 1100,
                "subaccount_id": 1101,
                "label_manager_name": "Subaccount Manager Name",
                "streams_28_days": 10,
                "streams_growth_percentage_28_day": 20,
                "track_downloads_28_days": 5,
                "track_downloads_growth_percentage_28_day": 6,
                "album_downloads_28_days": 7,
                "album_downloads_growth_percentage_28_day": 8,
            },
        ]
    }

    result = top_accounts_metrics.TopAccountsMetricsSchema().dump(
        mock_model_response_message
    )
    assert result == expected_result