import allure
import pytest
from assertpy import assert_that
from src.helpers.test_data_loader import DataLoader


@pytest.mark.usefixtures("one_time_set_up")
class TestAlbumView:
    artist_name = DataLoader.get_artist()

    @allure.title("Album View - base info")
    @allure.description("User should be able to view base info on Album View page")
    @allure.testcase("https://data-analytics.atlassian.net/browse/AMA-2057")
    @pytest.mark.smoke
    def test_view_album_info(self):
        self.login_page.login_as_super_user()
        self.welcome_page.click_skip_btn()
        self.artist_selector_page.select_an_artist_by_name(self.artist_name)

        self.library_page.view_all_albums()
        album_name = self.albums_page.list_albums.texts[1]
        self.albums_page.select_album_by_name(album_name)

        # Check album info attributes
        assert_that(self.album_view_page.is_image_present(), "Image should be shown").is_true()

        assert_that(self.album_view_page.is_album_name_present(album_name), "Name should be shown").is_true()
        assert_that(self.album_view_page.is_album_contains_tracks(), "Album should have tracks").is_true()
        assert_that(self.navigation_page.get_time_interval_value()).is_equal_to("Last 4 weeks")
        self.navigation_page.select_period_last_1_year()
        assert_that(self.navigation_page.get_time_interval_value()).is_equal_to("Last 1 year")
        assert_that(self.album_view_page.is_album_name_present(album_name), "Name should be shown").is_true()
        assert_that(self.album_view_page.is_album_contains_tracks(), "Album should have tracks").is_true()

    @allure.title("Album View - Compare Tracks")
    @allure.description("User should be able to Compare Tracks on Album View page")
    @allure.testcase("https://data-analytics.atlassian.net/browse/AMA-1900")
    @pytest.mark.smoke
    def test_view_album_compare_track(self):
        self.login_page.login_as_super_user()
        self.welcome_page.click_skip_btn()
        self.artist_selector_page.select_an_artist_by_name(self.artist_name)

        self.library_page.view_all_albums()
        self.albums_page.select_album_by_position(2)
        self.album_view_page.bn_album_compare_tracks.click()

        assert_that(self.album_view_page.is_label_present("ALL ")).is_true()
        assert_that(self.album_view_page.is_label_present("Track Performance")).is_true()
        assert_that(self.album_view_page.is_label_present("Tap bar chart to see the track details")).is_true()

        self.album_view_page.switch_range_to_1y()
        assert_that(self.album_view_page.is_label_present("Tap bar chart to see the track details")).is_true()
        self.album_view_page.switch_range_to_26w()
        assert_that(self.album_view_page.is_label_present("Tap bar chart to see the track details")).is_true()
        self.album_view_page.switch_range_to_8w()
        assert_that(self.album_view_page.is_label_present("Tap bar chart to see the track details")).is_true()
        self.album_view_page.switch_range_to_6w()
        assert_that(self.album_view_page.is_label_present("Tap bar chart to see the track details")).is_true()
        self.album_view_page.switch_range_to_4w()
        assert_that(self.album_view_page.is_label_present("Tap bar chart to see the track details")).is_true()
        self.album_view_page.switch_range_to_2w()
        assert_that(self.album_view_page.is_label_present("Tap bar chart to see the track details")).is_true()
        self.album_view_page.switch_range_to_chart_week()
        assert_that(self.album_view_page.is_label_present("Tap bar chart to see the track details")).is_true()