"""Tests for streams_by_store_schema module."""

from analytics.schemas import streams_by_store

STREAMS = [
    {
        "id": 1,
        "name": "Apple Music",
        "items": [
            {
                "date": "2021-01-16T00:00:00Z",
                "streams": 1000,
                "skip_rate": 0.325823532,
                "saves": 500,
            },
            {
                "date": "2021-01-17T00:00:00Z",
                "streams": 500,
                "skip_rate": 0.2597351,
                "saves": 125,
            },
        ],
    },
    {
        "id": 286,
        "name": "Spotify",
        "items": [
            {
                "date": "2021-01-16T00:00:00Z",
                "streams": 1500,
                "skip_rate": 0.421415667,
                "saves": 700,
            },
            {
                "date": "2021-01-17T00:00:00Z",
                "streams": 300,
                "skip_rate": 0.1245356,
                "saves": 100,
            },
        ],
    },
]


def test_streams_by_store_schema():
    """Test that streams_by_store schema converts to JSON object."""
    input_data = {"isrc": "TEST", "stores": STREAMS}

    expected_result = {"isrc": "TEST", "stores": STREAMS}

    result = streams_by_store.StreamsByStoreSchema().dump(input_data)

    assert result == expected_result