如何优雅地运行带功能开关的XUnit测试类变体？

在为代码添加功能开关后，需要保证原有测试用例在开关开启/关闭状态下均能正常通过，同时要避免循环遍历（状态不隔离）、复制测试数据/方法（维护成本高）等方案的弊端，这里提供一种基于xUnit参数化测试的优雅解决方案：

核心思路 #

利用xUnit的[MemberData]特性，将功能开关状态与原有测试数据做笛卡尔积组合，生成独立的隔离测试用例，既复用原有测试逻辑，又确保每个测试用例的独立性。

具体实现步骤 #

• 提取复用测试数据：将原有[InlineData]中的测试数据提取为静态成员，方便后续组合使用
• 生成组合测试数据：创建静态方法，将开关状态（true/false）与原有测试数据进行组合，生成所有可能的测试场景
• 修改测试方法：为原有测试方法添加开关状态参数，在测试执行前重置Mock的功能开关状态
• 替换测试数据特性：用[MemberData]替代[InlineData]，使用组合后的测试数据

完整代码示例 #

被测类与接口 #

using Moq;

public interface IFeatureFlags
{
    bool IsFlagEnabled(string flagName);
}

public class TestSubject(IFeatureFlags featureFlags)
{
    public void EnsureNumberIsOdd(int inputValue)
    {
        if (inputValue % 2 != 1)
            throw new ArgumentException("Must be an odd number", paramName: nameof(inputValue));

        if (featureFlags.IsFlagEnabled("ValueMustBeLessThan11"))
            ArgumentOutOfRangeException.ThrowIfGreaterThanOrEqual(inputValue, 11);
    }
}

优化后的测试类 #

using Xunit;
using Moq;

public class UnitTest2
{
    // 提取原有偶数测试数据
    public static readonly TheoryData<int> EvenNumbers = new() { 2, 4, 6 };
    // 提取原有奇数测试数据
    public static readonly TheoryData<int> OddNumbers = new() { 1, 3, 5 };

    // 生成开关状态与偶数数据的组合
    public static IEnumerable<object[]> EvenNumbersWithFlagStates()
    {
        foreach (var number in EvenNumbers)
        {
            yield return new object[] { number, true };
            yield return new object[] { number, false };
        }
    }

    // 生成开关状态与奇数数据的组合
    public static IEnumerable<object[]> OddNumbersWithFlagStates()
    {
        foreach (var number in OddNumbers)
        {
            yield return new object[] { number, true };
            yield return new object[] { number, false };
        }
    }

    private readonly Mock<IFeatureFlags> _mockFeatureFlags;
    private readonly TestSubject _subject;

    public UnitTest2()
    {
        _mockFeatureFlags = new Mock<IFeatureFlags>();
        _subject = new TestSubject(_mockFeatureFlags.Object);
    }

    [Theory]
    [MemberData(nameof(EvenNumbersWithFlagStates))]
    public void WhenValueIsEven_ThenThrowsException(int inputValue, bool isFlagEnabled)
    {
        // 设置当前测试用例的开关状态
        _mockFeatureFlags.Setup(x => x.IsFlagEnabled("ValueMustBeLessThan11"))
                         .Returns(isFlagEnabled);

        var exception = Assert.Throws<ArgumentException>(() => _subject.EnsureNumberIsOdd(inputValue));
        Assert.Equal("inputValue", exception!.ParamName);
    }

    [Theory]
    [MemberData(nameof(OddNumbersWithFlagStates))]
    public void WhenValueIsOdd_ThenDoesNotThrowException(int inputValue, bool isFlagEnabled)
    {
        _mockFeatureFlags.Setup(x => x.IsFlagEnabled("ValueMustBeLessThan11"))
                         .Returns(isFlagEnabled);

        _subject.EnsureNumberIsOdd(inputValue);
    }

    [Fact]
    public void WhenFeatureFlagIsOn_AndValueIsNotLessThan11_ThenThrowsException()
    {
        _mockFeatureFlags.Setup(x => x.IsFlagEnabled("ValueMustBeLessThan11"))
                         .Returns(true);

        foreach (int i in new[] { 11, 13, 15, 17, 19 })
        {
            var exception = Assert.Throws<ArgumentOutOfRangeException>(() => _subject.EnsureNumberIsOdd(i));
            Assert.Equal("inputValue", exception!.ParamName);
        }
    }
}

方案优势 #

• 测试完全隔离：每个测试用例都是独立执行，Mock状态不会在不同用例间污染
• 维护成本低：新增测试数据或功能开关时，只需修改静态数据或组合逻辑，无需复制测试方法或属性
• 扩展性强：多开关场景下，只需在组合数据中添加新的开关状态参数，自动生成所有场景的测试用例

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