MSTest中失败测试重试：自定义RetryAttribute无效问题排查

原代码存在的问题 #

• 循环逻辑错误：for (int i = 0; i <= _tryCount; i++)会导致测试执行_tryCount + 1次，超出预期的重试次数。
• 返回结果处理不当：返回所有执行的TestResult数组，MSTest会显示多条测试记录，且无法正确识别最终的测试状态。
• 测试实例未重置：复用同一个测试类实例，之前的测试状态会残留，导致重试时逻辑异常。

修正后的重试特性代码 #

[AttributeUsage(AttributeTargets.Method, AllowMultiple = false)]
public class RetryAttribute : TestMethodAttribute
{
    private readonly int _maxAttempts;

    public RetryAttribute(int maxAttempts)
    {
        // 确保重试次数至少为1
        _maxAttempts = maxAttempts < 1 ? 1 : maxAttempts;
    }

    public override TestResult[] Execute(ITestMethod testMethod)
    {
        TestResult finalResult = null;

        for (int attempt = 1; attempt <= _maxAttempts; attempt++)
        {
            // 每次重试创建全新的测试类实例，清除状态
            var testInstance = Activator.CreateInstance(testMethod.TestClass.ClassType);
            // 获取测试方法的默认参数
            var parameters = testMethod.GetParameters()
                                      .Select(param => param.DefaultValue)
                                      .ToArray();

            try
            {
                // 直接调用测试方法
                testMethod.Invoke(testInstance, parameters);
                finalResult = new TestResult
                {
                    Outcome = UnitTestOutcome.Passed,
                    TestMethodName = testMethod.TestMethodName,
                    DisplayName = $"Attempt {attempt}: {testMethod.TestMethodName}"
                };
                // 测试通过则终止重试
                break;
            }
            catch (Exception ex)
            {
                finalResult = new TestResult
                {
                    Outcome = UnitTestOutcome.Failed,
                    TestMethodName = testMethod.TestMethodName,
                    DisplayName = $"Attempt {attempt}: {testMethod.TestMethodName}",
                    TestFailureException = ex
                };

                // 未达到最大次数时记录重试日志
                if (attempt < _maxAttempts)
                {
                    System.Diagnostics.Debug.WriteLine(
                        $"Retry {attempt}/{_maxAttempts - 1} for test {testMethod.TestMethodName}"
                    );
                }
            }
        }

        // 返回最终的测试结果，仅显示一条记录
        return new[] { finalResult };
    }
}

关键修改说明 #

1. 修复循环逻辑：用attempt <= _maxAttempts控制执行次数，传入3则最多执行3次（2次重试），符合预期。
2. 重置测试实例：每次重试通过Activator.CreateInstance创建新的测试类对象，彻底清除之前的状态残留。
3. 直接控制测试执行：调用testMethod.Invoke替代base.Execute，更精准捕获测试异常，判断执行结果。
4. 返回单一结果：仅返回最后一次执行的结果，避免TestExplorer中出现多条重复测试记录，正确标记最终状态。

使用示例 #

[TestClass]
public class SampleTests
{
    // 允许最多3次执行（2次重试）
    [Retry(3)]
    [TestMethod]
    public void FlakyIntegrationTest()
    {
        // 偶尔会失败的测试逻辑，比如依赖外部服务的接口测试
        var success = new Random().Next(0, 2) == 1;
        Assert.IsTrue(success, "随机模拟测试失败");
    }
}

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