From d4e4c8cb7ce3f6f3ef16f10981a1e11a9342a6e3 Mon Sep 17 00:00:00 2001
From: chenbowen <you@example.com>
Date: Mon, 11 Aug 2025 09:22:01 +0800
Subject: [PATCH] =?UTF-8?q?1.=20=E4=BF=AE=E5=A4=8D=E5=8D=95=E5=85=83?=
 =?UTF-8?q?=E6=B5=8B=E8=AF=95=E7=94=A8=E4=BE=8B=E9=94=99=E8=AF=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

(cherry picked from commit 8673aee2816f288b19a4251cc4ad712547a3155c)
---
 .../infra/service/codegen/CodegenServiceImplTest.java  | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/yudao-module-infra/yudao-module-infra-server/src/test/java/cn/iocoder/yudao/module/infra/service/codegen/CodegenServiceImplTest.java b/yudao-module-infra/yudao-module-infra-server/src/test/java/cn/iocoder/yudao/module/infra/service/codegen/CodegenServiceImplTest.java
index da52e8c5..7d0fdbe1 100644
--- a/yudao-module-infra/yudao-module-infra-server/src/test/java/cn/iocoder/yudao/module/infra/service/codegen/CodegenServiceImplTest.java
+++ b/yudao-module-infra/yudao-module-infra-server/src/test/java/cn/iocoder/yudao/module/infra/service/codegen/CodegenServiceImplTest.java
@@ -76,10 +76,10 @@ public class CodegenServiceImplTest extends BaseDbUnitTest {
         // 准备参数
         String author = randomString();
         CodegenCreateListReqVO reqVO = randomPojo(CodegenCreateListReqVO.class,
-                o -> o.setDataSourceConfigId(1L).setTableNames(Collections.singletonList("t_yunai")));
+                o -> o.setDataSourceConfigId(1L).setTableNames(Collections.singletonList("t_yunai")).setIsStandardized(false));
         // mock 方法（TableInfo）
         TableInfo tableInfo = mock(TableInfo.class);
-        when(databaseTableService.getTable(eq(1L), eq("t_yunai"), false))
+        when(databaseTableService.getTable(eq(1L), eq("t_yunai"), eq(false)))
                 .thenReturn(tableInfo);
         when(tableInfo.getComment()).thenReturn("芋艿");
         // mock 方法（TableInfo fields）
@@ -91,7 +91,7 @@ public class CodegenServiceImplTest extends BaseDbUnitTest {
         when(tableInfo.getFields()).thenReturn(fields);
         // mock 方法（CodegenTableDO）
         CodegenTableDO table = randomPojo(CodegenTableDO.class);
-        when(codegenBuilder.buildTable(same(tableInfo), false)).thenReturn(table);
+        when(codegenBuilder.buildTable(same(tableInfo), eq(false))).thenReturn(table);
         // mock 方法（CodegenColumnDO）
         List<CodegenColumnDO> columns = randomPojoList(CodegenColumnDO.class);
         when(codegenBuilder.buildColumns(eq(table.getId()), same(fields)))
@@ -240,7 +240,7 @@ public class CodegenServiceImplTest extends BaseDbUnitTest {
         Long tableId = table.getId();
         // mock 方法（TableInfo）
         TableInfo tableInfo = mock(TableInfo.class);
-        when(databaseTableService.getTable(eq(1L), eq("t_yunai"), false))
+        when(databaseTableService.getTable(eq(1L), eq("t_yunai"), eq(false)))
                 .thenReturn(tableInfo);
         when(tableInfo.getComment()).thenReturn("芋艿");
         // mock 方法（TableInfo fields）
@@ -250,7 +250,7 @@ public class CodegenServiceImplTest extends BaseDbUnitTest {
         when(field03.getComment()).thenReturn("分类");
         List<TableField> fields = Arrays.asList(field01, field03);
         when(tableInfo.getFields()).thenReturn(fields);
-        when(databaseTableService.getTable(eq(1L), eq("t_yunai"), false))
+        when(databaseTableService.getTable(eq(1L), eq("t_yunai"), eq(false)))
                 .thenReturn(tableInfo);
         // mock 方法（CodegenTableDO）
         List<CodegenColumnDO> newColumns = randomPojoList(CodegenColumnDO.class, 2);