我可以创建一个模拟数据库上下文，我可以添加，然后搜索？

Question

我有一个简单的文件管理器，它被注入到我的控制器在asp.net c＃MVC项目中。该项目以数据库优先，Document表索引为documentId，一个自动递增整数。

我一直在试图编写一个测试，测试以下实现CreateNewDocument，它在成功添加文档后查找并返回新的文档ID。

问题是，我找不到方法来模拟MyEntityFrameWorkEntities，我可以添加一个文档，然后使用linq搜索该文档。我认为它不起作用，因为嘲讽的_context.Document.Add没有真正做任何事情。

我的问题是这样的：我需要改变我的DocumentManager，这样的代码是（通过与.FirstOrDefault更换.First并显示，如果返回null函数返回零例如）可测试的，或者我可以（我应该）集不同的我的嘲笑，所以我可以离开DocumentManager照原样，并写一个测试通过？

public class DocumentManager : IDocumentManager 
{ 
    private readonly MyEntityFrameWorkEntities _context; 

    public DocumentManager(MyEntityFrameWorkEntities context) 
    { 
     _context = context; 
    } 

    public int CreateNewDocument(int userId) 
    { 
     var newDocumentGuid = Guid.NewGuid(); 
     var newDocument = new Document 
     { 
      UserId = userId, 
      DateCreated = DateTime.Now, 
      DocumentGuid = newDocumentGuid 
     }; 
     _context.Document.Add(newDocument); 
     _context.SaveChanges(); 
     // the .First here doesn't return anything when called from tests 
     return _context.Document.First(d => d.DocumentGuid == newDocumentGuid).DocumentId; 
    } 
} 

public partial class MyEntityFrameWorkEntities : DbContext 
{ 
    public MyEntityFrameWorkEntities() : base("name=MyEntityFrameWorkEntities") 
    { 
    } 

    public virtual DbSet<Document> Document { get; set; } 
    /* ...etc... */ 
} 

和测试类：

[TestMethod] 
public void TestCreateNewDocument() 
{ 
    var mockContext = new Mock<MyEntityFrameWorkEntities>(); 

    var mockDocumentDbSet = GetQueryableMockDocumentDbSet(); 

    mockContext.Setup(m => m.Document).Returns(mockDocumentDbSet.Object); 

    var documentManager = new DocumentManager(mockContext.Object); 

    var newDocId = documentManager.CreateNewDocument2(123); 

    // This line doesn't get hit as the .First falls over before here 
    Assert.AreNotEqual(newDocId, 0); 
} 

private static Mock<DbSet<Document>> GetQueryableMockDocumentDbSet() 
{ 
    var data = new List<Document> { GetDocument(111, 11), GetDocument(222, 22), GetDocument(333, 33) }.AsQueryable(); 
    var mockDocumentDbSet = new Mock<DbSet<Document>>(); 
    mockDocumentDbSet.As<IQueryable<Document>>().Setup(m => m.Provider).Returns(data.Provider); 
    mockDocumentDbSet.As<IQueryable<Document>>().Setup(m => m.Expression).Returns(data.Expression); 
    mockDocumentDbSet.As<IQueryable<Document>>().Setup(m => m.ElementType).Returns(data.ElementType); 
    mockDocumentDbSet.As<IQueryable<Document>>().Setup(m => m.GetEnumerator()).Returns(data.GetEnumerator()); 
    return mockDocumentDbSet; 
} 

private static Document GetDocument(int documentId, int userId) 
{ 
    return new Document 
    { 
     DocumentId = documentId, 
     UserId = userId, 
     DateCreated = DateTime.Now.AddDays(-1), 
     DocumentGuid = Guid.NewGuid(), 
    }; 
} 

Answer 1

你可以设置你的模拟DbSet的Add（）方法的回调将在您的支持列表添加菜单项：

private static Mock<DbSet<Document>> GetQueryableMockDocumentDbSet() 
{ 
    var data = new List<Document> { GetDocument(111, 11), GetDocument(222, 22), GetDocument(333, 33) }; 

    var mockDocumentDbSet = new Mock<DbSet<Document>>(); 
    mockDocumentDbSet.As<IQueryable<Document>>().Setup(m => m.Provider).Returns(data.AsQueryable().Provider); 
    mockDocumentDbSet.As<IQueryable<Document>>().Setup(m => m.Expression).Returns(data.AsQueryable().Expression); 
    mockDocumentDbSet.As<IQueryable<Document>>().Setup(m => m.ElementType).Returns(data.AsQueryable().ElementType); 
    mockDocumentDbSet.As<IQueryable<Document>>().Setup(m => m.GetEnumerator()).Returns(data.GetEnumerator()); 
    mockDocumentDbSet.Setup(m => m.Add(It.IsAny<Document>())).Callback<Document>(data.Add); 
    return mockDocumentDbSet; 

}

你的后续调用First（）应该能够检索该项目。

Answer 2

考虑在较高的抽象层嘲笑出来。 在这种情况下，考虑嘲笑Respository。 你甚至可以更高兴地嘲笑服务本身。

Construct testable business layer logic