from selenium.webdriver.common.by import By

from apps.lyric.models import User


class MyNotesPage:
    def __init__(self, test):
        self.test = test

    def go_to_my_notes_page(self, email):
        self.test.browser.get(self.test.live_server_url)
        user = User.objects.get(email=email)
        self.test.browser.get(
            self.test.live_server_url + f'/dashboard/users/{user.id}/'
        )
        self.test.wait_for(
            lambda: self.test.assertIn(
                email,
                self.test.browser.find_element(By.TAG_NAME, "h2").text.lower(),
            )
        )
        return self