From 1515fdf191bf71f63a382d4dc790be87809b9a25 Mon Sep 17 00:00:00 2001
From: Suomm <1474983351@qq.com>
Date: Tue, 25 Jul 2023 21:17:01 +0800
Subject: [PATCH] =?UTF-8?q?test:=20=E4=BB=A3=E7=A0=81=E7=94=9F=E6=88=90?=
 =?UTF-8?q?=E5=99=A8=E5=A4=9A=E6=AC=A1=E8=B0=83=E7=94=A8=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../mybatisflex/codegen/test/GeneratorTest.java  | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

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 cf0155a5..15a5e8d8 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
@@ -23,6 +23,7 @@ import com.mybatisflex.codegen.config.TableConfig;
 import com.mybatisflex.codegen.config.TableDefConfig;
 import com.mybatisflex.spring.service.impl.CacheableServiceImpl;
 import com.zaxxer.hikari.HikariDataSource;
+import org.junit.Test;
 
 import java.util.function.UnaryOperator;
 
@@ -149,7 +150,7 @@ public class GeneratorTest {
         generator.generate();
     }
 
-    //    @Test
+    @Test
     public void testCodeGen3() {
         //配置数据源
         HikariDataSource dataSource = new HikariDataSource();
@@ -157,6 +158,13 @@ public class GeneratorTest {
         dataSource.setUsername("root");
         dataSource.setPassword("12345678");
 
+        //通过 datasource 和 globalConfig 创建代码生成器
+        new Generator(dataSource, globalConfig()).generate();
+        new Generator(dataSource, globalConfig()).generate();
+        new Generator(dataSource, globalConfig()).generate();
+    }
+
+    private GlobalConfig globalConfig() {
         GlobalConfig globalConfig = new GlobalConfig();
 
         //用户信息表，用于存放用户信息。 -> 用户信息
@@ -215,11 +223,7 @@ public class GeneratorTest {
             .setOverwriteEnable(true)
             .setMapperAnnotation(true);
 
-        //通过 datasource 和 globalConfig 创建代码生成器
-        Generator generator = new Generator(dataSource, globalConfig);
-
-        //开始生成代码
-        generator.generate();
+        return globalConfig;
     }
 
 }