c# - 如何模拟添加到 DbContext 的记录

Question

我有这样的方法:

private void CreateTaskFromModel(ForgotPasswordViewModel fpModel)
{
    var message = _dbContext.Create<Message>();

    message.MessageType = "TASK".PadLeft(10);
    message.Assigned_User_K = fpModel.SendPasswordRequestTo.Trim();
    message.Assigned_Date = DateTime.Today;
    message.Source_User_K = string.Empty;
    message.Title_Focus = "Request Web Password";

    _dbContext.Messages.Add(message);
}

所以我能够像这样模拟 Create 方法:

Message msg = new Message();
IMyDbContext fakeDbContext = NSubstitute.Substitute.For<IMyDbContext>();

fakeDbContext.Create<Message>().Returns(msg);

但是请注意最后一行它仍然有这段代码，我不知道如何模拟这段代码？

_dbContext.Messages.Add(message);

我正在使用 NSubstittue

Answer 1

消息类型为 DbSet<Message> .所以你需要创建一个 DbSet<Message> 的假/替代实例并让我在 _dbContext.Messages 之前返回.

Message msg = new Message();
IMyDbContext fakeDbContext = NSubstitute.Substitute.For<IMyDbContext>();

var messagesSet = NSubstitute.Substitute.For<DbSet<Message>>();
fakeDbContext.Create<Message>().Returns(msg);
fakeDbContext.Messages.Returns(messagesSet);

您也有兴趣查看是否调用了 Add。您可以在断言中执行此操作。

messagesSet.Received().Add(NSubstitute.Any<Message>());

或者如果您想测试是否从您的创建中添加了相同的实例

messagesSet.Received().Add(msg); // message created earlier in your test