在SpecFlow测试中控制IFeatureManager并读取配置的实现方案

问题背景 #

我有一个注入了FeatureManager的Controller，业务逻辑会根据_featureManager.IsEnabledAsync("NameOfFeature")的返回值做分支处理。用SpecFlow做测试时，需要创建Given步骤来修改特定FeatureManager特性，测试启用/禁用场景。同时部分特性要从appsettings.json的FeatureManagement节点读取（适配本地/CI Docker环境差异），应用内的特性可Mock，但SpecFlow需读取配置中部分特性值。

实现步骤 #

1. 自定义FeatureManager包装类 #

创建一个支持覆盖特性状态的IFeatureManager实现，默认读取配置值，允许临时指定特定特性的开关：

public class OverridableFeatureManager : IFeatureManager
{
    private readonly IFeatureManager _originalFeatureManager;
    private readonly Dictionary<string, bool> _overriddenFeatures = new();

    public OverridableFeatureManager(IFeatureManager originalFeatureManager)
    {
        _originalFeatureManager = originalFeatureManager;
    }

    // 用于覆盖特性状态的方法
    public void SetFeatureState(string featureName, bool isEnabled)
    {
        lock (_overriddenFeatures)
        {
            if (_overriddenFeatures.ContainsKey(featureName))
                _overriddenFeatures[featureName] = isEnabled;
            else
                _overriddenFeatures.Add(featureName, isEnabled);
        }
    }

    // 清除所有覆盖，恢复默认配置
    public void ClearOverrides()
    {
        lock (_overriddenFeatures)
        {
            _overriddenFeatures.Clear();
        }
    }

    // 实现IFeatureManager接口方法，优先使用覆盖值
    public Task<bool> IsEnabledAsync(string featureName)
    {
        lock (_overriddenFeatures)
        {
            if (_overriddenFeatures.TryGetValue(featureName, out bool isEnabled))
                return Task.FromResult(isEnabled);
        }
        return _originalFeatureManager.IsEnabledAsync(featureName);
    }

    // 实现其他IFeatureManager接口方法（如果用到的话）
    public Task<bool> IsEnabledAsync<TContext>(string featureName, TContext context)
    {
        lock (_overriddenFeatures)
        {
            if (_overriddenFeatures.TryGetValue(featureName, out bool isEnabled))
                return Task.FromResult(isEnabled);
        }
        return _originalFeatureManager.IsEnabledAsync(featureName, context);
    }
}

2. 修改SpecFlow Hooks配置 #

在Hooks中替换默认的IFeatureManager为自定义实现，同时确保加载appsettings.json配置：

[Binding]
public sealed class Hooks
{
    private static IHost _host;
    private static IConfiguration _configuration;

    [BeforeTestRun]
    public static void BeforeTestRun()
    {
        // 加载测试用的配置文件
        _configuration = new ConfigurationBuilder()
            .SetBasePath(Directory.GetCurrentDirectory())
            .AddJsonFile("appsettings.json", optional: false, reloadOnChange: true)
            .AddEnvironmentVariables() // 支持CI环境的环境变量覆盖
            .Build();

        var hostBuilder = Program.CreateHostBuilder(Array.Empty<string>());

        hostBuilder.ConfigureServices((hostBuilderContext, services) => {
            // 移除默认的IFeatureManager注册
            var featureManagerDescriptor = services.SingleOrDefault(d => d.ServiceType == typeof(IFeatureManager));
            if (featureManagerDescriptor != null)
                services.Remove(featureManagerDescriptor);

            // 先添加原始的FeatureManagement服务（读取配置）
            services.AddFeatureManagement(_configuration.GetSection("FeatureManagement"));

            // 替换为自定义的OverridableFeatureManager
            services.AddScoped<IFeatureManager>(sp => 
                new OverridableFeatureManager(sp.GetRequiredService<IFeatureManager>()));
        });

        _host = hostBuilder.Build();
        _host.Start();
    }

    [BeforeScenario]
    public void BeforeScenario(ScenarioContext scenarioContext)
    {
        // 每个场景前清除特性覆盖，避免场景间污染
        var featureManager = _host.Services.GetRequiredService<IFeatureManager>() as OverridableFeatureManager;
        featureManager?.ClearOverrides();
    }

    [ScenarioDependencies]
    public static IServiceCollection CreateServices()
    {
        var services = new ServiceCollection();
        
        // 注入配置和自定义FeatureManager到SpecFlow步骤
        services.AddSingleton(_configuration);
        services.AddFeatureManagement(_configuration.GetSection("FeatureManagement"));
        services.AddScoped<IFeatureManager>(sp => 
            new OverridableFeatureManager(sp.GetRequiredService<IFeatureManager>()));

        return services;
    }
}

3. 实现SpecFlow的Given步骤 #

在StepDefinitions中注入OverridableFeatureManager，实现控制特性开关的步骤：

[Binding]
public class StepDefinitions
{
    private readonly OverridableFeatureManager _featureManager;

    // 构造函数注入自定义FeatureManager
    public StepDefinitions(IFeatureManager featureManager)
    {
        _featureManager = featureManager as OverridableFeatureManager;
    }

    [Given(@"the feature '([^']*)' is enabled")]
    public void GivenTheFeatureIsEnabled(string feature)
    {
        _featureManager?.SetFeatureState(feature, true);
    }

    [Given(@"the feature '([^']*)' is disabled")]
    public void GivenTheFeatureIsDisabled(string feature)
    {
        _featureManager?.SetFeatureState(feature, false);
    }
}

关键说明 #

• 自定义的OverridableFeatureManager会优先使用通过SetFeatureState设置的覆盖值，未覆盖的特性仍从appsettings.json读取配置
• 每个Scenario前会清除覆盖状态，保证测试隔离性
• 支持环境变量覆盖配置，适配本地和CI Docker环境的差异
• 应用项目的Startup.cs无需修改，测试时会替换服务注册

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