from resonance_engine.dsp.enums import DSPClientName
from resonance_engine.tasks.utils import queue_url_for

from tests.unit.helpers import override_settings


def test_returns_configured_queue_url_per_client() -> None:
    with override_settings(
        sqs_fan_collect_spotify_songwhip_queue_url="https://sqs/songwhip",
        sqs_fan_collect_spotify_smf_sme_queue_url="https://sqs/sme",
        sqs_fan_collect_spotify_smf_orch_queue_url="https://sqs/orch",
    ):
        assert queue_url_for(DSPClientName.spotify_songwhip) == "https://sqs/songwhip"
        assert queue_url_for(DSPClientName.spotify_smf_sme) == "https://sqs/sme"
        assert queue_url_for(DSPClientName.spotify_smf_orch) == "https://sqs/orch"


def test_returns_none_when_queue_url_unset() -> None:
    with override_settings(sqs_fan_collect_spotify_smf_sme_queue_url=""):
        assert queue_url_for(DSPClientName.spotify_smf_sme) is None