tiddl/tests/test_utils.py

import unittest

from tiddl.models.resource import Track
from tiddl.utils import TidalResource, formatTrack


class TestTidalResource(unittest.TestCase):
    def test_resource_parsing(self):
        positive_cases = [
            ("https://tidal.com/browse/track/12345678", "track", "12345678"),
            ("track/12345678", "track", "12345678"),
            ("https://tidal.com/browse/album/12345678", "album", "12345678"),
            ("album/12345678", "album", "12345678"),
            ("https://tidal.com/browse/playlist/12345678", "playlist", "12345678"),
            ("playlist/12345678", "playlist", "12345678"),
            ("https://tidal.com/browse/artist/12345678", "artist", "12345678"),
            ("artist/12345678", "artist", "12345678"),
        ]

        for resource, expected_type, expected_id in positive_cases:
            with self.subTest(resource=resource):
                tidal_resource = TidalResource.fromString(resource)
                self.assertEqual(tidal_resource.type, expected_type)
                self.assertEqual(tidal_resource.id, expected_id)

    def test_failing_cases(self):
        failing_cases = [
            "https://tidal.com/browse/invalid/12345678",
            "invalid/12345678",
            "https://tidal.com/browse/track/invalid",
            "track/invalid",
            "",
            "invalid",
            "https://tidal.com/browse/track/",
            "track/",
            "/12345678",
        ]

        for resource in failing_cases:
            with self.subTest(resource=resource):
                with self.assertRaises(ValueError):
                    TidalResource.fromString(resource)


class TestFormatTrack(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        cls.track = Track(
            **{
                "id": 66421438,
                "title": "Shutdown",
                "duration": 189,
                "replayGain": -9.95,
                "peak": 0.966051,
                "allowStreaming": True,
                "streamReady": True,
                "adSupportedStreamReady": True,
                "djReady": True,
                "stemReady": False,
                "streamStartDate": "2016-11-15T00:00:00.000+0000",
                "premiumStreamingOnly": False,
                "trackNumber": 9,
                "volumeNumber": 1,
                "version": None,
                "popularity": 24,
                "copyright": "(P) 2016 Boy Better Know",
                "bpm": 69,
                "url": "http://www.tidal.com/track/66421438",
                "isrc": "GB7QY1500024",
                "editable": False,
                "explicit": True,
                "audioQuality": "LOSSLESS",
                "audioModes": ["STEREO"],
                "mediaMetadata": {"tags": ["LOSSLESS", "HIRES_LOSSLESS"]},
                "artist": {
                    "id": 3566984,
                    "name": "Skepta",
                    "type": "MAIN",
                    "picture": "747af850-fa9c-4178-a3e6-49259b67df86",
                },
                "artists": [
                    {
                        "id": 3566984,
                        "name": "Skepta",
                        "type": "MAIN",
                        "picture": "747af850-fa9c-4178-a3e6-49259b67df86",
                    }
                ],
                "album": {
                    "id": 66421429,
                    "title": "Konnichiwa",
                    "cover": "e0c2f05e-e21f-47c5-9c37-2993437df27d",
                    "vibrantColor": "#ae3b31",
                    "videoCover": None,
                },
                "mixes": {"TRACK_MIX": "001aa4abeb471e8f55f5784772b478"},
                "playlistNumber": None,
            }
        )

    def test_templating(self):
        test_cases = [
            ("{id}", "66421438"),
            ("{title}", "Shutdown"),
            ("{version}", ""),
            ("{artist}", "Skepta"),
            ("{artists}", "Skepta"),
            ("{album}", "Konnichiwa"),
            ("{number}", "9"),
            ("{disc}", "1"),
            ("{date:%m-%d-%y}", "11-15-16"),
            ("{date:%Y}", "2016"),
            ("{year}", "2016"),
            ("{playlist_number}", "0"),
            ("{playlist_number:02d}", "00"),
            ("{bpm}", "69"),
            ("{quality}", "high"),
            ("{artist}/{album}/{title}", "Skepta/Konnichiwa/Shutdown"),
            ("{number:02d}. {title}", "09. Shutdown"),
        ]

        for template, expected_result in test_cases:
            with self.subTest(template=template, expected_result=expected_result):
                result = formatTrack(template, self.track)
                self.assertEqual(result, expected_result)

    def test_invalid_characters(self):
        test_cases = ["\\", ":", '"', "?", "<", ">", "|", "{number}:{title}", "{date}"]

        for template in test_cases:
            with self.subTest(template=template):
                with self.assertRaises(ValueError):
                    formatTrack(template, self.track)


if __name__ == "__main__":
    unittest.main()