"""Tests for top_markets_schema module."""

from analytics.schemas import top_countries_videos


def test_top_countries_videos_schema():
    """Test top_countries_videos_schema converts to JSON object."""
    mock_model_response = {
        "video_id": "TEST",
        "top_countries_videos": [
            {"country_code": "US", "views": 666},
            {"country_code": "PH", "views": 66},
            {"country_code": "MX", "views": 6},
        ],
        "sources": [
            {"id": 453, "name": "YouTube"},
        ],
    }

    expected_result = {
        "video_id": "TEST",
        "top_countries_videos": [
            {"country_code": "US", "views": 666},
            {"country_code": "PH", "views": 66},
            {"country_code": "MX", "views": 6},
        ],
        "sources": [
            {"id": 453, "name": "YouTube"},
        ],
    }

    result = top_countries_videos.TopCountriesVideosSchema().dump(mock_model_response)
    assert result == expected_result