|
@@ -24,12 +24,11 @@ import cn.iocoder.yudao.module.system.api.user.AdminUserApi;
|
|
|
import cn.iocoder.yudao.module.system.api.user.dto.AdminUserRespDTO;
|
|
|
import com.baomidou.mybatisplus.generator.config.po.TableField;
|
|
|
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
|
|
|
-import org.junit.jupiter.api.Disabled;
|
|
|
+import jakarta.annotation.Resource;
|
|
|
import org.junit.jupiter.api.Test;
|
|
|
import org.springframework.boot.test.mock.mockito.MockBean;
|
|
|
import org.springframework.context.annotation.Import;
|
|
|
|
|
|
-import jakarta.annotation.Resource;
|
|
|
import java.util.Arrays;
|
|
|
import java.util.Collections;
|
|
|
import java.util.List;
|
|
@@ -234,17 +233,16 @@ public class CodegenServiceImplTest extends BaseDbUnitTest {
|
|
|
}
|
|
|
|
|
|
@Test
|
|
|
- @Disabled // TODO @芋艿:这个单测会随机性失败,需要定位下;
|
|
|
public void testSyncCodegenFromDB() {
|
|
|
// mock 数据(CodegenTableDO)
|
|
|
CodegenTableDO table = randomPojo(CodegenTableDO.class, o -> o.setTableName("t_yunai")
|
|
|
.setDataSourceConfigId(1L).setScene(CodegenSceneEnum.ADMIN.getScene()));
|
|
|
codegenTableMapper.insert(table);
|
|
|
CodegenColumnDO column01 = randomPojo(CodegenColumnDO.class, o -> o.setTableId(table.getId())
|
|
|
- .setColumnName("id"));
|
|
|
+ .setColumnName("id").setPrimaryKey(true).setOrdinalPosition(0));
|
|
|
codegenColumnMapper.insert(column01);
|
|
|
CodegenColumnDO column02 = randomPojo(CodegenColumnDO.class, o -> o.setTableId(table.getId())
|
|
|
- .setColumnName("name"));
|
|
|
+ .setColumnName("name").setOrdinalPosition(1));
|
|
|
codegenColumnMapper.insert(column02);
|
|
|
// 准备参数
|
|
|
Long tableId = table.getId();
|
|
@@ -263,7 +261,7 @@ public class CodegenServiceImplTest extends BaseDbUnitTest {
|
|
|
when(databaseTableService.getTable(eq(1L), eq("t_yunai")))
|
|
|
.thenReturn(tableInfo);
|
|
|
// mock 方法(CodegenTableDO)
|
|
|
- List<CodegenColumnDO> newColumns = randomPojoList(CodegenColumnDO.class);
|
|
|
+ List<CodegenColumnDO> newColumns = randomPojoList(CodegenColumnDO.class, 2);
|
|
|
when(codegenBuilder.buildColumns(eq(table.getId()), argThat(tableFields -> {
|
|
|
assertEquals(2, tableFields.size());
|
|
|
assertSame(tableInfo.getFields(), tableFields);
|
|
@@ -457,9 +455,11 @@ public class CodegenServiceImplTest extends BaseDbUnitTest {
|
|
|
.setTemplateType(CodegenTemplateTypeEnum.ONE.getType()));
|
|
|
codegenTableMapper.insert(table);
|
|
|
// mock 数据(CodegenColumnDO)
|
|
|
- CodegenColumnDO column01 = randomPojo(CodegenColumnDO.class, o -> o.setTableId(table.getId()));
|
|
|
+ CodegenColumnDO column01 = randomPojo(CodegenColumnDO.class, o -> o.setTableId(table.getId())
|
|
|
+ .setOrdinalPosition(1));
|
|
|
codegenColumnMapper.insert(column01);
|
|
|
- CodegenColumnDO column02 = randomPojo(CodegenColumnDO.class, o -> o.setTableId(table.getId()));
|
|
|
+ CodegenColumnDO column02 = randomPojo(CodegenColumnDO.class, o -> o.setTableId(table.getId())
|
|
|
+ .setOrdinalPosition(2));
|
|
|
codegenColumnMapper.insert(column02);
|
|
|
// mock 执行生成
|
|
|
Map<String, String> codes = MapUtil.of(randomString(), randomString());
|
|
@@ -486,9 +486,11 @@ public class CodegenServiceImplTest extends BaseDbUnitTest {
|
|
|
.setTemplateType(CodegenTemplateTypeEnum.MASTER_NORMAL.getType()));
|
|
|
codegenTableMapper.insert(table);
|
|
|
// mock 数据(CodegenColumnDO)
|
|
|
- CodegenColumnDO column01 = randomPojo(CodegenColumnDO.class, o -> o.setTableId(table.getId()));
|
|
|
+ CodegenColumnDO column01 = randomPojo(CodegenColumnDO.class, o -> o.setTableId(table.getId())
|
|
|
+ .setOrdinalPosition(1));
|
|
|
codegenColumnMapper.insert(column01);
|
|
|
- CodegenColumnDO column02 = randomPojo(CodegenColumnDO.class, o -> o.setTableId(table.getId()));
|
|
|
+ CodegenColumnDO column02 = randomPojo(CodegenColumnDO.class, o -> o.setTableId(table.getId())
|
|
|
+ .setOrdinalPosition(2));
|
|
|
codegenColumnMapper.insert(column02);
|
|
|
// mock 数据(sub CodegenTableDO)
|
|
|
CodegenTableDO subTable = randomPojo(CodegenTableDO.class,
|