Unitils是单元测试的一组工具集,介绍见http://www.iteye.com/topic/160004
Dbdeploy是Thoughtworks公司开发的数据库工具,获得Jolt大奖。
Unitils有自己的一个管理类似于dbdeploy的管理数据库的module -- DBMaintainer,其要求的sql文件名称格式如下所示,要求使用下划线分隔
001_create_users_table.sql
Dbdeploy要求的sql文件格式使用空格分隔
1 create users table.sql
如果在项目中同时使用了dbdeploy和unitils(dbdeploy比dbmaintainer更强大,同时支持undo),由于两种格式不匹配,就需要维护两套sql文件,比较麻烦。目前在dbdeploy中没有发现可以配置文件名的方法,只能从unitils入手,查看unitils的文档找到解决方法。
使用unitils时可以自定义ScriptSource(获取sql file source的一个接口),在unitils.properties中加入
org.unitils.dbmaintainer.script.ScriptSource.implClassName=com.andyao.unitils.DbDeployScriptSource
DbDeployScriptSource为
public class DbDeployScriptSource extends FileScriptSource {
public static final String SEPARATOR = " ";
/**
* Indicates if the given file is regarded as a script file
*
* @param file The file
* @param scriptFileSpecification Specification describing the files that can be regarded as a script file
* @return True if the given file is regarded as a script file.
*/
@Override
protected boolean isScriptFile(File file, ScriptFilesSpecification scriptFileSpecification) {
String name = file.getName();
boolean fileExtensionSupported = false;
for (String fileExtension : scriptFileSpecification.getFileExtensions()) {
if (name.endsWith(fileExtension)) {
fileExtensionSupported = true;
break;
}
}
if (!fileExtensionSupported) {
return false;
}
if (scriptFileSpecification.isExcludeFilesWithoutIndex()) {
if (!StringUtils.contains(name, SEPARATOR)) {
return false;
}
String indexNrStr = StringUtils.substringBefore(name, SEPARATOR);
if (!StringUtils.isNumeric(indexNrStr)) {
return false;
}
}
return true;
}
/**
* Returns the version index of the given script file
*
* @param scriptFile The file containing a script
* @return The version of the script file
*/
@Override
protected Long getIndex(File scriptFile) {
if (StringUtils.contains(scriptFile.getName(), SEPARATOR)) {
try {
return new Long(StringUtils.substringBefore(scriptFile.getName(), SEPARATOR));
} catch (NumberFormatException e) {
return -1L;
}
} else {
return -1L;
}
}
}
这样就可以同时使用dbdeploy和unitils,做到自动化维护测试数据库。