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 3bd55f87..4808532c 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.codegen.constant.TemplateConst;
 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();
@@ -189,7 +190,7 @@ public class GeneratorTest {
 
         TableConfig tableConfig = new TableConfig();
         tableConfig.setTableName("sys_user");
-        tableConfig.addColumnConfig(columnConfig);
+        tableConfig.setColumnConfig(columnConfig);
 
         ColumnConfig logicDelete = new ColumnConfig();
         logicDelete.setColumnName("del_flag");
@@ -225,7 +226,7 @@ public class GeneratorTest {
         return globalConfig;
     }
 
-//    @Test
+    //    @Test
     public void testCodeGen4() {
         // 配置数据源
         HikariDataSource dataSource = new HikariDataSource();
@@ -276,4 +277,72 @@ public class GeneratorTest {
         generator.generate();
     }
 
+    @Test
+    public void testCodeGen5() {
+        // 配置数据源
+        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();
+
+        // 用户信息表，用于存放用户信息。 -> 用户信息
+        UnaryOperator<String> tableFormat = (e) -> e.split("，")[0].replace("表", "");
+
+        // 设置注解生成配置
+        globalConfig.getJavadocConfig()
+            .setAuthor("王帅")
+            .setTableCommentFormat(tableFormat);
+
+        // 设置生成文件目录和根包
+        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", "sys_role");
+
+        // 全局表配置
+        TableConfig tableConfig = TableConfig.builder()
+            .camelToUnderline(false)
+            .mapperGenerateEnable(false)
+            .build();
+
+        // sys_user 字段单独配置
+        ColumnConfig userColumnConfig = ColumnConfig.create()
+            .setColumnName("update_time")
+            .setOnUpdateValue("NOW()");
+
+        // sys_user 表单独配置
+        TableConfig userTableConfig = TableConfig.create()
+            .setTableName("sys_user")
+            .setColumnConfig(userColumnConfig);
+
+        // 全局字段配置
+        ColumnConfig columnConfig = ColumnConfig.builder()
+            .columnName("create_time")
+            .onInsertValue("NOW()")
+            .build();
+
+        globalConfig.getStrategyConfig()
+            .setTableConfig(tableConfig)
+            .setTableConfig(userTableConfig)
+            .setColumnConfig(columnConfig);
+
+        // 配置生成 entity
+        globalConfig.enableEntity()
+            .setOverwriteEnable(true)
+            .setWithLombok(true);
+
+        //通过 datasource 和 globalConfig 创建代码生成器
+        Generator generator = new Generator(dataSource, globalConfig);
+
+        //开始生成代码
+        generator.generate();
+    }
+
 }