MSTest中如何获取DynamicData驱动测试的当前迭代序号？

MSTest框架本身没有直接提供获取DynamicData测试迭代序号的API，但可以通过两种无需修改原始测试数据结构的简洁方案实现，替代你目前手动添加序号参数的临时方案。

方案一：解析TestContext的测试名称 #

MSTest会为DynamicData的每个测试迭代自动生成包含参数值的测试名称（格式为测试方法名(参数值)），我们可以通过TestContext获取这个名称，再结合原始数据集的顺序匹配出当前迭代的序号。

实现步骤： #

1. 在测试类中注入TestContext：

public class YourTestClass
{
    public TestContext TestContext { get; set; }

    // 测试方法和数据定义...
}

2. 在测试方法中解析序号：

[TestMethod]
[DynamicData(nameof(TheData), typeof(TheSetup), DynamicDataSourceType.Method)]
public async Task Task_Should_Succeed(int id)
{
    // 从测试名称中提取参数值
    var paramPart = TestContext.TestName.Split('(', ')')[1];
    if (!int.TryParse(paramPart, out var paramValue))
    {
        Assert.Fail("无法解析测试参数值");
    }

    // 将原始数据集转为列表，按顺序查找当前参数的索引
    var dataList = TheSetup.TheData().ToList();
    var currentIdxOfTest = dataList.FindIndex(item => (int)item[0] == paramValue);

    if (currentIdxOfTest == 1) 
        Assert.IsTrue(id == 60);
}

注意：该方案适用于参数为简单类型（如int、string）的场景，且需保证原始数据集的顺序固定不变。

方案二：自定义DynamicData特性传递序号 #

如果需要更可靠的实现（比如支持复杂参数、并行测试），可以自定义一个继承自DynamicDataAttribute的特性，通过AsyncLocal在测试迭代间传递序号，无需修改测试方法签名或原始数据集。

实现步骤： #

1. 创建自定义特性：

public class DynamicDataWithIndexAttribute : DynamicDataAttribute
{
    private static readonly AsyncLocal<int?> _currentIterationIndex = new AsyncLocal<int?>();

    public DynamicDataWithIndexAttribute(string dataSourceName, Type dataSourceType) 
        : base(dataSourceName, dataSourceType)
    {
    }

    public override IEnumerable<object[]> GetData(MethodInfo methodInfo)
    {
        var originalData = base.GetData(methodInfo);
        int index = 0;
        foreach (var dataItem in originalData)
        {
            _currentIterationIndex.Value = index;
            yield return dataItem;
            index++;
        }
    }

    public static int? GetCurrentIterationIndex() => _currentIterationIndex.Value;
}

2. 在测试中使用自定义特性并获取序号：

[TestMethod]
[DynamicDataWithIndex(nameof(TheData), typeof(TheSetup))]
public async Task Task_Should_Succeed(int id)
{
    var currentIdxOfTest = DynamicDataWithIndexAttribute.GetCurrentIterationIndex();
    if (currentIdxOfTest == 1) 
        Assert.IsTrue(id == 60);
}

该方案支持并行测试（AsyncLocal保证线程隔离），且不受参数类型限制，是更通用的解决方案。

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