This code example demonstrates a robust approach to testing a website using Python and Selenium, along with other libraries such as difflib, requests, os, and subprocess. The code navigates to a spec

This code example demonstrates a robust approach to testing a website using Python and Selenium, along with other libraries such as difflib, requests, os, and subprocess. The code navigates to a specified URL, takes a screenshot of the page, and compares the text of the page with an expected text using the difflib library.

It also checks for any broken links on the page using the requests library and takes a screenshot of any broken links if they are found.

Additionally, the code checks for any visual changes on the page using the ImageMagick command-line tool and the Python os and subprocess libraries.

from selenium import webdriver

from https://lnkd.in/eyphV6mv import By

from selenium.webdriver.support.ui import WebDriverWait

from selenium.webdriver.support import expected_conditions as EC

import difflib

import time

import requests

import os

url = "https://www.example.com"

expected_text = "Welcome to Example.com"

driver_path = "/path/to/chromedriver"

screenshot_dir = "/path/to/screenshots"

def take_screenshot(driver, file_name):

if not os.path.exists(screenshot_dir):

os.makedirs(screenshot_dir)

driver.save_screenshot(os.path.join(screenshot_dir, file_name))

def compare_text(driver):

page_text = driver.find_element_by_tag_name("body").text

ratio = difflib.SequenceMatcher(None, expected_text, page_text).ratio()

return ratio >= 0.9

options = webdriver.ChromeOptions()

options.add_argument("--headless")

driver = webdriver.Chrome(driver_path, options=options)

driver.set_window_size(1920, 1080)

driver.get(url)

wait = WebDriverWait(driver, 10)

wait.until(EC.presence_of_element_located((By.TAG_NAME, "body")))

screenshot_name = f"{url.replace('https://', '')}_{int(time.time())}.png"

take_screenshot(driver, screenshot_name)

assert compare_text(driver)

broken_links = []

links = driver.find_elements_by_tag_name("a")

for link in links:

url = link.get_attribute("href")

if url.startswith("mailto:"):

continue

try:

response = requests.head(url, allow_redirects=True, timeout=5)

if response.status_code >= 400:

broken_links.append(url)

except:

broken_links.append(url)

if broken_links:

screenshot_name = f"{url.replace('https://', '')}_broken_links_{int(time.time())}.png"

take_screenshot(driver, screenshot_name)

raise Exception(f"Found broken links: {broken_links}")

expected_image = "/path/to/expected_image.png"

screenshot_image = os.path.join(screenshot_dir, screenshot_name)

if not os.path.exists(expected_image):

os.rename(screenshot_image, expected_image)

else:

os.system(f"compare -metric AE -fuzz 5% {expected_image} {screenshot_image} /dev/null 2>&1")

difference = int(os.popen(f"identify -format %[distortion] {screenshot_image}").read().strip())

if difference > 1000:

raise Exception(f"Found visual changes: {screenshot_image}")

driver.quit()