import unittest

import allure
import pytest
from ui_automation_framework.core import TestStatus
from ui_automation_framework.utils import AppConfig

from app.api_client import ApiClient
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 DemographicsTrackPageTest(unittest.TestCase):
    @pytest.fixture(autouse=True)
    def classSetup(self):
        self.loginPage = LoginPage(self.driver)
        self.trackPage = TrackPage(self.driver)
        self.api_client = ApiClient()
        self.ts = TestStatus(self.driver)

    @allure.story("AP-4262")
    @allure.severity(allure.severity_level.NORMAL)
    @allure.title("Demographics. Unknown age group verification - STEP 13")
    @allure.description("Demographics. Unknown age group verification - STEP 13")
    def test_unknown_age_group_step_13(self):
        start_date = "2017-09-01"
        end_date = "2017-12-31"
        user = self.loginPage.get_user_email_with_worker()
        token = self.api_client.authorize(user=user)

        # step 13
        spotify_us_0_17 = self.trackPage.get_streams_age_gender_api(
            dsp=AppConfig.get("dsp")["spotify"],
            market=AppConfig.get("market")["us"]["market"],
            isrc=AppConfig.get("track39")["isrc"],
            start_date=start_date,
            end_date=end_date,
            token=token,
        )[
            "ages"
        ][0]
        apple_us_0_17 = self.trackPage.get_streams_age_gender_api(
            dsp=AppConfig.get("dsp")["apple"],
            market=AppConfig.get("market")["us"]["market"],
            isrc=AppConfig.get("track39")["isrc"],
            start_date=start_date,
            end_date=end_date,
            token=token,
        )[
            "ages"
        ][0]
        spotify_gl_0_17 = self.trackPage.get_streams_age_gender_api(
            dsp=AppConfig.get("dsp")["spotify"],
            market=AppConfig.get("market")["global"]["api"],
            isrc=AppConfig.get("track39")["isrc"],
            start_date=start_date,
            end_date=end_date,
            token=token,
        )[
            "ages"
        ][0]
        apple_gl_0_17 = self.trackPage.get_streams_age_gender_api(
            dsp=AppConfig.get("dsp")["apple"],
            market=AppConfig.get("market")["global"]["api"],
            isrc=AppConfig.get("track39")["isrc"],
            start_date=start_date,
            end_date=end_date,
            token=token,
        )[
            "ages"
        ][0]
        self.ts.markFinal(spotify_us_0_17 > 0, "spotify us 0-17: {}".format(spotify_us_0_17))
        self.ts.markFinal(apple_us_0_17 > 0, "apple us 0-17: {}".format(apple_us_0_17))
        self.ts.markFinal(spotify_gl_0_17 > 0, "spotify gl 0-17: {}".format(spotify_gl_0_17))
        self.ts.markFinal(apple_gl_0_17 > 0, "apple gl 0-17: {}".format(apple_gl_0_17))