"""Tests for track role."""

from src.constants import service
from src.logic.track_role import _include_existing_track_role_fields


def test_include_existing_track_role_fields(request_engine):
    """Test include existing track role fields."""
    request_engine[service.OWS_TRACK].add_spec(
        'GET', '/tracks',
        status=200,
        response={'items': [
            {'tuid': 1, 'artists': [{'foo': 'bar'}]},
            {'tuid': 2, 'artists': [{'foo': 'baz'}]}
        ]}
    )
    mock_output = {'1': {}, '2': {}}
    _include_existing_track_role_fields(mock_output)
    assert mock_output == {
        '1': {'artists_tag': [{'foo': 'bar'}]},
        '2': {'artists_tag': [{'foo': 'baz'}]}
    }