# import logging # # import pandas # import pytest # from ui_automation_framework.utils import logger as cl # from ui_automation_framework.utils import CoreConfig # # from app.api_client import ApiClient # # log = cl.Logger(logging.DEBUG) # # new_env = 'https://test-proxy.apollo.stream' # old_env = 'https://proxy.apollo.stream' # api_client_old = ApiClient(host_endpoint=old_env) # api_client_new = ApiClient(host_endpoint=new_env) # token_new = api_client_new.authorize() # prod_user = 'anna.sheremet@dataart.com' # request_url = 'Request URL' # # # def f(dat, c='red'): # colors = [] # for i in dat: # if i: # c = 'green' # elif i == '': # c = 'white' # else: # c = 'red' # colors.append(f'background-color: {c}') # return colors # # # charts_stat_keys = ['isrc', 'peak_position', 'entry_date'] # markets = ['global'] # vendors = ['apple', 'spotify'] # # # @pytest.mark.parametrize("vendor", vendors) # @pytest.mark.parametrize("market", markets) # def test_my_starred_tracks_e_1(market, vendor): # old_endpoint = f'/apollo-api/charts/stats/?isrc=USSM12105970&isrc=USSM12103949&isrc=QMFME2066837&isrc=USQX91901353&isrc=NOEPO2000028&isrc=RUA012100417&isrc=USUG12101245&vendor={vendor}' # new_endpoint = f'/gate-api/charts/stats/?isrc=USSM12105970&isrc=USSM12103949&isrc=QMFME2066837&isrc=USQX91901353&i&isrc=NOEPO2000028&isrc=RUA012100417&isrc=USUG12101245&vendor={vendor}' # # get old response # old_resp = api_client_old.get_response(endpoint=old_endpoint, user=prod_user, env=old_env, token=token_old)['body'] # new_resp = api_client_old.get_response(endpoint=new_endpoint, user=CoreConfig.USER_EMAIL, env=new_env)['body'] # # old_resp = sorted(old_resp, key=lambda d: d['isrc']) # new_resp = sorted(new_resp, key=lambda d: d['isrc']) # log.info(f'new resp 1: {new_resp}') # log.info(f'old resp 1: {old_resp}') # # indexes_all = [request_url] + charts_stat_keys * len(old_resp) # log.info(f'len: {len(old_resp)}, \nlen: {len(new_resp)}') # log.info(f'indexes all: {indexes_all}') # old_endpoint_values = [] # new_endpoint_values = [] # diffs = [] # # old_endpoint_values.append(old_endpoint) # new_endpoint_values.append(new_endpoint) # diffs.append('') # # for to, tn in zip(old_resp, new_resp): # diffs.extend([to[k] == tn[k] for k in charts_stat_keys]) # old_endpoint_values.extend([to[k] for k in charts_stat_keys]) # new_endpoint_values.extend([tn[k] for k in charts_stat_keys]) # # df = pandas.DataFrame({ # 'IS CORRECT': diffs, # 'OLD API VALUE': old_endpoint_values, # 'NEW API VALUE': new_endpoint_values, # }, # index=indexes_all) # # df.reset_index().style.apply(f, subset=['IS CORRECT']) \ # .to_excel(f'my_starred_tracks_{vendor}.xlsx')