使用Moq模拟参数的属性更改

时间:2008-10-08 08:40:48

标签: c# tdd mocking moq

我正在使用Moq来模拟我的Repository层,以便进行单元测试。

我的存储库层插入方法在成功进行数据库插入时更新我的​​实体的Id属性。

如何在调用Insert方法时配置moq更新实体的Id属性?

存储库代码: -

void IAccountRepository.InsertAccount(AccountEntity account);

单元测试: -

[TestInitialize()]
public void MyTestInitialize() 
{
    accountRepository = new Mock<IAccountRepository>();
    contactRepository = new Mock<IContactRepository>();
    contractRepository = new Mock<IContractRepository>();
    planRepository = new Mock<IPlanRepository>();
    generator = new Mock<NumberGenerator>();

    service = new ContractService(contractRepository.Object, accountRepository.Object, 
                    planRepository.Object, contactRepository.Object, generator.Object);   
}


[TestMethod]
public void SubmitNewContractTest()
{
    // Setup Mock Objects
    planRepository
        .Expect(p => p.GetPlan(1))
        .Returns(new PlanEntity() { Id = 1 });

    generator
        .Expect(p => p.GenerateAccountNumber())
        .Returns("AC0001");

    // Not sure what to do here? 
    // How to mock updating the Id field for Inserts?
    //                                                 
    // Creates a correctly populated NewContractRequest instance
    NewContractRequest request = CreateNewContractRequestFullyPopulated();

    NewContractResponse response = service.SubmitNewContract(request);
    Assert.IsTrue(response.IsSuccessful);
}
来自ContractService类(WCF服务合同)的

实现片段。

AccountEntity account = new AccountEntity()
{
    AccountName = request.Contact.Name,
    AccountNumber = accountNumber,
    BillingMethod = BillingMethod.CreditCard,
    IsInvoiceRoot = true,
    BillingAddressType = BillingAddressType.Postal,
    ContactId = request.Contact.Id.Value
};

accountRepository.InsertAccount(account);
if (account.Id == null)
{
    // ERROR
}

如果这些信息可能有点缺乏,我道歉。我今天才开始学习moq和模拟框架。 AC

2 个答案:

答案 0 :(得分:5)

您可以使用Callback方法来模拟副作用。类似的东西:

accountRepository
    .Expect(r => r.InsertAccount(account))
    .Callback(() => account.ID = 1);

这是未经测试的,但它是沿着正确的路线。

答案 1 :(得分:2)

我不确定这是如何工作的,因为在方法中创建了帐户,所以当我将ID的值设置为1时,它不是我将要引用的实例。

也许我的设计存在缺陷,我应该检查ID&gt; 0在IAccountRepository.InsertAccount实现中,如果没问题则返回bool。虽然那时我在插入一个有相关对象需要插入的帐户时遇到了问题(并且Id已经被处理)。

我发现这是我的问题的答案

accountRepository
    .Expect(p => p.InsertAccount(It.Is<AccountEntity>(x => x.Id == null)))
    .Callback<AccountEntity>(a => a.Id = 1);

感谢。