"""Tests for streams_by_sos_schema module."""

from sound_recordings.schemas import streams_by_sos

SOURCE_OF_STREAMS = [
    {
        "source": "active",
        "items": [
            {"date": "2021-01-16T00:00:00Z", "streams": 1000},
            {"date": "2021-01-17T00:00:00Z", "streams": 500},
        ],
    },
    {
        "source": "passive",
        "items": [
            {"date": "2021-01-16T00:00:00Z", "streams": 1500},
            {"date": "2021-01-17T00:00:00Z", "streams": 300},
        ],
    },
    {
        "source": "collection",
        "items": [{"date": "2021-01-16T00:00:00Z", "streams": 2500}],
    },
    {"source": "unknown", "items": [{"date": "2021-01-16T00:00:00Z", "streams": 4310}]},
]


def test_streams_by_sos_schema():
    """Test that streams_by_sos schema serializes correctly."""
    input_data = {"isrc": "TEST", "source_of_streams": SOURCE_OF_STREAMS}

    expected_result = {"isrc": "TEST", "source_of_streams": SOURCE_OF_STREAMS}

    result = streams_by_sos.StreamsBySosSchema().dump(input_data)

    assert result == expected_result