我有一个简单的文档管理器,它被插入到asp.net c#MVC项目中的控制器中。该项目是数据库优先的,并且Document
表由索引documentId
,该索引是一个自动递增的整数。
我一直在尝试编写一个测试,测试的以下实现,该测试CreateNewDocument
在成功添加文档后会对其进行查找并返回新的文档ID。
问题是我找不到一种模拟方法MyEntityFrameWorkEntities
,可以向linq添加文档,然后搜索该文档。我认为这是行不通的,因为被嘲弄的_context.Document.Add
人实际上并没有做任何事情。
我的问题是:我可以以不同的方式设置模拟,以便可以保持现状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.CreateNewDocument(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(),
};
}
您可以使用回调设置模拟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()的后续调用应能够检索该项目。
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句