如何在ITestListener中获取当前失败测试类的WebDriver

咱们先从最常见的根源问题排查起，一步步解决这个Driver获取混乱的问题：

1. 先确认BaseTest的WebDriver是非静态成员变量 #

这是绝大多数情况下导致Driver串用的核心原因：如果你的BaseTest里把WebDriver声明为static，所有测试类的实例都会共享同一个Driver对象，后面的测试类初始化时会覆盖这个静态变量，导致前面的测试用例失败时，拿到的是最后一个初始化的Driver。

正确的BaseTest写法应该是每个测试实例拥有独立的Driver：

public class BaseTest {
    // 非静态private成员变量，每个测试类实例都有自己的Driver副本
    private WebDriver driver;

    @BeforeMethod
    public void setUp() {
        // 每个@Test方法执行前初始化专属的Driver
        driver = new ChromeDriver(new ChromeOptions());
        driver.manage().window().maximize();
    }

    @AfterMethod
    public void tearDown() {
        // 每个@Test方法执行后关闭当前实例的Driver
        if (driver != null) {
            driver.quit();
        }
    }

    // 提供获取当前实例Driver的方法
    public WebDriver getDriver() {
        return driver;
    }
}

2. 检查测试类是否正确继承BaseTest #

确保所有测试类只是单纯继承BaseTest，没有自己重新定义WebDriver变量（尤其是静态的）：

// 正确的测试类写法
public class LoginTest extends BaseTest {
    @Test
    public void invalidLoginTest() {
        // 直接调用getDriver()获取当前实例的专属Driver
        getDriver().get("https://your-app-url.com/login");
        // 测试逻辑...
    }
}

如果测试类里自己定义了static WebDriver driver，会直接覆盖BaseTest的成员变量，导致Listener拿到错误的Driver。

3. 优化ITestListener的onTestFailure逻辑 #

你的现有代码逻辑核心是对的，但可以加一层空值和类型校验，避免空指针和类型转换错误：

@Override
public void onTestFailure(ITestResult iTestResult) {
    String testMethodName = getTestMethodName(iTestResult);
    System.out.println("测试方法 " + testMethodName + " 执行失败");

    // 获取当前失败测试方法所属的测试类实例
    Object testClassInstance = iTestResult.getInstance();
    if (testClassInstance instanceof BaseTest) {
        WebDriver currentDriver = ((BaseTest) testClassInstance).getDriver();
        if (currentDriver != null) {
            // 截取Base64格式截图
            String base64Screenshot = "data:image/png;base64," + 
                ((TakesScreenshot) currentDriver).getScreenshotAs(OutputType.BASE64);
            
            // 日志记录和截图上传
            ExtentTest currentTest = ExtentTestManager.getTest();
            currentTest.log(LogStatus.FAIL, "测试执行失败", 
                currentTest.addBase64ScreenShot(base64Screenshot));
        } else {
            System.out.println("当前测试方法对应的Driver为空，无法截图");
        }
    } else {
        System.out.println("测试类未继承BaseTest，无法获取Driver");
    }
}

这里的关键是iTestResult.getInstance()会返回当前失败测试方法所属的测试类实例，只要BaseTest的Driver是非静态的，这个实例的getDriver()就会返回该测试方法专属的Driver。

4. 并行执行场景的额外处理 #

如果你是并行执行测试（比如testng.xml里设置了parallel="tests"或parallel="classes"），一定要用ThreadLocal管理WebDriver和ExtentTest，避免线程间资源冲突。比如修改BaseTest为：

public class BaseTest {
    private ThreadLocal<WebDriver> driverThreadLocal = new ThreadLocal<>();

    @BeforeMethod
    public void setUp() {
        WebDriver driver = new ChromeDriver(new ChromeOptions());
        driverThreadLocal.set(driver);
    }

    @AfterMethod
    public void tearDown() {
        WebDriver driver = driverThreadLocal.get();
        if (driver != null) {
            driver.quit();
        }
        driverThreadLocal.remove(); // 清理ThreadLocal引用，避免内存泄漏
    }

    public WebDriver getDriver() {
        return driverThreadLocal.get();
    }
}

同时，ExtentTestManager也要用ThreadLocal存储每个测试方法的ExtentTest实例，确保日志和截图能正确对应到测试用例。

内容的提问来源于stack exchange，提问作者kritika agarwal