"""Unit tests for the track artist model."""


from artist.models import track_artist


def test_to_dict_properties():
    """Test that the expected properties are being returned."""
    new_track_artist = track_artist.TrackArtist(
        id=1, track_id=5, artist_name='Bob',
        role='Singer')

    assert new_track_artist.to_dict() == {
        'id': new_track_artist.id,
        'name': new_track_artist.artist_name,
        'role': new_track_artist.role,
        'track_id': new_track_artist.track_id
    }