Question

我有这个域名：

public class ADomainClass
{
    public int Id { get; set; }
}

public interface IMyClass : IEnumerable<ADomainClass>
{
}

public class MyClass : IMyClass
{
    public IEnumerator<ADomainClass> GetEnumerator()
    {
        IList<ADomainClass> list = new List<ADomainClass>();
        //list = GetData...();

        foreach (var item in list)
        {
            yield return item;
        }
    }

    ...
}

并希望构建以下测试：

[Test]
public void TestSample()
{
    //Arrange
    IMyClass myclass = Substitute.For<IMyClass>();
    IList<ADomainClass> testdata = new List<ADomainClass>()
    {
        new ADomainClass(){ Id = 1, },
        new ADomainClass(){ Id = 2, },
        new ADomainClass(){ Id = 3, },
        new ADomainClass(){ Id = 4, },
    };
    int count = 0;

    myclass.ReturnsForAnyArgs(testdata); //How to set the "return" value?

    //Act
    foreach (ADomainClass item in myclass)
    {
        count++;
    }

    //Assert
    count.Should().Be(testdata.Count);
}

设置方法的返回值很简单，如下所示：

myclass.GetData().Returns(data);

我不记得当它是一个可枚举的类时如何设置返回值。我之前已经解决了这个问题，但不记得我在哪里使用它了。

Answer 1

您只需要告诉您的替代人员从您的测试数据中返回枚举器：

myclass.GetEnumerator().Returns(testdata.GetEnumerator());

Answer 2

我的问题是NSubstitute存储了枚举的状态，因此如果您在被测系统中两次使用枚举器，它将在第二次继续第二个元素。我使用下面的代码作为解决方法（使用lambda而不是返回Enumerator，第1 + 2行仅用于演示）。我只用一些Linq和foreach进行了测试。

var bar = new List<string> { "a", "b", "c" };
var foo = Substitute.For<ICollection<string>>();
foo.GetEnumerator().Returns((ci) => { return bar.GetEnumerator(); });

如何使用NSubstitute设置IEnumerable类的返回值

2 个答案: