grunge_ishaan/grunge/tests/test_tracks.py

from uuid import UUID

from furl import furl
from rest_framework import status
from rest_framework.reverse import reverse as drf_reverse

from . import BaseAPITestCase


class TrackTests(BaseAPITestCase):
    def setUp(self):
        self.track_name = "Last Exit"
        self.track_uuid = UUID("b3083319-47a9-40ed-a4e0-a79d050d9df7")
        self.album_uuid = UUID("b4fee0db-0c93-4470-96b3-cebd158033a0")

    def test_list_tracks(self):
        url = drf_reverse("track-list", kwargs={"version": self.version})
        r = self.client.get(url)
        self.assertEqual(r.status_code, status.HTTP_200_OK)
        self.assertEqual(r.data["count"], 3695)

    def test_search_tracks(self):
        url = drf_reverse("track-list", kwargs={"version": self.version})
        url = furl(url).set({"name": self.track_name}).url
        r = self.client.get(url)
        self.assertEqual(r.status_code, status.HTTP_200_OK)
        self.assertEqual(r.data["count"], 4)
        self.assertEqual(r.data["results"][0]["uuid"], self.track_uuid)

    def test_get_track(self):
        url = drf_reverse(
            "track-detail", kwargs={"version": self.version, "uuid": self.track_uuid}
        )
        r = self.client.get(url)
        self.assertEqual(r.status_code, status.HTTP_200_OK)
        self.assertEqual(r.data["name"], self.track_name)
        self.assertEqual(r.data["album"]["uuid"], self.album_uuid)