admin/mybatis-flex
1
0
Fork 0
mirror of https://gitee.com/mybatis-flex/mybatis-flex.git synced 2025-12-07 17:18:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: 更新测试代码，支持新生成器格式。

Browse Source
This commit is contained in:
Suomm 2023-05-15 20:04:55 +08:00
parent 19069a129a
commit 9165d8d44a
2 changed files with 58 additions and 57 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

91
mybatis-flex-codegen/src/test/java/com/mybatisflex/codegen/test/GeneratorTest.java
View File

@ -38,40 +38,36 @@ public class GeneratorTest {
// JdbcTypeMapping.registerMapping(Integer.class, Long.class);
GlobalConfig globalConfig = new GlobalConfig();
globalConfig.setSourceDir(System.getProperty("user.dir") + "/src/test/java");
// globalConfig.setTablePrefix("tb_");
// globalConfig.setEntityWithLombok(true);
globalConfig.setEntitySupperClass(BaseEntity.class);
//设置生成文件目录和根包
globalConfig.getPackageConfig()
.setSourceDir(System.getProperty("user.dir") + "/src/test/java")
.setBasePackage("com.test");
//设置只生成哪些表
globalConfig.addGenerateTable("account", "account_session");
globalConfig.getStrategyConfig()
.addGenerateTable("account", "account_session");
//设置 entity 的包名
globalConfig.setEntityPackage("com.test.entity");
globalConfig.setEntityClassPrefix("My");
globalConfig.setEntityClassSuffix("Entity");
//设置生成 entity
globalConfig.enableEntity()
.setWithLombok(true)
.setClassPrefix("My")
.setClassSuffix("Entity")
.setSupperClass(BaseEntity.class);
//设置 entity 的包名
globalConfig.setTableDefGenerateEnable(true);
globalConfig.setTableDefPackage("com.test.entity.tables");
globalConfig.setTableDefClassPrefix("My");
globalConfig.setTableDefClassSuffix("TableDef");
//是否生成 mapper 类，默认为 false
globalConfig.setMapperGenerateEnable(true);
globalConfig.setMapperClassPrefix("Flex");
globalConfig.setMapperClassSuffix("Dao");
//设置 mapper 类的包名
globalConfig.setMapperPackage("com.test.mapper");
globalConfig.setMapperSupperClass(MyBaseMapper.class);
//设置生成 tableDef
globalConfig.enableTableDef();
//设置生成 mapper
globalConfig.enableMapper()
.setClassPrefix("Flex")
.setClassSuffix("Dao")
.setSupperClass(MyBaseMapper.class);
TableConfig tableConfig = new TableConfig();
tableConfig.setTableName("account");
tableConfig.setUpdateListenerClass(MyUpdateListener.class);
globalConfig.addTableConfig(tableConfig);
globalConfig.getStrategyConfig().addTableConfig(tableConfig);
//可以单独配置某个列
@ -79,7 +75,7 @@ public class GeneratorTest {
columnConfig.setColumnName("tenant_id");
columnConfig.setLarge(true);
columnConfig.setVersion(true);
globalConfig.addColumnConfig("account", columnConfig);
globalConfig.getStrategyConfig().addColumnConfig("account", columnConfig);
//通过 datasource 和 globalConfig 创建代码生成器
@ -91,39 +87,44 @@ public class GeneratorTest {
@Test
public void testCodeGen() {
// 配置数据源
//配置数据源
HikariDataSource dataSource = new HikariDataSource();
dataSource.setJdbcUrl("jdbc:mysql://127.0.0.1:3306/test?characterEncoding=utf-8");
dataSource.setUsername("root");
dataSource.setPassword("12345678");
GlobalConfig globalConfig = new GlobalConfig();
globalConfig.setSourceDir(System.getProperty("user.dir") + "/src/test/java");
globalConfig.setTablePrefix("sys_");
globalConfig.setBasePackage("com.test");
globalConfig.setEntityWithLombok(true);
globalConfig.setEntitySupperClass(BaseEntity.class);
//设置生成文件目录和根包
globalConfig.getPackageConfig()
.setSourceDir(System.getProperty("user.dir") + "/src/test/java")
.setBasePackage("com.test");
// 设置只生成哪些表
globalConfig.addGenerateTable("sys_user");
//设置表前缀和只生成哪些表
globalConfig.getStrategyConfig()
.setTablePrefix("sys_")
.addGenerateTable("sys_user");
// 设置 entity 的包名
globalConfig.setTableDefGenerateEnable(true);
//配置生成 entity
globalConfig.enableEntity()
.setWithLombok(true)
.setSupperClass(BaseEntity.class);
// 是否生成 mapper 类，默认为 false
globalConfig.setMapperGenerateEnable(true);
// 是否生成 service 类，默认为 false
globalConfig.setServiceGenerateEnable(true);
// 是否生成 serviceImpl 类，默认为 false
globalConfig.setServiceImplGenerateEnable(true);
// 是否生成 controller 类，默认为 false
globalConfig.setControllerGenerateEnable(true);
//配置生成 mapper
globalConfig.enableMapper();
//配置生成 service
globalConfig.enableService();
//配置生成 serviceImpl
globalConfig.enableServiceImpl();
//配置生成 controller
globalConfig.enableController();
//配置生成 tableDef
globalConfig.enableTableDef();
// 通过 datasource 和 globalConfig 创建代码生成器
//通过 datasource 和 globalConfig 创建代码生成器
Generator generator = new Generator(dataSource, globalConfig);
// 开始生成代码
//开始生成代码
generator.generate();
}

24
mybatis-flex-codegen/src/test/java/com/mybatisflex/codegen/test/SqliteGeneratorTest.java
View File

@ -4,14 +4,13 @@ import com.mybatisflex.codegen.Generator;
import com.mybatisflex.codegen.config.GlobalConfig;
import com.mybatisflex.codegen.dialect.IDialect;
import com.zaxxer.hikari.HikariDataSource;
import org.junit.Test;
import java.sql.Connection;
import java.sql.Statement;
public class SqliteGeneratorTest {
// @Test
// @Test
public void testGenerator3() {
//配置数据源
@ -24,20 +23,21 @@ public class SqliteGeneratorTest {
GlobalConfig globalConfig = new GlobalConfig();
globalConfig.setSourceDir(System.getProperty("user.dir") + "/src/test/java");
//配置生成文件目录与根包
globalConfig.getPackageConfig()
.setSourceDir(System.getProperty("user.dir") + "/src/test/java")
.setBasePackage("com.test");
//设置只生成哪些表
globalConfig.addGenerateTable("person");
globalConfig.getStrategyConfig()
.addGenerateTable("person");
//设置 entity 的包名
globalConfig.setEntityPackage("com.test.entity");
globalConfig.enableEntity()
.setWithLombok(true);
//是否生成 mapper 类，默认为 false
globalConfig.setMapperGenerateEnable(true);
globalConfig.setEntityWithLombok(true);
//设置 mapper 类的包名
globalConfig.setMapperPackage("com.test.mapper");
//设置生成 mapper 类
globalConfig.enableMapper();
Generator generator = new Generator(dataSource, globalConfig, IDialect.SQLITE);