1

我正在尝试使用 Unitils 来帮助我进行数据库测试。我想使用 Unitils/DBMaintain 功能来禁用约束。然而，这有一些问题。我不希望使用 DBMaintain 为我创建数据库，但我希望使用它的约束禁用功能。我能够通过使用下面列出的自定义模块来实现这一点：

public class DisableConstraintModule implements Module {

     private boolean disableConstraints = false;

     public void afterInit() {
         if (disableConstraints) {
             DatabaseUnitils.disableConstraints();
         }
     }

     public void init(Properties configuration) {
         disableConstraints = PropertyUtils.getBoolean("Database.disableConstraints", false, configuration);
     }      
}

这部分解决了我想要的，但是我希望能够只禁用我将在测试中使用的表的约束。我的测试将针对具有多个模式的数据库运行，每个模式都有数百个不同的表。DatabaseUnitils.disableConstraints() 禁用每个模式中每个表的约束，这太耗时且没有必要。

在搜索 dbmaintain 代码后，我发现 Db2Database 类确实包含一个用于禁用特定模式和表名基础上的约束的函数，但是此方法受到保护。我可以通过扩展 Db2Database 类或使用反射来访问它。

接下来，我需要能够确定我感兴趣的模式和表。我可以通过观察 @DataSet 注释来根据 xml 中的内容确定哪些模式和表是重要的。为此，我需要覆盖 TestListener，以便在它尝试插入数据集之前指示它使用 xml 禁用约束。这是我的尝试：

public class DisableConstraintModule extends DbUnitModule {

    private boolean disableConstraints = false;

    private TableBasedConstraintsDisabler disabler;

    public void afterInit() {
    }

    public void init(Properties configuration) {
        disableConstraints = PropertyUtils.getBoolean("Database.disableConstraints", false, configuration);

        PropertyUtils.getInstance("org.unitils.dbmaintainer.structure.ConstraintsDisabler.implClassName", configuration);
    }

    public void disableConstraintsForDataSet(MultiSchemaDataSet dataSet) {
        disabler.disableConstraints(dataSet);
    }


    protected class DbUnitCustomListener extends DbUnitModule.DbUnitListener {

        @Override
        public void beforeTestSetUp(Object testObject, Method testMethod) {
            disableConstraintsForDataSet(getDataSet(testMethod, testObject));
            insertDataSet(testMethod, testObject);
        }
    }

 }

这是我想做的，但是我无法获得@DataSet 注释来触发我的 DbUnitCustomListener，而是调用默认的 DBUnitModule DbUnitListener。无论如何，我是否可以覆盖在使用 @DataSet 注释时调用哪个侦听器，或者是否有更好的方法一起禁用 DB2 数据库的特定模式和表级别的约束？

谢谢

4

1 回答 1

1

您必须告诉 Unitils 使用您的 DbUnitModule 子类。您可以使用 unitils.properties 文件中的 unitils.module.dbunit.className 属性来执行此操作。听起来你已经弄清楚了这部分。

第二部分是重写 DbUnitModule 的 getTestListener() 以返回您的自定义侦听器。

有关示例，请参见这篇文章。

于 2012-07-27T04:20:09.050 回答