test: 测试代码生成器。

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

@@ -20,6 +20,7 @@ import com.mybatisflex.codegen.config.ColumnConfig;
 import com.mybatisflex.codegen.config.GlobalConfig;
 import com.mybatisflex.codegen.config.TableConfig;
 import com.zaxxer.hikari.HikariDataSource;
+import org.junit.Test;
 
 public class GeneratorTest {
 
@@ -87,4 +88,43 @@ public class GeneratorTest {
         //开始生成代码
         generator.generate();
     }
+
+    @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.addGenerateTable("sys_user");
+
+        // 设置 entity 的包名
+        globalConfig.setTableDefGenerateEnable(true);
+
+        // 是否生成 mapper 类，默认为 false
+        globalConfig.setMapperGenerateEnable(true);
+        // 是否生成 service 类，默认为 false
+        globalConfig.setServiceGenerateEnable(true);
+        // 是否生成 serviceImpl 类，默认为 false
+        globalConfig.setServiceImplGenerateEnable(true);
+        // 是否生成 controller 类，默认为 false
+        globalConfig.setControllerGenerateEnable(true);
+
+        // 通过 datasource 和 globalConfig 创建代码生成器
+        Generator generator = new Generator(dataSource, globalConfig);
+
+        // 开始生成代码
+        generator.generate();
+    }
+
 }