IbatisTableConfigTest.java

package cn.org.rapid_framework.generator.ext.tableconfig.model;

import cn.org.rapid_framework.generator.GeneratorProperties;
import cn.org.rapid_framework.generator.GeneratorTestCase;
import cn.org.rapid_framework.generator.ext.tableconfig.builder.TableConfigXmlBuilder;
import cn.org.rapid_framework.generator.provider.db.sql.model.Sql;
import cn.org.rapid_framework.generator.util.FileHelper;
import cn.org.rapid_framework.generator.util.IOHelper;
import cn.org.rapid_framework.generator.util.test.GeneratorTestHelper;

public class IbatisTableConfigTest extends GeneratorTestCase {
	
	static TableConfigSet tableConfigSet = new TableConfigXmlBuilder().parseFromXML("com.global.pkg", FileHelper.getFile("classpath:cn/org/rapid_framework/generator/ext/tableconfig/"), "user_info.xml");
	
	public void test_gen_operation() throws Exception {
		 g.addTemplateRootDir("classpath:generator/template/rapid/operation/dal");
		 g.addTemplateRootDir("classpath:generator/template/rapid/share/dal");
		 
		 String result = ""; 
		 TableConfig tableConfig = tableConfigSet.getBySqlName("user_info");
		 for(Sql sql : tableConfig.getSqls()) {
			 result = result + "\n" + GeneratorTestHelper.generateBy(g, Helper.newMapFromSql(sql, tableConfig));
		 }
		 System.out.println(result);
		 
		 assertContains(result, "package com.company.project.user_info.operation.userinfo;");
		 assertContains(result, IOHelper.readFile(FileHelper.getFile("classpath:IbatisTableConfigTest/expectedParameterAndResult.txt"),"UTF-8"));
		 
	}

	public void test_gen_TableConfig() throws Exception {
		 
		 g.addTemplateRootDir("classpath:generator/template/rapid/table_config/dal");
		 g.addTemplateRootDir("classpath:generator/template/rapid/share/dal");
		 
		 String str = GeneratorTestHelper.generateBy(g, Helper.newMapFromTableConfig(tableConfigSet.getBySqlName("user_info")));
		 System.out.println(str);
		 
		 assertContains(str,"public java.lang.Long insert(UserInfoDO userInfo) throws DataAccessException;");
		 assertContains(str,"public java.lang.Long insertWithFunction(UserInfoDO userInfo) throws DataAccessException;");
		 assertContains(str,"根据订单号查询订单");
		 assertContains(str,"public CountUsernameResult countUsername(Long userId ,com.iwallet.biz.common.util.money.Money age ,com.iwallet.biz.common.util.money.Money sex ,String maxUsername ,String minUsername) throws DataAccessException;");
	
		 assertContains(str,"用户ID 		db_column: USER_ID ");
		 assertContains(str,"private Money sex = new Money(0,0);");
		 assertContains(str,IOHelper.readFile(FileHelper.getFile("classpath:IbatisTableConfigTest/expectedIbatisDaoImpl.txt"),"UTF-8"));
		 assertContains(str,IOHelper.readFile(FileHelper.getFile("classpath:IbatisTableConfigTest/expectedUserInfoDao.txt"),"UTF-8"));
		 assertContains(str,IOHelper.readFile(FileHelper.getFile("classpath:IbatisTableConfigTest/expectedUserInfoDO.txt"),"UTF-8"));
		 assertContains(str,IOHelper.readFile(FileHelper.getFile("classpath:IbatisTableConfigTest/expectedUserInfoSqlMap.txt"),"UTF-8"));
		 
		 assertContains(str,"src/main/java/com/company/project/user_info/dataobject/UserInfoDO.java");
		 assertContains(str,"src/main/java/com/company/project/user_info/ibatis/IbatisUserInfoDAO.java");
		 assertContains(str,"src/main/resources/sqlmap/badqiu_test_module/UserInfo-sqlmap-mapping.xml");
		 assertContains(str,"src/main/java/com/company/project/user_info/daointerface/UserInfoDAO.java");
		 
	}
	
	
    public void test_gen_TableConfig_for_return_primitive_type() throws Exception {
         tableConfigSet = new TableConfigXmlBuilder().parseFromXML("com.global.pkg", FileHelper.getFile("classpath:cn/org/rapid_framework/generator/ext/tableconfig/"), "user_info.xml");
         g.addTemplateRootDir("classpath:generator/template/rapid/table_config/dal");
         g.addTemplateRootDir("classpath:generator/template/rapid/share/dal");
         
         GeneratorProperties.setProperty("java_typemapping.java.lang.Long", "long");
         String str = GeneratorTestHelper.generateBy(g, Helper.newMapFromTableConfig(tableConfigSet.getBySqlName("user_info")));
         System.out.println(str);
         assertContains(str,"   public long countByUserId(Long userId) throws DataAccessException {Number returnObject = (Number)getSqlMapClientTemplate().queryForObject(\"badqiu_test_app.UserInfo.countByUserId\",userId);if(returnObject == null) return (long)0; elsereturn returnObject.longValue();}");
         
    }
}