更改Unitils DbUnitModule的TestListener

Question

我正在尝试使用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，以便在它尝试插入dataset之前使用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数据库的特定模式和表级别上的约束？

谢谢

Answer 1

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

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

有关示例，请参阅此post。