"""Tests for GenreSubgenreMapping class."""

from unittest.mock import patch

from feed_sender.util.genre_subgenre_mapping import GenreSubgenreMapping


def test_genre_subgenre_mapping():
    """Test _genre_subgenre_mapping."""
    genre_subgenre_mapping = {
        '1_4': {'subgenre': 'Sub Rock', 'genre': 'Rock'}}
    store_id = 10
    mock_results = [
        {
            'genre_id': 1, 'subgenre_id': 4, 'dms_master_genre': 'Rock',
            'dms_master_subgenre': 'Sub Rock'
        }
    ]
    with patch(
            'feed_sender.util.genre_subgenre_mapping.'
            'mysql.fetchall', return_value=mock_results):
        singleton = GenreSubgenreMapping()
        actual = singleton.fetch_genre_mapping(store_id)
        assert actual == genre_subgenre_mapping