Selenium WebDriver-Handling Alerts in Python

import unittest
from selenium import webdriver
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.chrome.options import Options

class BaseTest(unittest.TestCase):
    def setUp(self):
        # Set up ChromeDriver (update path to your chromedriver executable)
        chrome_options = Options()
        chrome_options.add_argument("--start-maximized")  # Maximize browser window
        service = Service("C:\\Users\\change the path of chromedriver.exe\\drivers\\chromedriver.exe")  # Update to your path
        self.driver = webdriver.Chrome(service=service, options=chrome_options)
        
        # Set implicit wait for element detection
        self.driver.implicitly_wait(10)  # 10 seconds

    def tearDown(self):
        if self.driver:
            self.driver.quit()

import unittest
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.alert import Alert
from base_test import BaseTest

class TestAlerts(BaseTest):
    URL = "https://bonigarcia.dev/selenium-webdriver-java/dialog-boxes.html"

    def test_simple_alert(self):
        try:
            # Navigate to the test page
            self.driver.get(self.URL)

            # Click the button to trigger the simple alert
            self.driver.find_element(By.ID, "my-alert").click()

            # Wait for the alert to appear and switch to it
            alert = WebDriverWait(self.driver, 5).until(EC.alert_is_present())
            alert = Alert(self.driver)

            # Validate the alert text
            alert_text = alert.text
            self.assertEqual(alert_text, "Hello world!", "Simple alert text does not match expected.")
            print("Simple Alert Text:", alert_text)

            # Accept the alert (click OK)
            alert.accept()
            print("Simple Alert Executed Successfully")
        except Exception as e:
            self.fail(f"Failed to handle simple alert: {str(e)}")

    def test_confirm_alert(self):
        try:
            # Navigate to the test page
            self.driver.get(self.URL)

            # Click the button to trigger the confirmation alert
            confirm_button = WebDriverWait(self.driver, 5).until(EC.element_to_be_clickable((By.ID, "my-confirm")))
            confirm_button.click()

            # Wait for the alert to appear and switch to it
            confirm = WebDriverWait(self.driver, 5).until(EC.alert_is_present())
            confirm = Alert(self.driver)

            # Validate the alert text
            alert_text = confirm.text
            self.assertEqual(alert_text, "Is this correct?", "Confirmation alert text does not match expected.")
            print("Confirmation Alert Text:", alert_text)

            # Dismiss the alert (click Cancel)
            confirm.dismiss()

            print("Confirmation Alert Executed Successfully")
        except Exception as e:
            self.fail(f"Failed to handle confirmation alert: {str(e)}")

    def test_prompt_alert(self):
        try:
            # Navigate to the test page
            self.driver.get(self.URL)

            # Click the button to trigger the prompt alert
            self.driver.find_element(By.ID, "my-prompt").click()

            # Wait for the alert to appear and switch to it
            prompt = WebDriverWait(self.driver, 5).until(EC.alert_is_present())
            prompt = Alert(self.driver)

            # Send text to the prompt
            prompt.send_keys("vaibhav")

            # Accept the prompt (click OK)
            prompt.accept()

            print("Prompt Alert Executed Successfully")
        except Exception as e:
            self.fail(f"Failed to handle prompt alert: {str(e)}")