"""Unit tests for Spotify config."""
from unittest import mock

import pytest

from feed_ingestion.flows.spotify import config


@pytest.mark.parametrize(
    'licensor, expected', [
        ['sme', 'common_value'],
        ['theorchard', 'common_value'],
        ['altafonte', 'altafonte_specific'],
    ]
)
def test__get_environ(licensor, expected):
    environ_mock = {
        'SPOTIFY_API_CLIENT_ID': 'common_value',
        'SPOTIFY_API_CLIENT_ID_ALTAFONTE': 'altafonte_specific',
    }
    with mock.patch.dict(config.os.environ, environ_mock):
        value = config._get_environ(
            variable='SPOTIFY_API_CLIENT_ID',
            licensor=licensor
        )
        assert value == expected