"""Pytest conftest module for create_tracks unit tests.""" import pytest @pytest.fixture def context_event(): """State machine input event.""" return { 'correlation_id': 'test-correlation-id', 'product': { 'upc': '231232563123', 'product_id': 234324, 'product_name': 'Test Product', }, 'project': { 'name': 'Test Project', 'project_code': '1', 'project_id': 4498666, }, 'tracks': [ { 'isrc': 'QZCDB2000001', 'track_name': 'Track One', 'tuid': 45677, 'sequence_number': 1, 'volume': 1, 'explicit': 'C', 'version': 'Deluxe', 'display_artists': [ { 'name': 'Artist One', 'spotify_uri': 'spotify:artist:111', 'apple_id': 'apple_111', }, ], }, { 'isrc': 'QZCDB2000002', 'track_name': 'Track Two', 'tuid': 45678, 'sequence_number': 2, 'volume': 1, 'explicit': 'C', 'version': None, 'display_artists': [], }, { 'isrc': 'QZCDB2000003', 'track_name': 'Track Three', 'tuid': None, 'sequence_number': 3, 'volume': 1, 'explicit': 'E', 'version': 'Remix', 'display_artists': [], }, ], 'label_participants': [], } @pytest.fixture def orchard_tracks(): """Sample orchard tracks response.""" return [ {'isrc': 'QZCDB2000001', 'tuid': '45677'}, {'isrc': 'QZCDB2000002', 'tuid': '45678'}, {'isrc': 'QZCDB2000004', 'tuid': '45680'}, ] @pytest.fixture def get_tracks_response(orchard_tracks): """Response from productByUpc with tracks.""" return { 'data': { 'productByUpc': { 'tracks': orchard_tracks } } } @pytest.fixture def get_tracks_empty_response(): """Response from productByUpc when no product found.""" return { 'data': { 'productByUpc': None } } @pytest.fixture def save_tracks_response(): """Response from saveTracks mutation.""" return { 'data': { 'saveTracks': [ {'isrc': 'QZCDB2000003', 'tuid': '45679'}, ] } }