reportDaoTest.ftl 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475
  1. <#import "../function.ftl" as func>
  2. <#assign class=model.variables.class>
  3. <#assign classVar=model.variables.classVar>
  4. <#assign po=class + "Po">
  5. <#assign poVar=classVar + "Po">
  6. <#assign sys=model.variables.sys>
  7. <#assign module=model.variables.module>
  8. <#assign pk=func.getPk(model) >
  9. <#assign pkVar=func.convertUnderLine(pk) >
  10. <#assign baseClass=model.variables.baseClass>
  11. <#assign isBaseModule=model.variables.isBaseModule>
  12. <#if isBaseModule = 'true'>
  13. package com.${vars.cAlias}.${sys}.test.persistence.dao;
  14. <#else>
  15. package com.${vars.cAlias}.${sys}.${module}.test.persistence.dao;
  16. </#if>
  17. import java.util.Random;
  18. import java.util.Date;
  19. import java.util.List;
  20. import org.junit.Before;
  21. import org.junit.Test;
  22. import org.springframework.test.context.ContextConfiguration;
  23. import com.${vars.cAlias}.base.core.test.SimpleSpringTestCase;
  24. import com.${vars.cAlias}.base.mongodb.MongoConfigHelper;
  25. <#if isBaseModule = 'true'>
  26. import com.${vars.cAlias}.${sys}.persistence.dao.${class}MGDao;
  27. import com.${vars.cAlias}.${sys}.persistence.entity.${po};
  28. <#else>
  29. import com.${vars.cAlias}.${sys}.${module}.persistence.dao.${class}MGDao;
  30. import com.${vars.cAlias}.${sys}.${module}.persistence.entity.${po};
  31. </#if>
  32. @ContextConfiguration({"classpath:conf/gl-test.xml"})
  33. public class ${class}MGDaoTest extends SimpleSpringTestCase{
  34. @Before
  35. public void init(){
  36. MongoConfigHelper.connect();
  37. }
  38. @Test
  39. public void createAndGet() {
  40. ${class}MGDao ${classVar}Dao = ${class}MGDao.newInstance();
  41. ${po} ${poVar}=new ${po}();
  42. Integer randId=new Double(100000*Math.random()).intValue();
  43. <#list model.columnList as col>
  44. <#assign columnName=func.convertUnderLine(col.columnName)>
  45. <#if col.isPK>
  46. <#else>
  47. <#if col.isNotNull>
  48. <#if col.colType="java.util.Date">
  49. ${poVar}.set${columnName?cap_first}(new Date());
  50. <#elseif col.colType="Float">
  51. ${poVar}.set${columnName?cap_first}(Float.parseFloat(randId+""));
  52. <#elseif col.colType="Short">
  53. ${poVar}.set${columnName?cap_first}(new Short("1"));
  54. <#elseif col.colType="Integer">
  55. ${poVar}.set${columnName?cap_first}(randId);
  56. <#elseif col.colType="Long">
  57. ${poVar}.set${columnName?cap_first}(Long.parseLong(randId+""));
  58. <#elseif col.colType="String">
  59. ${poVar}.set${columnName?cap_first}("${poVar}" + randId);
  60. </#if>
  61. </#if>
  62. </#if>
  63. </#list>
  64. ${classVar}MGDao.insert(${poVar});
  65. }
  66. }