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.src.pages.amplitude_helper import AmplitudeHelper
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 Amplitude(unittest.TestCase):
    log = cl.Logger(logging.DEBUG)
    periods = ["2 Weeks", "4 Weeks", "8 Weeks", "12 Weeks", "26 Weeks", "1 Year"]
    page = "Track Page"

    @pytest.fixture(autouse=True)
    def classSetup(self):
        self.log = cl.Logger(logging.DEBUG)
        self.loginPage = LoginPage(self.driver)
        self.ts = TestStatus(self.driver)
        self.amplitude_helper = AmplitudeHelper(self.driver)
        self.track_page = TrackPage(self.driver)

    @parameterized.expand(
        [
            ("spotify", AppConfig.get("track45")["id"], AppConfig.get("track45")["name"]),
            ("apple", str(AppConfig.get("track44")["apple_id"]), AppConfig.get("track44")["name"]),
        ]
    )
    @allure.story("AP-3659")
    @allure.severity(allure.severity_level.NORMAL)
    @allure.title("[AP-3654][Amplitude] Verify event for changing date range on Track Page")
    @allure.description("""TC Verifies events when different date ranges are selected in Track Page""")
    def test_date_range_change_on_track_page(self, dsp, track_id, track_name):
        self.loginPage.login_with_worker()
        self.track_page.select_market(AppConfig.get("market")["us"]["name"])
        events = self.amplitude_helper.wait_and_get_amplitude_events(5, False)
        self.amplitude_helper.is_initializer_event(events[0])
        self.amplitude_helper.verify_event_user_properties(events[0])
        # step 1-2
        self.track_page.open_track_page(apple_spotify=dsp, track_id=track_id, market=AppConfig.get("market")["us"]["url"])
        self.track_page.is_track_page_title_present(track_name)
        events = self.amplitude_helper.wait_and_get_amplitude_events(2)
        for p in self.periods:
            self.track_page.change_period_in_date_picker(p)
            events = self.amplitude_helper.wait_and_get_amplitude_events(1)
            start_date = self.track_page.get_url_start_date()
            end_date = self.track_page.get_url_end_date()
            total_days = self.track_page.graph_get_count_days_from_calendar()
            self.amplitude_helper.verify_for_event_type(events[0], self.amplitude_helper.TRACK_PAGE["date_range_changed_event_type"])
            assert_that(events[0]["event_properties"]["page"]).is_equal_to(self.page)
            assert_that(events[0]["event_properties"]["period"]).is_equal_to(p)
            assert_that(events[0]["event_properties"]["start_date"]).is_equal_to(start_date)
            assert_that(events[0]["event_properties"]["end_date"]).is_equal_to(end_date)
            assert_that(events[0]["event_properties"]["total_days"]).is_equal_to(total_days)
        # custom period
        self.track_page.select_previous_mounth_period()
        events = self.amplitude_helper.wait_and_get_amplitude_events(1)
        start_date = self.track_page.get_url_start_date()
        end_date = self.track_page.get_url_end_date()
        total_days = self.track_page.graph_get_count_days_from_calendar()
        self.amplitude_helper.verify_for_event_type(events[0], self.amplitude_helper.TRACK_PAGE["date_range_changed_event_type"])
        assert_that(events[0]["event_properties"]["page"]).is_equal_to(self.page)
        assert_that(events[0]["event_properties"]["period"]).is_equal_to("Custom")
        assert_that(events[0]["event_properties"]["start_date"]).is_equal_to(start_date)
        assert_that(events[0]["event_properties"]["end_date"]).is_equal_to(end_date)
        assert_that(events[0]["event_properties"]["total_days"]).is_equal_to(total_days)