c# - 使用 Moq 模拟自定义集合

Question

我在第三方库中有以下类型

public interface IWorkbook : IPrintable
{
    ...
    IWorksheets Worksheets { get; }
}

工作表界面是

public interface IWorksheets : IEnumerable
{
    IWorksheet this[int index] { get; }
    IWorksheet this[string name] { get; }
    IWorkbookSet WorkbookSet { get; }
    IWorksheet Add();
    IWorksheet AddAfter(ISheet sheet);
    IWorksheet AddBefore(ISheet sheet);
    bool Contains(IWorksheet worksheet);
}

我有一个想要对其进行单元测试的方法，该方法采用 IWorkbook 并迭代所包含的工作表。我的问题是如何使用 Moq 为 IWorksheets 创建模拟集合。 IWorksheet 界面是

public IWorksheet
{
    ...
    public Name { get; set; } // This is the only property I am interested in.
}

那么如何生成一个假的 IWorkbook ，其中包含 IWorksheet 的假集合(IWorksheets)？

我已经开始

[TestInitialize]
public void Initialize()
{
    List<string> fakeSheetNames = new List<string>()
    {
        "Master",
        "A",
        "B",
        "C",
        "__ParentA", 
        "D",
        "wsgParentB", 
        "E",
        "F",
        "__ParentC",
        "__ParentD",
        "G"
    };

    List<IMock<IWorksheet>> sheetMockList = new List<IMock<IWorksheet>>();
    foreach (string name in fakeSheetNames)
    {
        Mock<IWorksheet> tmpMock = new Mock<IWorksheet>();
        tmpMock.Setup(p => p.Name).Returns(name);
        sheetMockList.Add(tmpMock);
    }

    var mockWorksheets = new Mock<IWorksheets>();
    mockWorksheets.Setup(p => p).Returns(sheetMockList);
    ...
}

但我不能这样做(显然)

cannot convert from 'System.Collections.Generic.List>' to 'SpreadsheetGear.IWorksheets'

如何模拟 IWorksheets 集合？

---

所以我现在有以下代码来根据下面的答案创建我的模拟

    [TestClass]
public class WorkbookStrucutreProviderTests
{
    private Mock<IWorkbookSet> mockWorkbookSet;
    private readonly List<string> parentPrefixes = new List<string>() { "__", "wsg" };

    [TestInitialize]
    public void Initialize()
    {
        List<string> fakeSheetNames = new List<string>()
        {
            "Master",
            "A",
            "B",
            "C",
            "__ParentA", 
            "D",
            "wsgParentB", 
            "E",
            "F",
            "__ParentC",
            "__ParentD",
            "G"
        };

        List<IWorksheet> worksheetMockList = new List<IWorksheet>();
        foreach (string name in fakeSheetNames)
        {
            Mock<IWorksheet> tmpMock = new Mock<IWorksheet>();
            tmpMock.Setup(p => p.Name).Returns(name);
            tmpMock.Setup(p => p.Visible)
                .Returns(parentPrefixes.Any(p => name.StartsWith(p)) ? 
                    SheetVisibility.Hidden : 
                    SheetVisibility.Visible);

            worksheetMockList.Add(tmpMock.Object);
        }

        List<IWorkbook> workbookMockList = new List<IWorkbook>();
        Mock<IWorkbook> mockWorkbook = new Mock<IWorkbook>();
        mockWorkbook
            .Setup(p => p.Worksheets.GetEnumerator())
            .Returns(worksheetMockList.GetEnumerator());
        workbookMockList.Add(mockWorkbook.Object);

        mockWorkbookSet = new Mock<IWorkbookSet>();
        mockWorkbookSet
            .Setup(p => p.Workbooks.GetEnumerator())
            .Returns(workbookMockList.GetEnumerator());
    }

    [TestMethod]
    public async Task StrucutreGenerationAsyncTest()
    {
        WorkbookStructureProvider provider = new WorkbookStructureProvider();
        await provider.GenerateWorkbookStructureAsync(mockWorkbookSet.Object);

        foreach (var item in provider.Structure)
        {
            Trace.WriteLine("--" + item.Name);
            if (item.HasChildren)
            {
                foreach (var child in item.Children)
                {
                    Trace.WriteLine("-- --" + child.Name);
                }
            }
        }
    }

但是在 GenerateWorkbookStructureAsync() 方法中我有这段代码

bool IsUserCostWorkbook = false;
if (workbook.Worksheets.Cast<IWorksheet>().Any(
    ws => ws.Name.CompareNoCase(Keywords.Master)))
{
    // TODO Extra check for UserCost template.
    IsUserCostWorkbook = true;          
}

这里的workbook.Worksheets集合是空的。我认为我的模拟 GetEnumerator 可以处理这个问题；事实并非如此。

那么我如何模拟 IWorksheets 以便我可以执行以下操作？

foreach (var ws in workbook.Worksheets.Cast<IWorksheet>())
{
    ...
}

Answer 1

以下示例经测试通过

[TestMethod]
public void Mock_Custom_Collection_Using_Moq() {
    //Arrange
    var parentPrefixes = new List<string>() { "__", "wsg" };
    var fakeSheetNames = new List<string>(){
        "Master",
        "A",
        "B",
        "C",
        "__ParentA", 
        "D",
        "wsgParentB", 
        "E",
        "F",
        "__ParentC",
        "__ParentD",
        "G"
    };

    var worksheetMockList = new List<IWorksheet>();
    foreach (string name in fakeSheetNames) {
        var worksheet = Mock.Of<IWorksheet>();
        worksheet.Name = name;
        worksheet.Visible = parentPrefixes.Any(p => name.StartsWith(p)) ?
                SheetVisibility.Hidden :
                SheetVisibility.Visible;

        worksheetMockList.Add(worksheet);
    }

    var mockWorkbook = new Mock<IWorkbook>();
    mockWorkbook
        .Setup(p => p.Worksheets.GetEnumerator())
        .Returns(() => worksheetMockList.GetEnumerator());

    var workbook = mockWorkbook.Object;

    //Act
    bool IsUserCostWorkbook = false;
    if (workbook.Worksheets.Cast<IWorksheet>()
        .Any(ws => ws.Name.Equals("Master", StringComparison.InvariantCultureIgnoreCase))) {
        IsUserCostWorkbook = true;
    }

    //Assert            
    Assert.IsTrue(IsUserCostWorkbook);
}