捕获Console输出的单元测试Assert结果异常排查求助

问题 #

我是编程新手，正在做简单算法练习，知道写单元测试是好的开发习惯。现在尝试捕获Console输出用于Assert断言，但遇到了异常问题。

待测试函数 #

该函数已通过HackerRank测试，功能正常：

public static void MinMaxSum(List<int> arr)
{
    long minSum = long.MaxValue;
    long maxSum = long.MinValue;
    long totalSum = default;
    int n = arr.Count;
    for (int i = 0; i < n; i++)
    {
        totalSum += arr[i];
    }
    for (int i = 0; i < n; i++)
    {
        long currentSum = totalSum - arr[i];
        if (currentSum > maxSum)
        {
            maxSum = currentSum;
        }
        if (currentSum < minSum)
        {
            minSum = currentSum;
        }
    }
    Console.WriteLine(minSum + " " + maxSum);
}

单元测试代码 #

namespace LCMinMaxSumTests
{
    [TestClass]
    public sealed class Test1
    {
        private TextWriter originalOut = null!;
        private StringWriter sw = null!;

        [TestInitialize]
        public void SetOriginalConsoleOutput()
        {
            originalOut = Console.Out;
            sw = new StringWriter();
            Console.SetOut(sw);
        }

        [TestCleanup]
        public void RestoreOriginalConsoleOutput()
        {
            Console.SetOut(originalOut);
            sw.Flush();
            sw.Close();
            sw.Dispose();
        }

        [TestMethod]
        public void TestMethod1()
        {
            List<int> list = new List<int> { 1, 3, 5, 7, 9 };
            HCRNKMinMaxSum.Program.MinMaxSum(list);
            string expected = string.Format($"16 24{Environment.NewLine}");
            Assert.AreEqual(expected, sw.ToString());
        }

        [TestMethod]
        public void TestMethod2()
        {
            List<int> list = new List<int> { 1, 2, 3, 4, 5 };
            HCRNKMinMaxSum.Program.MinMaxSum(list);
            string expected = string.Format($"10 14{Environment.NewLine}");
            Assert.AreEqual(expected, sw.ToString());
        }
    }
}

问题现象 #

• 第一个TestMethod的Assert结果是两个测试的输出组合（10 14 16 24）
• 第二个TestMethod的Assert结果为空
• 尝试用using包裹StringWriter，问题依旧
• 当前环境：VS 17.12.3、.NET 9

问题原因 #

核心问题是测试类实例复用导致的状态污染：

1. MSTest框架为了性能会复用测试类的实例，类字段sw会在不同测试方法之间共享，前一次测试的输出内容没有被清空，导致叠加。
2. 在TestCleanup中直接Dispose了sw，第二个测试执行时sw已经被释放，无法捕获新的Console输出。

解决方法 #

方法一：局部变量替代类字段（快速修复） #

将StringWriter和originalOut改为测试方法内的局部变量，彻底避免跨测试共享状态：

[TestMethod]
public void TestMethod1()
{
    var originalOut = Console.Out;
    using var sw = new StringWriter();
    try
    {
        Console.SetOut(sw);
        List<int> list = new List<int> { 1, 3, 5, 7, 9 };
        HCRNKMinMaxSum.Program.MinMaxSum(list);
        string expected = $"16 24{Environment.NewLine}";
        Assert.AreEqual(expected, sw.ToString());
    }
    finally
    {
        Console.SetOut(originalOut);
    }
}

[TestMethod]
public void TestMethod2()
{
    var originalOut = Console.Out;
    using var sw = new StringWriter();
    try
    {
        Console.SetOut(sw);
        List<int> list = new List<int> { 1, 2, 3, 4, 5 };
        HCRNKMinMaxSum.Program.MinMaxSum(list);
        string expected = $"10 14{Environment.NewLine}";
        Assert.AreEqual(expected, sw.ToString());
    }
    finally
    {
        Console.SetOut(originalOut);
    }
}

方法二：重构待测试函数（最佳实践） #

直接输出到Console的函数耦合性高，难以测试。更优的做法是拆分逻辑：让函数返回计算结果，单独处理打印逻辑：

// 仅负责计算的核心函数，便于测试
public static (long MinSum, long MaxSum) CalculateMinMaxSum(List<int> arr)
{
    long minSum = long.MaxValue;
    long maxSum = long.MinValue;
    long totalSum = default;
    int n = arr.Count;
    for (int i = 0; i < n; i++)
    {
        totalSum += arr[i];
    }
    for (int i = 0; i < n; i++)
    {
        long currentSum = totalSum - arr[i];
        if (currentSum > maxSum)
        {
            maxSum = currentSum;
        }
        if (currentSum < minSum)
        {
            minSum = currentSum;
        }
    }
    return (minSum, maxSum);
}

// 保留原打印函数，用于实际运行场景
public static void MinMaxSum(List<int> arr)
{
    var (minSum, maxSum) = CalculateMinMaxSum(arr);
    Console.WriteLine($"{minSum} {maxSum}");
}

对应的单元测试可以直接断言返回值，无需捕获Console输出：

[TestMethod]
public void CalculateMinMaxSum_Test1()
{
    List<int> list = new List<int> { 1, 3, 5, 7, 9 };
    var result = HCRNKMinMaxSum.Program.CalculateMinMaxSum(list);
    Assert.AreEqual(16, result.MinSum);
    Assert.AreEqual(24, result.MaxSum);
}

[TestMethod]
public void CalculateMinMaxSum_Test2()
{
    List<int> list = new List<int> { 1, 2, 3, 4, 5 };
    var result = HCRNKMinMaxSum.Program.CalculateMinMaxSum(list);
    Assert.AreEqual(10, result.MinSum);
    Assert.AreEqual(14, result.MaxSum);
}

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