如何在SpecFlow.DependencyInjection的BeforeScenario中注册WebDriver到DI容器

我正在使用SpecFlow.DependencyInjection组件，通过[ScenarioDependencies]特性创建服务，代码如下：

[ScenarioDependencies]
public static IServiceCollection CreateServices()
{
    var services = new ServiceCollection();

    // 注册其他服务

    return services;
}

问题在于CreateServices方法会在BeforeScenario特性执行前运行，同时我还使用SpecFlow.Contrib.Variants组件来选择要使用的浏览器驱动。此前我会在BeforeScenario钩子中编写如下代码：

[BeforeScenario]
public void BeforeScenario()
{
    _scenarioContext.TryGetValue("Browser", out var browser);

    switch (browser)
    {
        case "Chrome":
            _driver = SetupChromeDriver();
            break;
        case "Firefox":
            _driver = SetupFirefoxDriver();
            break;
        default:
            _driver = SetupChromeDriver();
            break;
    }
    _scenarioContext.ScenarioContainer.RegisterInstanceAs(_driver);
}

但使用SpecFlow.DependencyInjection后，_scenarioContext.ScenarioContainer.RegisterInstanceAs方法不再生效，且无法在CreateServices方法中完成驱动的注册。我希望能通过构造函数注入的方式使用IWebDriver，例如：

public void DoSomething(IWebDriver driver)
{
    _driver = driver;
    // 使用driver执行操作
}

方法一：通过延迟工厂注册（推荐） #

利用ASP.NET Core DI的延迟解析特性，在CreateServices中注册IWebDriver为Scoped服务，通过工厂方法在首次注入时根据ScenarioContext中的浏览器类型创建对应驱动。这种方式无需修改BeforeScenario钩子，直接实现构造函数注入。

修改[ScenarioDependencies]的服务注册代码：

[ScenarioDependencies]
public static IServiceCollection CreateServices()
{
    var services = new ServiceCollection();

    // 注册ScenarioContext（SpecFlow会自动为每个场景创建独立实例）
    services.AddSingleton<ScenarioContext>();

    // 注册其他业务服务...

    // 注册IWebDriver，Scoped生命周期对应每个场景
    services.AddScoped<IWebDriver>(serviceProvider =>
    {
        var scenarioContext = serviceProvider.GetRequiredService<ScenarioContext>();
        scenarioContext.TryGetValue("Browser", out var browserName);

        // 根据浏览器类型创建对应驱动
        return browserName switch
        {
            "Firefox" => SetupFirefoxDriver(),
            _ => SetupChromeDriver() // 默认使用Chrome
        };
    });

    return services;
}

// 驱动初始化方法（根据实际需求配置选项）
private static IWebDriver SetupChromeDriver()
{
    var options = new ChromeOptions();
    // 示例：添加无头模式等配置
    // options.AddArgument("--headless=new");
    return new ChromeDriver(options);
}

private static IWebDriver SetupFirefoxDriver()
{
    var options = new FirefoxOptions();
    // 示例：添加无头模式等配置
    // options.AddArgument("--headless");
    return new FirefoxDriver(options);
}

之后就可以直接在步骤类中通过构造函数注入IWebDriver：

[Binding]
public class HomepageSteps
{
    private readonly IWebDriver _driver;

    // 构造函数注入IWebDriver
    public HomepageSteps(IWebDriver driver)
    {
        _driver = driver;
    }

    [Given(@"我打开首页")]
    public void GivenIOpenHomepage()
    {
        _driver.Navigate().GoToUrl("https://example.com");
    }
}

方法二：通过包装类实现（适用于必须在BeforeScenario中处理驱动的场景） #

如果需要在BeforeScenario钩子中对驱动做额外操作，可通过包装类间接实现实例注入：

1. 创建驱动包装类：

public class WebDriverWrapper
{
    public IWebDriver Driver { get; set; }
}

2. 在CreateServices中注册包装类为Scoped服务：

[ScenarioDependencies]
public static IServiceCollection CreateServices()
{
    var services = new ServiceCollection();

    services.AddSingleton<ScenarioContext>();
    services.AddScoped<WebDriverWrapper>(); // 注册包装类

    // 注册其他服务...

    return services;
}

3. 修改BeforeScenario钩子，创建驱动并赋值给包装类：

[Binding]
public class Hooks
{
    private readonly ScenarioContext _scenarioContext;
    private readonly WebDriverWrapper _driverWrapper;

    public Hooks(ScenarioContext scenarioContext, WebDriverWrapper driverWrapper)
    {
        _scenarioContext = scenarioContext;
        _driverWrapper = driverWrapper;
    }

    [BeforeScenario]
    public void BeforeScenario()
    {
        _scenarioContext.TryGetValue("Browser", out var browser);
        IWebDriver driver = browser switch
        {
            "Firefox" => SetupFirefoxDriver(),
            _ => SetupChromeDriver()
        };

        _driverWrapper.Driver = driver;
        // 可在此添加驱动的额外初始化操作
    }

    private IWebDriver SetupChromeDriver()
    {
        var options = new ChromeOptions();
        return new ChromeDriver(options);
    }

    private IWebDriver SetupFirefoxDriver()
    {
        var options = new FirefoxOptions();
        return new FirefoxDriver(options);
    }
}

4. 在步骤类中注入包装类并使用驱动：

[Binding]
public class HomepageSteps
{
    private readonly IWebDriver _driver;

    public HomepageSteps(WebDriverWrapper driverWrapper)
    {
        _driver = driverWrapper.Driver;
    }

    [Given(@"我打开首页")]
    public void GivenIOpenHomepage()
    {
        _driver.Navigate().GoToUrl("https://example.com");
    }
}

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