如何在TestNG套件运行前验证WebDriver与浏览器版本匹配？

一、当前思路的问题分析 #

你现在的代码是在@Test方法内实例化WebDriver，一旦实例化失败（比如版本不匹配导致浏览器无法启动），TestNG会标记当前方法为失败，自动执行下一个@Test方法，这和你“停止整个套件”的需求不符。另外，通过检查登录按钮是否启用判断WebDriver版本的方式不够可靠——页面加载异常、元素定位错误等情况也会导致按钮无法被检测到，容易误判。

二、正确实现方向：用TestNG的@BeforeSuite做前置验证 #

TestNG的@BeforeSuite注解标记的方法会在整个测试套件运行前执行一次，若该方法抛出异常，整个套件会直接终止，完全匹配你的需求。

1. 实例化前预检查：WebDriver与浏览器版本匹配 #

可以通过读取本地浏览器版本，对比WebDriver版本实现预验证（WebDriver需与浏览器大版本一致）：

import java.io.BufferedReader;
import java.io.InputStreamReader;

public class WebDriverVersionChecker {
    // Windows系统下获取Chrome版本
    public static String getChromeVersion() throws Exception {
        Process process = Runtime.getRuntime().exec("reg query \"HKEY_CURRENT_USER\\Software\\Google\\Chrome\\BLBeacon\" /v version");
        BufferedReader reader = new BufferedReader(new InputStreamReader(process.getInputStream()));
        String line;
        while ((line = reader.readLine()) != null) {
            if (line.contains("version")) {
                return line.split("\\s+")[3];
            }
        }
        throw new Exception("未检测到Chrome浏览器版本");
    }

    // Windows系统下获取Edge版本
    public static String getEdgeVersion() throws Exception {
        Process process = Runtime.getRuntime().exec("reg query \"HKEY_CURRENT_USER\\Software\\Microsoft\\Edge\\BLBeacon\" /v version");
        BufferedReader reader = new BufferedReader(new InputStreamReader(process.getInputStream()));
        String line;
        while ((line = reader.readLine()) != null) {
            if (line.contains("version")) {
                return line.split("\\s+")[3];
            }
        }
        throw new Exception("未检测到Edge浏览器版本");
    }

    // 对比版本大号是否匹配
    public static boolean isVersionMatch(String browserVersion, String driverVersion) {
        String browserMajor = browserVersion.split("\\.")[0];
        String driverMajor = driverVersion.split("\\.")[0];
        return browserMajor.equals(driverMajor);
    }
}

2. 在@BeforeSuite中执行验证，失败终止套件 #

结合@BeforeSuite方法，验证不通过时抛出RuntimeException，TestNG会直接停止整个套件：

import org.testng.annotations.BeforeSuite;

public class SuitePreCheck {
    @BeforeSuite
    public void validateWebDrivers() {
        try {
            // 验证Chrome
            String chromeBrowserVer = WebDriverVersionChecker.getChromeVersion();
            String chromeDriverVer = new ChromeDriverService.Builder().getDriverVersion();
            if (!WebDriverVersionChecker.isVersionMatch(chromeBrowserVer, chromeDriverVer)) {
                throw new RuntimeException(String.format("Chrome版本不匹配：浏览器%s，Driver%s", chromeBrowserVer, chromeDriverVer));
            }

            // 验证Edge
            String edgeBrowserVer = WebDriverVersionChecker.getEdgeVersion();
            String edgeDriverVer = new EdgeDriverService.Builder().getDriverVersion();
            if (!WebDriverVersionChecker.isVersionMatch(edgeBrowserVer, edgeDriverVer)) {
                throw new RuntimeException(String.format("Edge版本不匹配：浏览器%s，Driver%s", edgeBrowserVer, edgeDriverVer));
            }

            // 额外验证：尝试启动浏览器确认可用性
            try (ChromeDriver chromeDriver = new ChromeDriver()) {
                LogFile.logfile("Chrome浏览器启动成功，WebDriver有效");
            }
            try (EdgeDriver edgeDriver = new EdgeDriver()) {
                LogFile.logfile("Edge浏览器启动成功，WebDriver有效");
            }
        } catch (Exception e) {
            LogFile.logfile("WebDriver验证失败：" + e.getMessage());
            throw new RuntimeException("WebDriver验证不通过，终止测试套件", e);
        }
    }
}

3. 更简便的替代方案：使用WebDriverManager #

如果不想手动处理版本匹配，可以用WebDriverManager库自动下载对应版本的WebDriver，彻底避免版本不匹配问题：

import io.github.bonigarcia.wdm.WebDriverManager;
import org.testng.annotations.BeforeSuite;

public class SuitePreCheck {
    @BeforeSuite
    public void setupAndValidateWebDrivers() {
        try {
            // 自动配置ChromeDriver并验证
            WebDriverManager.chromedriver().setup();
            try (ChromeDriver chromeDriver = new ChromeDriver()) {}
            
            // 自动配置EdgeDriver并验证
            WebDriverManager.edgedriver().setup();
            try (EdgeDriver edgeDriver = new EdgeDriver()) {}

            LogFile.logfile("所有WebDriver验证通过");
        } catch (Exception e) {
            LogFile.logfile("WebDriver配置失败：" + e.getMessage());
            throw new RuntimeException("终止测试套件", e);
        }
    }
}

三、关于当前代码跳转到下一个@Test方法的原因 #

TestNG默认会在单个@Test方法执行失败时，继续执行其他@Test方法。你使用的System.exit(2)虽然能终止JVM，但如果TestNG以多线程模式运行，可能无法立即生效。而通过@BeforeSuite抛出异常的方式，TestNG会直接停止整个套件，更符合你的需求。

内容的提问来源于stack exchange，提问作者Ian Thompson