使用Moq模拟自定义集合

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

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 s的假集合(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);
    ...
}

但是我不能这样做，因为

无法从"System.Collections.Generic"转换。列表>' 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>())
{
    ...
}

[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);
}

var fakeWorksheet = new Mock<IWorksheet>();
//You can use AutoFixture here to auto-populate properties or you can set only required props
fakeWorksheet.Setup(p => p.Name).Returns("TestName");
var worksheetsMock = new Mock<IWorksheets>()
//here mock some members that you need
worksheetsMock.Setup(w => w.Add()).Returns(fakeWorksheet.Object);
var workbookMock = new Mock<IWorkbook>();
workbookMock.Setup(w => w.Worksheets).Returns(worksheetsMock.Object);