from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys

from .base import FunctionalTest


class DashboardMaintenanceTest(FunctionalTest):
    def test_user_without_username_can_claim_unclaimed_username(self):
        # 1. Create a pre-authenticated session for discoman@example.com
        self.create_pre_authenticated_session("discoman@example.com")
        # 2. Navigate to self.live_server_url + "/"
        self.browser.get(self.live_server_url)
        # 3. Find the username applet on the page; look for a <section> or <div> with id="id_username_applet"
        self.browser.find_element(By.ID, "id_applet_username")
        # 4. Assert it shows the current display name (truncated email: di…an@e…e.com)
        self.assertIn("di…an@e…e.com", self.browser.find_element(By.ID, "id_applet_username").text)
        # 5. Find the username input field inside the applet & type a username
        username_input = self.browser.find_element(By.CSS_SELECTOR, "#id_new_username")
        # 6. Type a username, e.g., discoman
        username_input.send_keys("discoman")
        self.wait_for(
            lambda: self.browser.find_element(By.CSS_SELECTOR, "#id_new_username:valid")
        )
        # 7. Submit the form (click a btn or press Enter)
        username_input.send_keys(Keys.ENTER)
        # 8. Without a page reload, wait for the navbar to update; user wait_for() to check that the navbar text now contains "discoman"
        self.wait_for(
            lambda: self.assertIn(
                "discoman",
                self.browser.find_element(By.CLASS_NAME, "navbar-text").text
            )
        )
        # 9. Also assert the applet input now shows "discoman" as its value
        self.wait_for(
            lambda: self.assertEqual(
                "discoman",
                self.browser.find_element(By.CSS_SELECTOR, "#id_new_username").get_attribute("value")
            )
        )