如何对包含数据库事务的代码进行单元测试

Question

如何在以下代码周围放置单位：

public DbContextTransaction QTTransactionBegin()
    {
        return Database.BeginTransaction();
    }

public int CreateCampaign(CreateCampaignModel createCampaignModel)
    {
        using (var transaction = _qtContext.QTTransactionBegin())
        {
            try
            {
                var campaign = new Campaign();
                campaign.CampaignCode = createCampaignModel.CampaignCode;
                campaign.CampaignDescription = createCampaignModel.CampaignDescription;
                campaign.CampaignDate = createCampaignModel.CampaignDate;
                campaign.CampaignNotes = createCampaignModel.CampaignNotes;
                campaign.OwnerUserID = createCampaignModel.OwnerUserID;
                campaign.AddedOn = DateTime.Now;
                campaign.AddedBy = createCampaignModel.OwnerUserName;
                campaign.UpdatedOn = DateTime.Now;
                campaign.UpdatedBy = createCampaignModel.OwnerUserName;
                campaign.CampaignSegments = GetCampaignSegmentList(createCampaignModel);

                var campaignId = AddOrUpdateCampaign(campaign);
                transaction.Commit();
                return campaignId;
            }
            catch (Exception ex)
            {
                transaction.Rollback();
            }
        }
        return 0;
    }

有人可以建议我如何对上述代码进行单元测试吗？

我尝试了如下代码：

Database_database;

    [TestInitialize]
    public void SetUp()
    {
        _qtDALMock = _factory.CreateMock<IQTDAL>();
        _campaignRepository = new CampaignRepository(_qtDALMock.MockObject);
    }
 [TestMethod]
    public void Check_CreateCampaign_Test()
    {
        // arrange 
        const int expectedCampaignId = 1;
        var createCampaign = QueryToolDummies.CreateCampaignModel;

        _database.BeginTransaction();
        _qtDALMock.Expects.One.MethodWith(x => x.QTTransactionBegin())
            .WillReturn(_database.BeginTransaction());
        _qtDALMock.Expects.One.Method(x => x.AddOrUpdateCampaign(null))
            .With(Is.TypeOf<Campaign>())
            .WillReturn(expectedCampaignId);

        // act
        var result = _campaignRepository.CreateCampaign(createCampaign);

        // assert
        Assert.IsNotNull(result);
    }

此_database.BeginTransaction()有问题。错误说不能像这样使用。

请咨询。

Answer 1

一个问题？为什么要尝试将事务开始进行单元测试？

如果您使用Moq framework模拟存储库并返回从仓库中返回的内容，这将很容易。

实际上，我认为在单元测试中启动BeginTransaction()并不是一个好习惯。

我希望这会有所帮助

Answer 2

我也遇到过同样的问题，解决这个问题非常棘手。

我曾尝试为公开BeginTransaction()方法的上下文创建一个包装器类，但最终您最终需要在测试时模拟DbContextTransaction返回的BeginTransaction()，但是DbContextTransaction没有接口或公共构造函数。

最后，我编写了一个事务管理器类，该类创建和管理自己的事务，并公开用于开始，提交和回滚事务的方法。然后可以伪造该管理器类以及返回它的服务，从而使使用事务的代码得到充分测试。

我已经将其全部写成in this answer。

Answer 3

您正在尝试对多个单元进行单元测试。

假设上面的代码是您的“数据层” /存储库，那么您正在执行集成测试，因为该测试涉及多个单元。

您可以在测试类中包括数据库的设置/拆卸，并使用有效和无效的数据调用SUT（被测对象）以验证预期的行为。