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 TestSearch:
    data = DataLoader("test_search_data.json")
    artist_name = DataLoader.get_artist()
    artist_name_search = data.get("artist_name_search")
    track_name_search = data.get("track_name_search")
    video_name_search = data.get("video_name_search")
    video_name_found = data.get("video_name_found")
    album_name_search = data.get("album_name_search")

    @allure.title("Search Track")
    @allure.description("User is able to search track")
    @allure.testcase("https://data-analytics.atlassian.net/browse/AMA-1888")
    @pytest.mark.smoke
    def test_search_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_search)

        # Search for the track
        self.navigation_page.go_to_search()
        assert_that(self.search_page.is_search_screen_opened()).is_true()
        self.search_page.in_search_items.send_text(self.track_name_search)
        self.search_page.press_return_to_hide_keyboard()
        assert_that(self.common_page.is_label_present("Tracks")).is_true()
        self.search_page.list_items.select_by_text(self.track_name_search)

        # Check track name label
        assert_that(self.common_page.is_label_present(self.track_name_search)).is_true()

    @allure.title("Search Video")
    @allure.description("User is able to search video")
    @allure.testcase("https://data-analytics.atlassian.net/browse/AMA-1889")
    @pytest.mark.regression
    def test_search_video(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_search)

        # Search for the video
        self.navigation_page.go_to_search()
        assert_that(self.search_page.is_search_screen_opened()).is_true()
        self.search_page.in_search_items.send_text(self.video_name_search)
        self.search_page.press_return_to_hide_keyboard()
        assert_that(self.common_page.is_label_present("Videos")).is_true()
        self.search_page.list_items.select_by_text(self.video_name_found)

        # Check video name label
        assert_that(self.common_page.is_label_present(self.video_name_search)).is_true()

    @allure.title("Search Album")
    @allure.description("User is able to search album")
    @allure.testcase("https://data-analytics.atlassian.net/browse/AMA-1890")
    @pytest.mark.regression
    def test_search_album(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_search)

        # Search for the album
        self.navigation_page.go_to_search()
        assert_that(self.search_page.is_search_screen_opened()).is_true()
        self.search_page.in_search_items.send_text(self.album_name_search)
        self.search_page.press_return_to_hide_keyboard()
        assert_that(self.common_page.is_label_present("Albums, EPs, & LPs")).is_true()
        self.search_page.list_items.select_by_text(self.album_name_search)

        # Check album name label
        assert_that(self.common_page.is_label_present(self.album_name_search)).is_true()