from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
import unittest
class LoginTest(unittest.TestCase):
def setUp(self):
self.driver = webdriver.Chrome()
self.driver.get('https://example.com/login')
self.wait = WebDriverWait(self.driver, 10)
def test_valid_login(self):
driver = self.driver
wait = self.wait
driver.find_element(By.ID, 'username').send_keys('user@example.com')
driver.find_element(By.ID, 'password').send_keys('Password123')
driver.find_element(By.ID, 'login-button').click()
wait.until(EC.url_contains('/dashboard'))
self.assertIn('/dashboard', driver.current_url)
def test_invalid_login(self):
driver = self.driver
wait = self.wait
driver.get('https://example.com/login')
driver.find_element(By.ID, 'username').send_keys('wronguser@example.com')
driver.find_element(By.ID, 'password').send_keys('WrongPass')
driver.find_element(By.ID, 'login-button').click()
error = wait.until(EC.visibility_of_element_located((By.ID, 'error-message')))
self.assertTrue(error.is_displayed())
self.assertEqual(error.text, 'Invalid username or password.')
def tearDown(self):
self.driver.quit()
if __name__ == '__main__':
unittest.main()This script uses Selenium with Python's unittest framework to automate login tests.
In setUp, the browser opens the login page and sets an explicit wait.
The test_valid_login method enters valid credentials, clicks login, waits for the dashboard URL, and asserts the URL changed.
The test_invalid_login method enters wrong credentials, clicks login, waits for an error message, and asserts it is visible and correct.
Explicit waits ensure the test waits for page changes or elements before asserting.
Using IDs for locators keeps selectors simple and reliable.
The tearDown method closes the browser after each test.
This structure follows best practices for maintainability and clear assertions.
