Modifying element attributes with JS in Selenium Python

Introduction

Sometimes, you need to change a web page element's attribute to test how the page behaves. Using JavaScript lets you do this easily during tests.

When you want to change a button's disabled state to enabled for testing.

When you need to update an input field's placeholder text dynamically.

When you want to add or remove a CSS class to check style changes.

When you want to simulate a change in an element's attribute that normally requires user action.

When you want to test how the page reacts to attribute changes without reloading.

Syntax

Selenium Python

driver.execute_script("arguments[0].setAttribute('attributeName', 'newValue')", element)

Use execute_script to run JavaScript code in the browser from Selenium.

arguments[0] refers to the web element passed from Selenium to JavaScript.

Examples

This disables a button by setting its 'disabled' attribute to 'true'.

Selenium Python

driver.execute_script("arguments[0].setAttribute('disabled', 'true')", button_element)

This changes the placeholder text of an input field.

Selenium Python

driver.execute_script("arguments[0].setAttribute('placeholder', 'Enter your name')", input_element)

This sets the CSS class of a div to 'highlighted' to change its style.

Selenium Python

driver.execute_script("arguments[0].setAttribute('class', 'highlighted')", div_element)

Sample Program

This script opens a page with a button, prints its disabled attribute (should be None), then disables it using JavaScript and prints the attribute again (should be 'true').

Selenium Python

from selenium import webdriver
from selenium.webdriver.common.by import By
import time

# Setup WebDriver (make sure chromedriver is in PATH)
driver = webdriver.Chrome()

try:
    # Open a simple test page
    driver.get('data:text/html,<button id="btn1">Click me</button>')

    # Find the button element
    button = driver.find_element(By.ID, 'btn1')

    # Initially, button is enabled
    print('Initially disabled attribute:', button.get_attribute('disabled'))

    # Disable the button using JS
    driver.execute_script("arguments[0].setAttribute('disabled', 'true')", button)

    # Check if disabled attribute is set
    print('After JS modification disabled attribute:', button.get_attribute('disabled'))

finally:
    time.sleep(1)  # Pause to see the browser if needed
    driver.quit()

Important Notes

Always pass the element as an argument to execute_script to avoid errors.

Modifying attributes with JS does not always trigger events; you may need to trigger them manually if needed.

Summary

You can change element attributes during tests using JavaScript with Selenium's execute_script.

This helps test different states without user interaction.

Remember to check the attribute after modification to confirm the change.

Practice

1. What does the following Selenium Python code do?
driver.execute_script("arguments[0].setAttribute('disabled', 'true')", element)

easy

A. It changes the element's text to 'true'.

B. It enables the web element by removing the 'disabled' attribute.

C. It disables the web element by setting its 'disabled' attribute to true.

D. It deletes the element from the page.

Solution

1. Step 1: Understand the JavaScript command

   The JavaScript code uses setAttribute to add or change the 'disabled' attribute of the element to 'true'.

2. Step 2: Effect on the web element

   Setting 'disabled' to 'true' makes the element disabled, meaning user interaction is blocked.

3. Final Answer:

   It disables the web element by setting its 'disabled' attribute to true. -> Option C

4. Quick Check:

   setAttribute('disabled', 'true') disables element [OK]

Hint: setAttribute('disabled', 'true') disables element [OK]

Common Mistakes:

• Confusing 'disabled' attribute with changing text content
• Thinking it removes the element
• Assuming it enables the element instead

2. Which of the following is the correct syntax to change an element's attribute 'class' to 'active' using Selenium Python's execute_script?

easy

A. driver.execute_script("arguments[0].setAttribute('class', 'active')", element)

B. driver.execute_script("element.setAttribute('class', 'active')")

C. driver.execute_script("arguments.setAttribute('class', 'active')", element)

D. driver.execute_script("arguments[0].setAttribute('class' = 'active')", element)

Solution

1. Step 1: Identify correct use of arguments in execute_script

   In Selenium, arguments[0] refers to the first argument passed after the script string, which is the element.

2. Step 2: Check JavaScript syntax for setAttribute

   The correct syntax is setAttribute('attributeName', 'value'). driver.execute_script("arguments[0].setAttribute('class', 'active')", element) uses this correctly with arguments[0].

3. Final Answer:

   driver.execute_script("arguments[0].setAttribute('class', 'active')", element) -> Option A

4. Quick Check:

   Use arguments[0].setAttribute('class', 'active') [OK]

Hint: Use arguments[0] for element in execute_script [OK]

Common Mistakes:

• Using 'element' directly inside the script string
• Wrong JavaScript syntax like '=' inside setAttribute
• Using 'arguments' without index

3. What will be the output of the following code snippet?

element = driver.find_element(By.ID, 'submit-btn')
driver.execute_script("arguments[0].setAttribute('data-test', 'passed')", element)
attr_value = element.get_attribute('data-test')
print(attr_value)

medium

A. SyntaxError

B. data-test

C. None

D. passed

Solution

1. Step 1: Modify attribute using JavaScript

   The script sets the 'data-test' attribute of the element to 'passed'.

2. Step 2: Retrieve the attribute value

   Calling get_attribute('data-test') fetches the updated value, which is 'passed'.

3. Final Answer:

   passed -> Option D

4. Quick Check:

   setAttribute then get_attribute returns new value [OK]

Hint: get_attribute returns updated attribute after JS change [OK]

Common Mistakes:

• Expecting None because attribute was not originally present
• Confusing attribute name with value
• Thinking JavaScript changes don't reflect in Selenium

4. Identify the error in the following code snippet that tries to change an element's 'title' attribute:

element = driver.find_element(By.CSS_SELECTOR, '.btn')
driver.execute_script("arguments[0].setAttribute('title', title)", element)

medium

A. The variable 'title' is not defined in the JavaScript context.

B. The element locator is incorrect.

C. The setAttribute method is misspelled.

D. execute_script cannot modify attributes.

Solution

1. Step 1: Analyze JavaScript string in execute_script

   The script uses 'title' as a variable inside the string, but it is not defined in JavaScript context.

2. Step 2: Understand variable passing in execute_script

   To pass a Python variable, it must be included as an argument and referenced via arguments[1] or similar.

3. Final Answer:

   The variable 'title' is not defined in the JavaScript context. -> Option A

4. Quick Check:

   JS variables must be passed explicitly [OK]

Hint: Pass variables as arguments, not inline strings [OK]

Common Mistakes:

• Using Python variables directly inside JS string
• Ignoring need for arguments[n] in execute_script
• Assuming locator error without checking script

5. You want to toggle a checkbox's 'checked' attribute using Selenium Python and JavaScript. Which code snippet correctly toggles the checkbox state?

hard

A. driver.execute_script("arguments[0].setAttribute('checked', !arguments[0].checked)", checkbox_element)

B. driver.execute_script("arguments[0].checked = !arguments[0].checked", checkbox_element)

C. driver.execute_script("arguments[0].checked = 'false'", checkbox_element)

D. driver.execute_script("arguments[0].removeAttribute('checked')", checkbox_element)

Solution

1. Step 1: Understand checkbox 'checked' property vs attribute

   The 'checked' property is a boolean that controls checkbox state; toggling it flips the state.

2. Step 2: Evaluate each option's effect

   driver.execute_script("arguments[0].checked = !arguments[0].checked", checkbox_element) uses JavaScript to toggle the boolean property correctly. driver.execute_script("arguments[0].setAttribute('checked', !arguments[0].checked)", checkbox_element) tries to set attribute with a boolean, which converts to string and may not work as expected. driver.execute_script("arguments[0].checked = 'false'", checkbox_element) sets 'checked' to string 'false', which does not uncheck. driver.execute_script("arguments[0].removeAttribute('checked')", checkbox_element) removes 'checked' attribute but does not toggle back on.

3. Final Answer:

   driver.execute_script("arguments[0].checked = !arguments[0].checked", checkbox_element) -> Option B

4. Quick Check:

   Toggle boolean property 'checked' to flip checkbox [OK]

Hint: Toggle checkbox by flipping 'checked' boolean property [OK]

Common Mistakes:

• Using setAttribute with boolean values
• Setting 'checked' to string 'false' instead of boolean
• Removing attribute without toggling back