test: 测试生成内置数据脱敏处理器键(Masks)。

This commit is contained in:
Suomm 2024-03-29 12:22:18 +08:00
parent b0a3b358d7
commit e547e90dff

View File

@ -22,6 +22,7 @@ import com.mybatisflex.codegen.config.GlobalConfig;
import com.mybatisflex.codegen.config.TableConfig; import com.mybatisflex.codegen.config.TableConfig;
import com.mybatisflex.codegen.config.TableDefConfig; import com.mybatisflex.codegen.config.TableDefConfig;
import com.mybatisflex.codegen.constant.TemplateConst; import com.mybatisflex.codegen.constant.TemplateConst;
import com.mybatisflex.core.mask.Masks;
import com.zaxxer.hikari.HikariDataSource; import com.zaxxer.hikari.HikariDataSource;
import org.junit.Test; import org.junit.Test;
@ -317,9 +318,14 @@ public class GeneratorTest {
.setColumnName("update_time") .setColumnName("update_time")
.setOnUpdateValue("NOW()"); .setOnUpdateValue("NOW()");
ColumnConfig userNameConfig = ColumnConfig.create()
.setColumnName("user_name")
.setMaskType(Masks.CHINESE_NAME);
// sys_user 表单独配置 // sys_user 表单独配置
TableConfig userTableConfig = TableConfig.create() TableConfig userTableConfig = TableConfig.create()
.setTableName("sys_user") .setTableName("sys_user")
.setColumnConfig(userNameConfig)
.setColumnConfig(userColumnConfig); .setColumnConfig(userColumnConfig);
// 全局字段配置 // 全局字段配置
@ -335,7 +341,7 @@ public class GeneratorTest {
// 配置生成 entity // 配置生成 entity
globalConfig.enableEntity() globalConfig.enableEntity()
.setAlwaysGenColumnAnnotation(true) // .setAlwaysGenColumnAnnotation(true)
.setOverwriteEnable(true) .setOverwriteEnable(true)
.setWithLombok(true); .setWithLombok(true);