from unittest.mock import patch

from tests import BaseTestCase, mock_for_user
from tests.projects.fixtures.product_families_fixture import fixture

user = mock_for_user(1, "dev-marketing-user-1@example.com", "sme-dna|5cf0091d6c91d310fe6fcdc6")


@patch("auth.atlas.atlas_token_decoder.jwt.decode", user.jwt_decode)
class TestProductFamilies(BaseTestCase):
    fixture = fixture

    maxDiff = None

    @staticmethod
    def get_url(project_id):
        return "/projects/{}/product-families".format(project_id)

    def test_product_families_list_api(self):
        response = self.execute_get(self.get_url(1), headers=self.headers)

        self.assertEqual(200, response.status_code)
        self.assertEqual(
            response.json,
            [
                {
                    "id": 1,
                    "releaseDate": "2002-01-01",
                    "externalId": "1",
                    "title": "Test Family 1",
                    "tracks": [
                        {
                            "id": 1,
                            "isrc": "FR7U62000001",
                            "releaseDate": "2002-01-01",
                            "supplementary": "Album Version",
                            "title": "Test Track 1",
                        }
                    ],
                },
                {
                    "id": 2,
                    "releaseDate": "2002-02-01",
                    "title": "Test Family 2",
                    "externalId": "2",
                    "tracks": [
                        {
                            "id": 2,
                            "isrc": "DEE860900167",
                            "releaseDate": "2002-02-01",
                            "supplementary": None,
                            "title": "Test Track 2",
                        },
                        {
                            "id": 3,
                            "isrc": "USSM10010637",
                            "releaseDate": "2002-02-01",
                            "supplementary": None,
                            "title": "Test Track 3",
                        },
                    ],
                },
            ],
        )