如何将Scanner选择的WebDriver值传入TestNG套件的browser参数？

直接将selectDriver方法的返回值传入TestSuite.xml的browser参数是做不到的——因为TestNG的XML配置会在Java代码执行前完成加载解析，而Scanner获取用户输入是在代码运行时才发生的，二者时序不匹配，无法直接传递值到XML中。

不过有更简便的替代方案，不需要修改XML文件即可实现需求：

推荐方案：通过系统属性传递浏览器参数 #

核心思路是先获取用户选择的浏览器，将其设置为JVM系统属性，再启动TestNG套件，测试代码直接读取该系统属性初始化WebDriver。

代码示例 #

1. 编写驱动选择与TestNG启动类：

import org.testng.TestNG;
import java.util.List;
import java.util.Scanner;

public class DriverSelector {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        System.out.println("请选择要运行的浏览器：");
        System.out.println("1. Chrome");
        System.out.println("2. Firefox");
        System.out.println("3. Edge");
        
        int userChoice = scanner.nextInt();
        String selectedBrowser = switch(userChoice) {
            case 1 -> "chrome";
            case 2 -> "firefox";
            case 3 -> "edge";
            default -> throw new IllegalArgumentException("无效的浏览器选择");
        };
        
        // 将选择的浏览器设置为系统属性
        System.setProperty("test.browser", selectedBrowser);
        
        // 启动TestNG套件
        TestNG testng = new TestNG();
        testng.setTestSuites(List.of("TestSuite.xml"));
        testng.run();
    }
}

2. 测试类中读取系统属性初始化WebDriver：

import org.openqa.selenium.WebDriver;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.edge.EdgeDriver;
import org.openqa.selenium.firefox.FirefoxDriver;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;

public class SeleniumTestCases {
    private WebDriver driver;

    @BeforeMethod
    public void setupDriver() {
        String browser = System.getProperty("test.browser");
        switch(browser) {
            case "chrome" -> driver = new ChromeDriver();
            case "firefox" -> driver = new FirefoxDriver();
            case "edge" -> driver = new EdgeDriver();
            default -> throw new RuntimeException("未获取到有效浏览器参数");
        }
    }

    @Test
    public void sampleTest() {
        driver.get("https://example.com");
        // 测试逻辑...
    }
}

这个方案的优势是完全不需要修改XML文件，通过JVM系统属性完成参数传递，时序上完全匹配（先获取用户输入，再启动TestNG）。

备选方案：动态修改XML文件（不推荐） #

如果你坚持要通过修改XML的browser参数实现，可以用Java XML解析库（如DOM4J、JDOM）读取TestSuite.xml，找到对应参数节点并修改值，保存后再启动TestNG。但这种方法需要处理XML读写逻辑，容易出现格式错误，复杂度远高于上述方案，除非有特殊需求不建议使用。

内容的提问来源于stack exchange，提问作者Gabriel Dragomir