<#include "../function.ftl"> <#include "../variable.ftl"> <#assign pkVar=convertUnderLine(pk) > <#if isBaseModule = 'true'> package com.${cAlias}.${cPlatform}.${sys}.persistence.dao; <#else> package com.${cAlias}.${cPlatform}.${sys}.${module}.persistence.dao; import javax.annotation.Resource; import org.junit.Assert; import org.junit.Test; import org.springframework.test.annotation.Rollback; <#if isBaseModule = 'true'> import com.${cAlias}.${cPlatform}.${sys}.persistence.dao.${class}Dao; import com.${cAlias}.${cPlatform}.${sys}.persistence.dao.${class}QueryDao; import com.${cAlias}.${cPlatform}.${sys}.persistence.entity.${po}; <#else> import com.${cAlias}.${cPlatform}.${sys}.${module}.persistence.dao.${class}Dao; import com.${cAlias}.${cPlatform}.${sys}.${module}.persistence.dao.${class}QueryDao; import com.${cAlias}.${cPlatform}.${sys}.${module}.persistence.entity.${po}; import com.${cAlias}.${cPlatform}.${sys}.${baseClass}BaseTest; /** * ${model.tabComment} dao单元测试类 * *
 
 <#if vars.company?exists>
 * 开发公司：${vars.company}
 
 <#if vars.developer?exists>
 * 开发人员：${vars.developer}
 
 <#if vars.email?exists>
 * 邮箱地址：${vars.email}
 
 * 创建时间：${date?string("yyyy-MM-dd HH:mm:ss")}
 *
*/ public class ${class}DaoTest extends ${baseClass}BaseTest{ @Resource private ${class}Dao ${classVar}Dao; @Resource private ${class}QueryDao ${classVar}QueryDao; @Test @Rollback(true) public void testCrud(){ ${po} ${poVar}=new ${po}(); <#list model.columnList as col> <#assign columnName=convertUnderLine(col.columnName)> <#if col.isPK> ${poVar}.setId(idGenerator.getId()); <#else> <#if col.isNotNull> <#if col.colType="java.util.Date"> ${poVar}.set${columnName?cap_first}(new Date()); <#elseif col.colType="Float"> Integer randId=new Double(100000*Math.random()).intValue(); ${poVar}.set${columnName?cap_first}(Float.parseFloat(randId+"")); <#elseif col.colType="Short"> ${poVar}.set${columnName?cap_first}(new Short("1")); <#elseif col.colType="Integer"> Integer randId=new Double(100000*Math.random()).intValue(); ${poVar}.set${columnName?cap_first}(randId); <#elseif col.colType="Long"> Integer randId=new Double(100000*Math.random()).intValue(); ${poVar}.set${columnName?cap_first}(Long.parseLong(randId+"")); <#elseif col.colType="String"> Integer randId=new Double(100000*Math.random()).intValue(); ${poVar}.set${columnName?cap_first}("${poVar}" + randId); //创建一实体 ${classVar}Dao.create(${poVar}); Assert.assertNotNull(${poVar}.getId()); logger.debug("${poVar}1:"+ ${poVar}.getId()); //获取一实体 ${po} ${poVar}2=${classVar}QueryDao.get(${poVar}.getId()); Assert.assertNotNull(${poVar}2); logger.debug("${poVar}2:" + ${poVar}2.toString()); <#list model.columnList as col> <#assign columnName=convertUnderLine(col.columnName)> <#if !col.isPK> <#if col.isNotNull> <#if col.colType="java.util.Date"> ${poVar}2.set${columnName?cap_first}(new Date()); <#elseif col.colType="Float"> Integer randId2=new Double(100000*Math.random()).intValue(); ${poVar}2.set${columnName?cap_first}(Float.parseFloat(randId2+"")); <#elseif col.colType="Short"> ${poVar}2.set${columnName?cap_first}(new Short("1")); <#elseif col.colType="Integer"> Integer randId2=new Double(100000*Math.random()).intValue(); ${poVar}2.set${columnName?cap_first}(randId2); <#elseif col.colType="Long"> Integer randId2=new Double(100000*Math.random()).intValue(); ${poVar}2.set${columnName?cap_first}(Long.parseLong(randId2+"")); <#elseif col.colType="String"> Integer randId2=new Double(100000*Math.random()).intValue(); ${poVar}2.set${columnName?cap_first}("${poVar}" + randId2); //更新一实体 ${classVar}Dao.update(${poVar}2); ${po} ${poVar}3=${classVar}QueryDao.get(${poVar}.getId()); Assert.assertNotNull(${poVar}3); logger.debug("${poVar}3:"+${poVar}3.toString()); //删除一实体 //${classVar}Dao.remove(${poVar}.getId()); } }