"""Tests for utils."""

import pytest

from src import utils


@pytest.mark.parametrize('spotify_url, spotify_id', [
    ('https://open.spotify.com/artist/165ZgPlLkK7bf5bDoFc6Sb', '165ZgPlLkK7bf5bDoFc6Sb'),
    ('https://open.spotify.com/artist/165ZgPlLkK7bf5bDoFc6Sb/', '165ZgPlLkK7bf5bDoFc6Sb'),
    ('https://open.spotify.com/playlist/37i9dQZF1DWSyORK2fDImV', None),
    ('spotify:artist:165ZgPlLkK7bf5bDoFc6Sb', '165ZgPlLkK7bf5bDoFc6Sb'),
    ('spotify:playlist:37i9dQZF1DWSyORK2fDImV', None),
    ('https://open.spotify.com/artist/0w7cJQGzRiAIVIW8OMeCGL?si=fy_lY-vBQhGAGeoNuh0iZg&dl_branch=1',
     '0w7cJQGzRiAIVIW8OMeCGL')
])
def test_get_spotify_id(spotify_url, spotify_id):
    """Test parsing the spotify ID from Spotify URL."""
    assert utils.get_spotify_id(spotify_url) == spotify_id