"""Mocks for ows-sound-recordings responses."""


def mock_sound_recording(mock_tuid, mock_id=None):
    """Generate a sound recording wrapper object."""
    return {
        'id': mock_id or '463c4d20-c9d4-403b-8e7d-7b3a0272fde5',
        'isrc': 'USA370646506',
        'primary_track_id': mock_tuid,
        'assets': []
    }


def mock_asset(mock_track):
    """Generate a mock asset object."""
    return {
        'asset_id': 'ef719ed2-d402-4390-837d-33f4042514b0',
        'acr_id': 'd4aed4e91da0875d78afbd9e0a4042b3',
        'filename': '56bf4415_b210_4810_bfd4_f16f2372b13e',
        'extension': 'flac',
        'tuids': [mock_track['tuid']],
        'tracks': [mock_track],
        'source': 'ows-assets-db'
    }


def mock_track_match(main_track, other_tracks=None, osr_id=None):
    """Generate a mocked track match."""
    sr = mock_sound_recording(main_track['tuid'], osr_id)
    sr['assets'].append(mock_asset(main_track))
    if other_tracks:
        for other_track in other_tracks:
            sr['assets'].append(mock_asset(other_track))
    return sr