diff --git a/modules/core/src/test/java/com/jeesite/test/GenTableToolTest.java b/modules/core/src/test/java/com/jeesite/test/GenTableToolTest.java new file mode 100644 index 00000000..14bf1c16 --- /dev/null +++ b/modules/core/src/test/java/com/jeesite/test/GenTableToolTest.java @@ -0,0 +1,69 @@ +/** + * Copyright (c) 2013-Now http://jeesite.com All rights reserved. + */ +package com.jeesite.test; + +import org.junit.Test; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.boot.test.context.SpringBootTest; +import org.springframework.test.annotation.Rollback; +import org.springframework.test.context.ActiveProfiles; + +import com.jeesite.common.config.Global; +import com.jeesite.common.lang.StringUtils; +import com.jeesite.common.tests.BaseSpringContextTests; +import com.jeesite.modules.gen.entity.GenTable; +import com.jeesite.modules.gen.entity.GenTableColumn; +import com.jeesite.modules.gen.service.GenTableService; + +/** + * 代码生成工具(API) + * @author ThinkGem + * @version 2020-2-1 + */ +@ActiveProfiles("test") +@SpringBootTest(classes=ApplicationTest.class) +@Rollback(false) +public class GenTableToolTest extends BaseSpringContextTests { + + @Autowired + private GenTableService genTableService; + + @Test + public void execute() throws Exception{ + GenTable genTable = new GenTable(); + genTable.setIsNewRecord(true); + genTable.setTableName("test_data_demo"); // 表名 + genTable = genTableService.getFromDb(genTable); + genTable.setClassName(StringUtils.capCamelCase(genTable.getTableName())); // 实体类名 + genTable.setFunctionAuthor("ThinkGem"); // 作者名称 + genTable.setTplCategory("crud"); // 生成模板:crud、treeGrid、service、dao、query + genTable.setPackageName("com.jeesite.modules");// 生成包路径 + genTable.setModuleName("test"); // 生成模块名 + genTable.setSubModuleName(""); // 生成子模块名 + genTable.setFunctionName(genTable.getComments()); // 生成功能名 + genTable.setFunctionNameSimple(genTable.getComments()); // 功能名(简称) + genTable.getOptionMap().put("isHaveDisableEnable", Global.NO); // 是否有停用启用 + genTable.getOptionMap().put("isHaveDelete", Global.YES); // 是否有删除功能 + genTable.getOptionMap().put("isFileUpload", Global.NO); // 是否有上传附件 + genTable.getOptionMap().put("isImageUpload", Global.NO); // 是否有上传图片 + for(GenTableColumn column : genTable.getColumnList()){ + // 字段配置(下拉框) + if ("trade_type".equals(column.getColumnName())){ + column.setShowType("select"); + column.getOptionMap().put("dictType", "sys_menu_type"); + } + } + // 删除旧配置 + genTableService.delete(genTable); + // 保存新配置 + genTable.setIsNewRecord(true); + genTableService.save(genTable); + // 编译并生成代码 + genTable.setGenFlag("1"); // 1:编译输出;2:生成文件 + genTable.setReplaceFile(Global.YES); // 如果存在则替换 + String result = genTableService.generateCode(genTable); + System.out.println(result); + } + +}