from playlist.queries.fetch_queries import PlacementsCountQuery
from tests.unit.utils import raises_validation_error


def test_invalid_isrc():
    with raises_validation_error({"isrc": ["String does not match expected pattern."]}):
        PlacementsCountQuery({"isrc": "AHHHHH"})


def test_invalid_keys_are_deleted():
    q = PlacementsCountQuery({"isrc": "QM6N22019571", "a": 1, "b": 2})

    assert "a" not in q.query_params
    assert "b" not in q.query_params
    assert "isrc" in q.query_params


def test_invalid_without_isrc():
    with raises_validation_error(
        {"_schema": ["isrc, or global_participant_id, or product_id required"]}
    ):
        PlacementsCountQuery({})