C#实现单次枚举IEnumerable及搭配Moq编写单测的最优方案

通用C#场景的无依赖实现 #

完全不需要手动从头实现IEnumerable<T>和IEnumerator<T>两个接口，利用C#原生迭代器特性就能写出等价行为，代码量极小：

• 所有用yield return生成的枚举器，原生就会在调用Reset()方法时抛出NotSupportedException，天然匹配你之前手写OnceEnumerator的行为要求，不需要额外写枚举器的逻辑
• 只需要在外层用闭包捕获一个访问标记位，阻断第二次GetEnumerator()的调用即可，完整实现不到15行代码：

public static IEnumerable<T> CreateSingleUseEnumerable<T>(IEnumerable<T> source)
{
    // 标记位被外层方法闭包捕获，不会随GetEnumerator调用重置
    bool getEnumeratorCalled = false;
    return EnumerateWrapper();

    IEnumerable<T> EnumerateWrapper()
    {
        if (getEnumeratorCalled)
        {
            throw new NotSupportedException("仅支持对该IEnumerable执行单次枚举");
        }
        getEnumeratorCalled = true;
        // 内部用yield遍历源数据，自带Reset异常行为
        foreach (var item in source)
        {
            yield return item;
        }
    }
}

• 这个实现和你之前手写的两个自定义类行为完全对齐：第一次枚举可正常遍历源数据、第二次及以上调用GetEnumerator直接抛异常、枚举器调用Reset自动抛异常，且没有任何第三方依赖，任意C#开发场景都可以直接用。

Moq框架下的测试适配方案 #

Moq本身没有提供开箱即用的单次枚举内置类型，但完全不需要手写独立的自定义枚举类，两种简便方式就能达到测试效果：

• 直接复用上述通用实现，在配置Mock返回值时包装测试数据即可，示例代码：

  var mockClient = new Mock<IDemoClient>();
  // 直接把测试用的List数据包装成单次枚举实例
  mockClient.Setup(c => c.GetData())
      .Returns(CreateSingleUseEnumerable(new List<string>{"测试值1", "测试值2"}));
  

• 如果不想写独立的包装方法，可以直接在Mock配置环节完成全行为模拟：用Moq分别模拟IEnumerable<T>和IEnumerator<T>的行为，给GetEnumerator加调用次数限制、给Reset配置抛异常的规则，所有逻辑都在测试setup块里写完，不需要单独定义OnceEnumerable/OnceEnumerator类。
• 之前单元测试漏判问题的核心原因，就是测试环节用的List<T>是支持无限次遍历、支持Reset的全功能实现，和生产环境的单次枚举行为完全错位；只要把测试环节的返回值替换成上述实现，就能完整覆盖调用方重复枚举、误调用Reset的异常路径。

常见踩坑提醒：别直接把带yield return的方法返回值当成单次枚举对象——这类方法每次调用GetEnumerator时都会重置方法内部的局部状态，如果没有外层闭包的标记位阻断，是支持任意次数遍历的，必须加上访问次数判断才能匹配单次枚举的约束。

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