From 82e189f405e44f85a0718ed4a8a41b35f4aaf54a Mon Sep 17 00:00:00 2001
From: Suomm <1474983351@qq.com>
Date: Tue, 4 Jun 2024 11:13:53 +0800
Subject: [PATCH] =?UTF-8?q?test:=20=E6=B5=8B=E8=AF=95=20Kotlin=20=E4=BB=A3?=
 =?UTF-8?q?=E7=A0=81=E7=94=9F=E6=88=90=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../codegen/test/GeneratorTest.java           | 35 ++++++++++++++++++-
 1 file changed, 34 insertions(+), 1 deletion(-)

diff --git a/mybatis-flex-codegen/src/test/java/com/mybatisflex/codegen/test/GeneratorTest.java b/mybatis-flex-codegen/src/test/java/com/mybatisflex/codegen/test/GeneratorTest.java
index e5a853b1..8e621190 100644
--- a/mybatis-flex-codegen/src/test/java/com/mybatisflex/codegen/test/GeneratorTest.java
+++ b/mybatis-flex-codegen/src/test/java/com/mybatisflex/codegen/test/GeneratorTest.java
@@ -279,7 +279,7 @@ public class GeneratorTest {
         generator.generate();
     }
 
-//    @Test
+    // @Test
     public void testCodeGen5() {
         // 配置数据源
         HikariDataSource dataSource = new HikariDataSource();
@@ -353,4 +353,37 @@ public class GeneratorTest {
         generator.generate();
     }
 
+    @Test
+    public void testKotlinCodegen() {
+        // 配置数据源
+        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.FileType.KOTLIN);
+
+        // 设置生成文件目录和根包
+        globalConfig.getPackageConfig()
+            .setSourceDir(System.getProperty("user.dir") + "/src/test/java")
+            .setMapperXmlPath(System.getProperty("user.dir") + "/src/test/resources/mapper")
+            .setBasePackage("com.test");
+
+        // 设置表前缀和只生成哪些表
+        globalConfig.getStrategyConfig()
+            .setTablePrefix("sys_")
+            .setGenerateTable("sys_user");
+
+        globalConfig.enableEntity();
+        globalConfig.enableService();
+        globalConfig.enableServiceImpl();
+        globalConfig.enableMapper();
+        globalConfig.enableMapperXml();
+        globalConfig.enableController();
+        globalConfig.enableTableDef();
+        globalConfig.enablePackageInfo();
+
+        new Generator(dataSource, globalConfig).generate();
+    }
+
 }