RegionDbDAOTestCase;
import xxx.constant.group.Nic;
import xxx.constant.group.Policy;
import xxx.dao.regiondb.XXXDAO;
import xxx.model.db.XXXDO;
public class XXXDAOTest extends BaseRegionDbDAOTestCase {
@SpringBeanByName
private XXXDAO YYYDAO;
@Test
@DbFit(when="XXXDAOTest.initBlank.when.wiki", then="XXXDAOTest.queryOneRecord.then.wiki")
public void testInsertXXXDO() {
XXXDO YYY = new XXXDO();
$setFields
YYY.setGmtCreate(new Date());
YYY.setGmtModify(new Date());
YYYDAO.insertXXXDO(YYY);
}
@Test
@DbFit(when="XXXDAOTest.initRecords.when.wiki")
public void testCountXXXDOByExample() {
XXXDO YYY = new XXXDO();
Assert.assertTrue(YYYDAO.countXXXDOByExample(YYY) == );
}
@Test
@DbFit(when="XXXDAOTest.initRecords.when.wiki", then="XXXDAOTest.testUpdate.then.wiki")
public void testUpdateXXXDO() {
XXXDO YYY = YYYDAO.findXXXDOByPrimaryKey();
$setFields
YYYDAO.updateXXXDO(YYY);
}
@Test
@DbFit(when="XXXDAOTest.initRecords.when.wiki")
public void testFindListByExample() {
XXXDO YYY = new XXXDO();
$setFields
List list = YYYDAO.findListByExample(YYY);
Assert.assertEquals(list.size(), );
for (XXXDO YYYDO: list) {
$AssertGetValues
}
}
@Test
@DbFit(when="XXXDAOTest.initRecords.when.wiki")
public void testFindXXXDOByPrimaryKey() {
XXXDO YYY = YYYDAO.findXXXDOByPrimaryKey();
$AssertGetValues
}
@Test
@DbFit(when="XXXDAOTest.initRecords.when.wiki", then="")
public void testDeleteXXXDOByPrimaryKey() {
Integer count = YYYDAO.deleteXXXDOByPrimaryKey();
Assert.assertEquals(count.intValue(), );
Integer nodelete = YYYDAO.deleteXXXDOByPrimaryKey();
Assert.assertEquals(nodelete.intValue(), );
}
}
运行： $ python create_daotest_wiki.py
生成以下文件：
其中：
VmDAOTest.initBlank.when.wiki
|connect|
|clean table|vm|
VmDAOTest.initRecords.when.wiki
|connect|
|clean table|vm|
|insert|vm|
|id|gmt_create|gmt_modify|vm_name|cores|mem|disk|status|nc_id|is_deleted|
VmDAOTest.queryOneRecord.when.wiki / VmDAOTest.testUpdate.when.wiki
|connect|
|query|select id, vm_name, cores, mem, disk, status, nc_id, is_deleted from vm|
|id|vm_name|cores|mem|disk|status|nc_id|is_deleted|
生成的DAOTEST Java 文件:
package com.aliyun.houyi.regionmaster.dao.regiondb.impl;
import java.util.Date;
import java.util.List;
import org.jtester.unitils.dbfit.DbFit;
import org.testng.Assert;
import org.testng.annotations.Test;
import org.unitils.spring.annotation.SpringBeanByName;
import com.aliyun.houyi.regionmaster.BaseRegionDbDAOTestCase;
import com.aliyun.houyi.regionmaster.constant.group.Nic;
import com.aliyun.houyi.regionmaster.constant.group.Policy;
import com.aliyun.houyi.regionmaster.dao.regiondb.VmDAO;
import com.aliyun.houyi.regionmaster.model.db.VmDO;
public class VmDAOTest extends BaseRegionDbDAOTestCase {
@SpringBeanByName
private VmDAO vmDAO;
@Test
@DbFit(when="VmDAOTest.initBlank.when.wiki", then="VmDAOTest.queryOneRecord.then.wiki")
public void testInsertVmDO() {
VmDO vm = new VmDO();
vm.setId();
vm.setVmName();
vm.setCores();