"""Tests for streams module."""

from analytics.schemas import streams

STREAMS = [
    {
        "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,
    },
]


def test_streams_all_schema():
    """Test that streams_all schema converts to JSON object."""
    input_data = {"isrc": "TEST", "items": STREAMS}

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

    result = streams.StreamsAllSchema().dump(input_data)

    assert result == expected_result