zgty/zt-dsc
8
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

1. 修复单元测试用例错误

Browse Source
This commit is contained in:
chenbowen
2025-08-11 09:22:01 +08:00
parent 3812611b04
commit 8673aee281
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
yudao-module-infra/yudao-module-infra-server/src/test/java/cn/iocoder/yudao/module/infra/service/codegen/CodegenServiceImplTest.java
View File

@@ -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);