如何在Selenium中恢复元素原始DOM以避免StaleElementReferenceException

你的核心问题在于：当元素原本没有style属性时，直接将获取到的null/空字符串设置回style属性，会导致元素多出一个空的style=""属性，而非完全恢复到无style的原始状态，甚至可能触发StaleElementReferenceException。

修改后的解决方案代码 #

public static void objHighlight(WebDriver webdriver, WebElement element) {
    JavascriptExecutor js = (JavascriptExecutor) webdriver;
    String originalStyle = element.getAttribute("style");
    
    // 应用高亮样式
    js.executeScript("arguments[0].setAttribute('style', 'background: yellow; border: 2px solid red;');", element);
    
    try {
        Thread.sleep(500);
    } catch (InterruptedException e) {
        e.printStackTrace();
    }
    
    // 恢复原始状态：分情况处理
    if (originalStyle == null) {
        // 原本无style属性，直接移除该属性
        js.executeScript("arguments[0].removeAttribute('style');", element);
    } else {
        // 原本有style属性，恢复原有值
        js.executeScript("arguments[0].setAttribute('style', '" + originalStyle + "');", element);
    }
}

关键逻辑说明 #

• 判断原始状态：通过element.getAttribute("style")的返回值判断元素是否原本存在style属性——返回null表示元素从未有过该属性，返回空字符串则表示元素有style属性但值为空（这种情况需要保留空属性，而非移除）。
• 精准恢复：
  • 若原本无style属性，调用JavaScript的removeAttribute('style')彻底移除该属性，完全还原元素初始状态；
  • 若原本有style属性（包括空值），则将属性值设置回原始内容，保证样式完全一致。

额外优化建议 #

如果仍然出现StaleElementReferenceException，说明sleep期间元素可能已被页面重新渲染（DOM结构变化），可以尝试：

• 在恢复样式前，通过原始定位器重新查找元素；
• 使用显式等待替代Thread.sleep，确保元素处于可交互状态后再执行恢复操作。

内容的提问来源于stack exchange，提问作者Benny Thomas