"""Test genre mapping."""
from unittest.mock import MagicMock

from ddex_ingester_common.lambda_exceptions import InvalidGenreException
from ddex_ingester_common.utils import genre_mapping
import pytest


def test_genre_mapping_exists():
    """Test genre mapping exists."""
    assert genre_mapping.get_orchard_genre_mapping_for_itunes('AFRICA-00') == {
        'genre_id': 11,
        'subgenre_id': 548
    }


def test_genre_mapping_not_exist():
    """Test genre mapping not exist."""
    assert not genre_mapping.get_orchard_genre_mapping_for_itunes('blah')


def test_orchard_genre_mapping_not_exist_exception():
    """Test orchard genre mapping raises exception genre does not exist."""
    with pytest.raises(InvalidGenreException):
        genre_mapping.get_orchard_genre_mapping(
            MagicMock(),
            '',
            '',
            raise_exception=True
        )


def test_orchard_genre_mapping_not_exist_no_exception():
    """Test orchard genre mapping raise exception flag works."""
    genre_mapping.get_orchard_genre_mapping(
        MagicMock(),
        '',
        '',
        raise_exception=False
    )