|
|
@@ -88,7 +88,7 @@ public class CodegenServiceImplTest extends BaseDbUnitTest {
|
|
|
TableInfo tableInfo = mock(TableInfo.class);
|
|
|
when(databaseTableService.getTable(eq(1L), eq("t_yunai")))
|
|
|
.thenReturn(tableInfo);
|
|
|
- when(tableInfo.getComment()).thenReturn("芋艿");
|
|
|
+ when(tableInfo.getComment()).thenReturn("张三");
|
|
|
// mock 方法(TableInfo fields)
|
|
|
TableField field01 = mock(TableField.class);
|
|
|
when(field01.getComment()).thenReturn("主键");
|
|
|
@@ -139,7 +139,7 @@ public class CodegenServiceImplTest extends BaseDbUnitTest {
|
|
|
assertServiceException(() -> codegenService.validateTableInfo(tableInfo),
|
|
|
CODEGEN_TABLE_INFO_TABLE_COMMENT_IS_NULL);
|
|
|
// 情况三
|
|
|
- when(tableInfo.getComment()).thenReturn("芋艿");
|
|
|
+ when(tableInfo.getComment()).thenReturn("张三");
|
|
|
assertServiceException(() -> codegenService.validateTableInfo(tableInfo),
|
|
|
CODEGEN_IMPORT_COLUMNS_NULL);
|
|
|
// 情况四
|
|
|
@@ -252,7 +252,7 @@ public class CodegenServiceImplTest extends BaseDbUnitTest {
|
|
|
TableInfo tableInfo = mock(TableInfo.class);
|
|
|
when(databaseTableService.getTable(eq(1L), eq("t_yunai")))
|
|
|
.thenReturn(tableInfo);
|
|
|
- when(tableInfo.getComment()).thenReturn("芋艿");
|
|
|
+ when(tableInfo.getComment()).thenReturn("张三");
|
|
|
// mock 方法(TableInfo fields)
|
|
|
TableField field01 = mock(TableField.class);
|
|
|
when(field01.getComment()).thenReturn("主键");
|
|
|
@@ -327,7 +327,7 @@ public class CodegenServiceImplTest extends BaseDbUnitTest {
|
|
|
// mock 数据
|
|
|
CodegenTableDO tableDO = randomPojo(CodegenTableDO.class, o -> {
|
|
|
o.setTableName("t_yunai");
|
|
|
- o.setTableComment("芋艿");
|
|
|
+ o.setTableComment("张三");
|
|
|
o.setClassName("SystemYunai");
|
|
|
o.setCreateTime(buildTime(2021, 3, 10));
|
|
|
}).setScene(CodegenSceneEnum.ADMIN.getScene());
|
|
|
@@ -538,7 +538,7 @@ public class CodegenServiceImplTest extends BaseDbUnitTest {
|
|
|
// mock 方法
|
|
|
TableInfo tableInfo01 = mock(TableInfo.class);
|
|
|
when(tableInfo01.getName()).thenReturn("t_yunai");
|
|
|
- when(tableInfo01.getComment()).thenReturn("芋艿");
|
|
|
+ when(tableInfo01.getComment()).thenReturn("张三");
|
|
|
TableInfo tableInfo02 = mock(TableInfo.class);
|
|
|
when(tableInfo02.getName()).thenReturn("t_yunai_02");
|
|
|
when(tableInfo02.getComment()).thenReturn("芋艿_02");
|
|
|
@@ -556,7 +556,7 @@ public class CodegenServiceImplTest extends BaseDbUnitTest {
|
|
|
// 断言
|
|
|
assertEquals(1, result.size());
|
|
|
assertEquals("t_yunai", result.get(0).getName());
|
|
|
- assertEquals("芋艿", result.get(0).getComment());
|
|
|
+ assertEquals("张三", result.get(0).getComment());
|
|
|
}
|
|
|
|
|
|
}
|