Jtester+unitils+testng:DAO单元测试文件模板自动生成(一)

2014-11-24 03:31:13 · 作者: · 浏览: 20

在使用 jtester+unitils+testng 做数据库接口的单元测试框架中, 常常需要编写一些 wiki 及 DAOTest java 文件, 比如:

public class XXXDefaultDAOTest extends BaseRegionDbDAOTestCase {

	@SpringBeanByName
	private XXXDefaultDAO XXXDefaultDAO;
	
	@Test
	@DbFit(when="XXXDefaultDAOTest.initBlank.when.wiki", then="XXXDefaultDAOTest.queryOneRecord.then.wiki")
	public void testInsertXXXDefaultDO() {
		XXXDefaultDO XXXDefaultDO = new XXXDefaultDO();
		XXXDefaultDO.setId(1L);
		XXXDefaultDO.setCidrBlock("192.168.10.10");
		XXXDefaultDO.setIpProtocol("tcp");
		XXXDefaultDO.setPortRange("3000:4000");
		XXXDefaultDO.setPolicy(Policy.POLICY_ACCEPT);
		XXXDefaultDO.setNic(Nic.INTRANET);
		XXXDefaultDO.setPriority(65533L);
		XXXDefaultDO.setType(1L);
		XXXDefaultDO.setIsDeleted(0L);
		XXXDefaultDO.setDescription("test1");
		XXXDefaultDO.setGmtCreate(new Date());
		XXXDefaultDO.setGmtModify(new Date());
		XXXDefaultDAO.insertXXXDefaultDO(XXXDefaultDO);
	}

	@Test
	@DbFit(when="XXXDefaultDAOTest.initRecords.when.wiki")
	public void testCountXXXDefaultDOByExample() {
		XXXDefaultDO XXXDefaultDO = new XXXDefaultDO();
		Assert.assertTrue(XXXDefaultDAO.countXXXDefaultDOByExample(XXXDefaultDO) == 1);
	}

	@Test
	@DbFit(when="XXXDefaultDAOTest.initRecords.when.wiki", then="XXXDefaultDAOTest.testUpdate.then.wiki")
	public void testUpdateXXXDefaultDO() {
		XXXDefaultDO found = XXXDefaultDAO.findXXXDefaultDOByPrimaryKey(6L);
		found.setIpProtocol("udp");
		found.setNic(Nic.INTERNET);
		found.setDescription("desc");
		XXXDefaultDAO.updateXXXDefaultDO(found);
	}

	@Test
	@DbFit(when="XXXDefaultDAOTest.initRecords.when.wiki")
	public void testFindListByExample() {
		String cidrBlock = "10.152.126.83";
		Policy policy = Policy.POLICY_ACCEPT;
		XXXDefaultDO XXXDefault = new XXXDefaultDO();
		XXXDefault.setCidrBlock(cidrBlock);
		XXXDefault.setPolicy(policy);
		List
list = XXXDefaultDAO.findListByExample(XXXDefault); Assert.assertEquals(list.size(), 1); for (XXXDefaultDO XXXDefaultDO: list) { Assert.assertEquals(XXXDefaultDO.getCidrBlock(), cidrBlock); Assert.assertEquals(XXXDefaultDO.getPolicy(), policy); } } @Test @DbFit(when="XXXDefaultDAOTest.initRecords.when.wiki") public void testFindXXXDefaultDOByPrimaryKey() { XXXDefaultDO found = XXXDefaultDAO.findXXXDefaultDOByPrimaryKey(6L); Assert.assertEquals(found.getCidrBlock(), "10.152.126.83"); Assert.assertEquals(found.getIpProtocol(), "all"); Assert.assertEquals(found.getPortRange(), ""); Assert.assertEquals(found.getPolicy(), Policy.POLICY_ACCEPT); Assert.assertEquals(found.getNic(), Nic.BOTH); Assert.assertEquals(found.getPriority().longValue(),1L); Assert.assertEquals(found.getType().intValue(), 1); Assert.assertEquals(found.getIsDeleted().intValue(), 0); Assert.assertEquals(found.getDescription(), "bie dong"); } @Test @DbFit(when="XXXDefaultDAOTest.initRecords.when.wiki", then="") public void testDeleteXXXDefaultDOByPrimaryKey() { Integer count = XXXDefaultDAO.deleteXXXDefaultDOByPrimaryKey(6L); Assert.assertEquals(count.i