import logging
import unittest

import allure
import pytest
from assertpy import assert_that
from parameterized import parameterized
from ui_automation_framework.core import TestStatus
from ui_automation_framework.utils import AppConfig
from ui_automation_framework.utils import logger as cl

from app.api_client import ApiClient
from app.src.pages.artist_page import ArtistPage
from app.src.pages.chart_digest import ChartDigestPage
from app.src.pages.left_navigation_panel import LeftNavigationPanel
from app.src.pages.login_page import LoginPage
from app.src.pages.track_page import TrackPage


@pytest.mark.usefixtures("set_up", "one_time_set_up")
class TopChartsPageTest(unittest.TestCase):
    log = cl.Logger(logging.DEBUG)

    @pytest.fixture(autouse=True)
    def classSetup(self):
        self.loginPage = LoginPage(self.driver)
        self.left_panel = LeftNavigationPanel(self.driver)
        self.chart_digest = ChartDigestPage(self.driver)
        self.trackPage = TrackPage(self.driver)
        self.artist_page = ArtistPage(self.driver)
        self.api_client = ApiClient()
        self.ts = TestStatus(self.driver)

    @parameterized.expand(["us", "argentina"])
    @allure.story("AP-661")
    @allure.severity(allure.severity_level.NORMAL)
    @allure.title("Verification of the 'Top 10' section for Apple tracks")
    @allure.description("""TC verifies Top 10 section on the Dashboard page functionality works as expected for Apple tracks.""")
    def test_top_10_apple(self, market):
        user = self.loginPage.get_user_email_with_worker()
        token = self.api_client.authorize(user=user)
        self.loginPage.login_with_worker()
        self.loginPage.is_logged_in()
        self.left_panel.open_chart_digest_page()
        self.chart_digest.is_opened()
        self.trackPage.select_market(AppConfig.get("market")[market]["name"])
        # step 1
        self.chart_digest.select_tab(tab="{}-{}".format(self.chart_digest.TOP_VIEW, self.chart_digest.TAB_APPLE))
        # step 3
        tracks = self.chart_digest.get_tracks_for_table(table=self.chart_digest.TOP_VIEW)
        t = self.chart_digest.get_text(element=tracks[1][0])
        self.chart_digest.click_element(element=tracks[1][0])
        self.trackPage.is_track_page_title_present(track=t)
        self.trackPage.go_back()
        self.chart_digest.select_tab(tab="{}-{}".format(self.chart_digest.TOP_VIEW, self.chart_digest.TAB_APPLE))
        # step 4
        self.chart_digest.verify_tooltip_apple_artist(table=self.chart_digest.TOP_VIEW)
        # step 2, 5-6
        self.compare_ui_api_apple_top_10(token=token, market=AppConfig.get("market")[market])
        self.chart_digest.verify_trend_calculation(table=self.chart_digest.TOP_VIEW, vendor=self.chart_digest.TAB_APPLE, market=AppConfig.get("market")[market]["market"], token=token)
        # step 7-9
        for c in ["name", "trend", "position"]:
            self.chart_digest.verify_sorting(table=self.chart_digest.TOP_VIEW, column=c)
        self.chart_digest.refresh()
        self.chart_digest.select_tab(tab="{}-{}".format(self.chart_digest.TOP_VIEW, self.chart_digest.TAB_APPLE))
        apple_api = self.api_client.get_vendor_api_tracks(vendor=self.chart_digest.TAB_APPLE, market=AppConfig.get("market")[market]["market"], token=token)
        top_apple_api = apple_api["tracks"][0:10]
        tracks_trends_api = [top_apple_api[t]["trend"] for t in range(len(top_apple_api))]
        trends_for_arrows = [-trend for trend in tracks_trends_api]
        trends_for_arrows_indexes = [t for t in range(len(trends_for_arrows))]
        for index, trend in zip(trends_for_arrows_indexes, trends_for_arrows):
            self.chart_digest.verify_trend_arrow(table=self.chart_digest.TOP_VIEW, index=index + 1, trend=trend)

    def compare_ui_api_apple_top_10(self, token, market):
        apple_ui = self.chart_digest.get_all_from_table(table=self.chart_digest.TOP_VIEW)
        self.ts.markFinal(all(p in list(range(1, 11)) for p in apple_ui["positions"]), "positions are correct")
        self.ts.markFinal(apple_ui["positions"] == sorted(apple_ui["positions"]), "tracks are sorted by positions")
        apple_api = self.api_client.get_vendor_api_tracks(vendor=self.chart_digest.TAB_APPLE, market=market["market"], token=token)
        top_apple_api = apple_api["tracks"]
        tracks_names_api = [top_apple_api[n]["name"] for n in range(len(top_apple_api))]
        tracks_artists_api = [top_apple_api[a]["artist"] for a in range(len(top_apple_api))]
        tracks_trends_api = [abs(top_apple_api[t]["trend"]) for t in range(len(top_apple_api))]
        positions_api = [top_apple_api[p]["position"] for p in range(len(top_apple_api))]
        apple_ui_tracks = [a.replace("  ", " ") for a in apple_ui["tracks"]]
        track_names_api_f = [a.replace("  ", " ") for a in tracks_names_api]
        assert_that(apple_ui_tracks).is_equal_to(track_names_api_f)
        assert_that(apple_ui["artists"]).is_equal_to(tracks_artists_api)
        blue_dot_indexes = [tracks_trends_api.index(t_api) for t_api in tracks_trends_api if abs(t_api) > 1000]
        if len(blue_dot_indexes) > 0:
            for index in blue_dot_indexes:
                self.chart_digest.verify_trend_blue_dot_present(index=index + 1, table=self.chart_digest.BUBBLING_UNDER)
        else:
            assert_that(apple_ui["trends"]).is_equal_to(tracks_trends_api)
        assert_that(apple_ui["positions"]).is_equal_to(positions_api)