模拟异步查询操作
Posted
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了模拟异步查询操作相关的知识,希望对你有一定的参考价值。
我创建了单元测试,我必须模拟我的EF的context
。我使用Moq
库和Xunit
。我有一个这样的测试方法:
[Fact]
public async Task DeleteAttachmentCommandHandler_WithValidCommand_ShouldCallSaveChangesAsyncOnce()
{
var command = new DeleteAttachmentCommand { Id = Guid.NewGuid() };
var attachments = new List<Attachment>();
var dbSetMock = attachments.AsQueryable().BuildMockDbSetForAsyncQueryOperations();
_context.Setup(x => x.Set<Attachment>()).Returns(dbSetMock.Object);
_context.Setup(x => x.SaveChangesAsync(It.IsAny<CancellationToken>())).ReturnsAsync(1).Verifiable();
await Act(command);
_context.Verify(x => x.SaveChangesAsync(It.IsAny<CancellationToken>()), Times.Once);
}
_context
是Mock<IEmployeeSettlementsDbContext>
类型
由于MSDN文档 - > BuildMockDbSetForAsyncQueryOperations
,https://docs.microsoft.com/en-us/ef/ef6/fundamentals/testing/mocking是我的扩展方法,它使用像TestDbAsyncEnumerable
,TestDbAsyncEnumerator
,TestDbAsyncQueryProvider
这样的异步提供程序。我的扩展BuildMockDbSetForAsyncQueryOperations
看起来像这样:
public static Mock<IDbSet<TEntity>> BuildMockDbSetForAsyncQueryOperations<TEntity>(this IQueryable<TEntity> data) where TEntity : class
{
var dbSetMock = new Mock<IDbSet<TEntity>>();
dbSetMock.As<IDbAsyncEnumerable<TEntity>>().Setup(x => x.GetAsyncEnumerator()).Returns(new TestDbAsyncEnumerator<TEntity>(data.GetEnumerator()));
dbSetMock.As<IQueryable<TEntity>>().Setup(x => x.Provider).Returns(new TestDbAsyncQueryProvider<TEntity>(data.Provider));
dbSetMock.As<IQueryable<TEntity>>().Setup(x => x.Expression).Returns(data.Expression);
dbSetMock.As<IQueryable<TEntity>>().Setup(x => x.ElementType).Returns(data.ElementType);
dbSetMock.As<IQueryable<TEntity>>().Setup(x => x.GetEnumerator()).Returns(data.GetEnumerator());
return dbSetMock;
}
在我测试的实际处理程序方法中,我有一行:
var attachment = await _context.Set<Attachment>()
.SingleAsync(x => x.Id == command.Id);
当我运行测试时它失败并向我显示消息
消息:System.InvalidOperationException:序列不包含匹配的元素。
但是当我将处理程序中的查询更改为ListAsync()
而不是SingleAsync()
时,模拟设置正确并返回一个空列表。但它对使用SingleAsync()
的单个元素不起作用。
编辑:这是我的完整处理程序方法:
public async Task<Unit> Handle(DeleteAttachmentCommand command, CancellationToken cancellationToke = default(CancellationToken))
{
ValidateCommand(command);
var attachment = await _context.Set<Attachment>().SingleAsync(x => x.Id == command.Id);
attachment.IsDeleted = true;
await _context.SaveChangesAsync();
return Unit.Value;
}
在SingleAsync()
的MSDN文档
异步返回序列的唯一元素,如果序列中没有一个元素,则抛出异常。
考虑使用FirstOrDefault()
or FirstOrDefaultAsync()
方法而不是SingleAsync()
。 Here和here是关于此的链接,这不会抛出异常。
以上是关于模拟异步查询操作的主要内容,如果未能解决你的问题,请参考以下文章