from tests.conftest import MOCK_DATE_RANGE, MOCK_ISRCS, MOCK_PLAYLIST_IDS CONTAINER_ID = '1111111' COUNTRY_CODE = 'us' START_DATE = MOCK_DATE_RANGE[0] END_DATE = MOCK_DATE_RANGE[-1] INT_VAL = 7 STR_VAL = 'testval' MOCK_ISRC1 = MOCK_ISRCS[0] MOCK_ISRC2 = MOCK_ISRCS[1] PLAYLIST_ID1 = MOCK_PLAYLIST_IDS[0] PLAYLIST_ID2 = MOCK_PLAYLIST_IDS[1] MOCK_TOKEN_RESPONSE = { 'access_token': 'fake-access-token', 'expires_at': 1572029502, 'expires_in': 86400, 'token_type': 'Bearer' } PARAM_CONTAINER_ID = {'container_id': CONTAINER_ID} PARAM_COUNTRY_CODE = {'country_code': COUNTRY_CODE} PARAM_END_DATE = {'end_date': END_DATE} PARAM_ISRC1 = {'isrc': MOCK_ISRC1} PARAM_ISRC_ARRAY = {'isrc': [MOCK_ISRC1, MOCK_ISRC2]} PARAM_PLAYLIST_ID = {'playlist_id': PLAYLIST_ID1} PARAM_PLAYLIST_ARRAY = {'playlist_id': [PLAYLIST_ID1, PLAYLIST_ID2]} PARAM_START_DATE = {'start_date': START_DATE} PARAM_GROUP_BY_DATE = {'group_by': 'date'} PARAMS_START_END_ISRC = {**PARAM_START_DATE, **PARAM_END_DATE, **PARAM_ISRC1} PARAMS_START_END_PLAYLIST_ARRAY = {**PARAM_START_DATE, **PARAM_END_DATE, **PARAM_PLAYLIST_ARRAY} PARAMS_START_END_ISRC_ARRAY = {**PARAM_START_DATE, **PARAM_END_DATE, **PARAM_ISRC_ARRAY} #: Format is { route: dict of query params } AMAZON_ROUTES = [ { 'route': '/v2/amazon/tracks/streams', 'params': {**PARAMS_START_END_ISRC_ARRAY, **PARAM_COUNTRY_CODE, **PARAM_GROUP_BY_DATE} }, { 'route': '/v2/amazon/tracks/streams', 'params': {**PARAMS_START_END_ISRC_ARRAY} # no country or group_by }, { 'route': '/v2/amazon/tracks/streams', 'params': {**PARAMS_START_END_ISRC_ARRAY, **PARAM_COUNTRY_CODE} }, { 'route': '/v2/amazon/playlist/tracks/streams', 'params': { **PARAMS_START_END_ISRC_ARRAY, **PARAM_PLAYLIST_ID, **PARAM_COUNTRY_CODE, **PARAM_GROUP_BY_DATE } }, { 'route': '/v2/amazon/playlist/tracks/streams', 'params': {**PARAMS_START_END_ISRC_ARRAY, **PARAM_PLAYLIST_ID} # no country or group_by }, { 'route': '/v2/amazon/playlist/tracks/streams', 'params': {**PARAMS_START_END_ISRC_ARRAY, **PARAM_PLAYLIST_ID, **PARAM_COUNTRY_CODE} }, { 'route': '/v2/amazon/playlists/streams', 'params': {**PARAMS_START_END_PLAYLIST_ARRAY, **PARAM_COUNTRY_CODE, **PARAM_GROUP_BY_DATE} }, { 'route': '/v2/amazon/playlists/streams', 'params': {**PARAMS_START_END_PLAYLIST_ARRAY} # no country or group_by }, { 'route': '/v2/amazon/playlists/streams', 'params': {**PARAMS_START_END_PLAYLIST_ARRAY, **PARAM_COUNTRY_CODE} }, ] APPLE_ROUTES = [ { 'route': '/v2/apple/tracks/streams', 'params': {**PARAMS_START_END_ISRC_ARRAY, **PARAM_COUNTRY_CODE, **PARAM_GROUP_BY_DATE} }, { 'route': '/v2/apple/tracks/streams', 'params': {**PARAMS_START_END_ISRC_ARRAY} # no country or group_by }, { 'route': '/v2/apple/tracks/streams', 'params': {**PARAMS_START_END_ISRC_ARRAY, **PARAM_COUNTRY_CODE} }, ] SPOTIFY_ROUTES = [ { 'route': '/v2/spotify/tracks/streams', 'params': {**PARAMS_START_END_ISRC_ARRAY, **PARAM_COUNTRY_CODE, **PARAM_GROUP_BY_DATE} }, { 'route': '/v2/spotify/tracks/streams', 'params': {**PARAMS_START_END_ISRC_ARRAY} # no country or group_by }, { 'route': '/v2/spotify/tracks/streams', 'params': {**PARAMS_START_END_ISRC_ARRAY, **PARAM_COUNTRY_CODE} }, ] YOUTUBE_ROUTES = [ { 'route': '/v2/youtube/tracks/streams', 'params': {**PARAMS_START_END_ISRC_ARRAY, **PARAM_COUNTRY_CODE, **PARAM_GROUP_BY_DATE} }, { 'route': '/v2/youtube/tracks/streams', 'params': {**PARAMS_START_END_ISRC_ARRAY} # no country or group_by }, { 'route': '/v2/youtube/tracks/streams', 'params': {**PARAMS_START_END_ISRC_ARRAY, **PARAM_COUNTRY_CODE} }, ] HEALTH_ROUTES = [ { 'route': '/health', 'params': {} }, { 'route': '/v2/health', 'params': {} }, ] SECRETS_MANAGER_CLIENTS = { 'test_client_id_1': 'Test Client Name 1', 'test_client_id_2': 'Test Client Name 2', 'test_client_id_3': 'Test Client Name 3', 'test_client_id_4': 'Test Client Name 4', }