from unittest import mock

import pytest

from playlist.queries.fetch_queries import fetch_placement_metrics_by_country

store_id = "1"
store_playlist_id = "pl.df3f10ca27b1479087de2cd3f9f6716b"
isrc = "QZK6M1903348"


def test_minimal(mock_execute):
    query_params = {
        "store_id": store_id,
        "store_playlist_id": store_playlist_id,
        "isrc": isrc,
    }

    permissions = {
        "permission_label_ids": ["1", "2", "3"],
    }

    fetch_placement_metrics_by_country(query_params, permissions)
    assert mock_execute.call_count == 1


@pytest.mark.disable_mock_execute
def test_countries(mock_template_engine):
    query_params = {
        "store_id": store_id,
        "store_playlist_id": store_playlist_id,
        "isrc": isrc,
        "stream_countries": ["GB", "DE"],
    }
    permissions = {
        "permission_label_ids": ["1", "2", "3"],
    }
    fetch_placement_metrics_by_country(query_params, permissions)
    db_call_args = mock_template_engine.prepare_query.call_args[0][1]
    assert db_call_args["stream_countries"] == ["GB", "DE"]