Selenium：使用TestNG监听器捕获异常后终止@Test方法后续步骤

针对你的需求——不在@Test方法中写try-catch，同时在元素找不到时终止当前测试的剩余步骤，我整理了一套基于TestNG和WebDriverEventListener的优化方案，核心思路是通过自定义RuntimeException让TestNG自动识别并终止测试，具体如下：

1. 自定义RuntimeException类 #

因为WebDriverEventListener的方法不允许抛出检查型异常，我们需要自定义一个继承自RuntimeException的异常类，让TestNG能够捕获并处理它：

public class ElementNotFoundException extends RuntimeException {
    public ElementNotFoundException(String message, Throwable cause) {
        super(message, cause);
    }
}

2. 修改CustomListeners监听逻辑 #

在beforeFindBy方法中，捕获NoSuchElementException后抛出我们自定义的异常，而不是仅仅打印日志。这样TestNG会检测到未捕获的异常，自动终止当前测试方法的后续执行：

public class CustomListeners extends TestListenerAdapter implements WebDriverEventListener {

    @Override
    public void beforeFindBy(By by, WebElement element, WebDriver driver) {
        try {
            driver.findElement(by);
        } catch (NoSuchElementException e) {
            String errorMsg = "元素定位失败：" + by;
            System.out.println(errorMsg);
            // 抛出自定义异常，触发TestNG终止测试
            throw new ElementNotFoundException(errorMsg, e);
        }
    }

    // 可选：重写onTestFailure方法，添加失败后的额外处理（比如截图、日志记录）
    @Override
    public void onTestFailure(ITestResult result) {
        if (result.getThrowable() instanceof ElementNotFoundException) {
            // 这里可以结合ExtentTest添加失败日志或截图
            ExtentTest test = (ExtentTest) result.getAttribute("test");
            test.log(LogStatus.FAIL, "测试失败：" + result.getThrowable().getMessage());
            // 调用你的截图方法
            // takeScreenshot(driver, test);
        }
    }

    // 其他WebDriverEventListener方法保持默认实现即可
}

3. 调整Page类的异常处理 #

把Page类方法中的检查型异常（比如IOException）包装成RuntimeException抛出，避免在@Test方法中声明throws：

import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.support.FindBy;
import org.openqa.selenium.support.PageFactory;
import org.testng.Assert;
import com.relevantcodes.extentreports.LogStatus;
import com.relevantcodes.extentreports.ExtentTest;
import java.io.IOException;

public class LoginPage {
    private WebDriver driver;
    private ExtentTest test;

    @FindBy(xpath = "//img[@src='/webres_5786257bd7c8a5.72130757/themes/default/images/login/logo.png']")
    WebElement loginLogo;

    @FindBy(id = "username")
    WebElement userNameField;
    @FindBy(id = "password")
    WebElement passwordField;
    @FindBy(id = "loginBtn")
    WebElement loginBtn;

    public LoginPage(WebDriver driver, ExtentTest test) {
        this.driver = driver;
        PageFactory.initElements(driver, this);
        this.test = test;
    }

    public void verifyLoginPageLogo() {
        try {
            Assert.assertTrue(loginLogo.isDisplayed());
            test.log(LogStatus.PASS, "Logo显示正常", takeScreenshot(driver, test));
        } catch (IOException e) {
            // 截图失败时包装成RuntimeException抛出
            throw new RuntimeException("截图失败：" + e.getMessage(), e);
        }
    }

    public void setUserName(String username) {
        userNameField.sendKeys(username);
    }

    public void setPassword(String password) {
        passwordField.sendKeys(password);
    }

    public void clickLoginBtn() {
        loginBtn.click();
    }

    private String takeScreenshot(WebDriver driver, ExtentTest test) throws IOException {
        // 你的截图逻辑，返回截图路径
        return "screenshot_path";
    }
}

4. 保持Test类简洁 #

不需要在@Test方法中添加任何try-catch，TestNG会自动捕获Listener抛出的异常，终止后续步骤并标记测试失败：

import org.testng.annotations.Listeners;
import org.testng.annotations.Test;

@Listeners(CustomListeners.class)
public class Test1 extends BaseTestSuite {
    LoginPage lp;
    TabMenu tm;

    @Test(priority = 0, testName = "Verify Login")
    public void login() {
        lp = new LoginPage(driver, test);
        tm = new TabMenu(driver, test);
        driver.get(Constants.url);
        lp.verifyLoginPageLogo(); // 如果元素不存在，这里会触发异常，终止后续步骤
        lp.setUserName("admin");
        lp.setPassword("admin");
        lp.clickLoginBtn();
        tm.verifyTabMenu();
        tm.isCurrentTab("Dashboard");
    }
}

方案原理说明 #

• 自定义RuntimeException：绕过WebDriverEventListener方法的异常声明限制，让TestNG能够识别并处理异常。
• Listener抛出异常：当元素找不到时，抛出的异常会打断当前的元素查找流程，TestNG检测到未捕获的RuntimeException后，会立即标记测试为失败，并且不再执行@Test方法的剩余代码。
• 无侵入式测试代码：不需要在@Test方法中添加任何异常捕获逻辑，保持测试代码的简洁性和可读性。

内容的提问来源于stack exchange，提问作者Mehavarnan Murugan