|
@@ -5,10 +5,8 @@ import cn.iocoder.yudao.framework.common.pojo.PageResult;
|
|
|
import cn.iocoder.yudao.framework.tenant.config.TenantProperties;
|
|
|
import cn.iocoder.yudao.framework.tenant.core.context.TenantContextHolder;
|
|
|
import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest;
|
|
|
-import cn.iocoder.yudao.module.system.controller.admin.tenant.vo.tenant.TenantCreateReqVO;
|
|
|
-import cn.iocoder.yudao.module.system.controller.admin.tenant.vo.tenant.TenantExportReqVO;
|
|
|
import cn.iocoder.yudao.module.system.controller.admin.tenant.vo.tenant.TenantPageReqVO;
|
|
|
-import cn.iocoder.yudao.module.system.controller.admin.tenant.vo.tenant.TenantUpdateReqVO;
|
|
|
+import cn.iocoder.yudao.module.system.controller.admin.tenant.vo.tenant.TenantSaveReqVO;
|
|
|
import cn.iocoder.yudao.module.system.dal.dataobject.permission.MenuDO;
|
|
|
import cn.iocoder.yudao.module.system.dal.dataobject.permission.RoleDO;
|
|
|
import cn.iocoder.yudao.module.system.dal.dataobject.tenant.TenantDO;
|
|
@@ -152,7 +150,7 @@ public class TenantServiceImplTest extends BaseDbUnitTest {
|
|
|
}))).thenReturn(300L);
|
|
|
|
|
|
// 准备参数
|
|
|
- TenantCreateReqVO reqVO = randomPojo(TenantCreateReqVO.class, o -> {
|
|
|
+ TenantSaveReqVO reqVO = randomPojo(TenantSaveReqVO.class, o -> {
|
|
|
o.setContactName("芋道");
|
|
|
o.setContactMobile("15601691300");
|
|
|
o.setPackageId(100L);
|
|
@@ -160,7 +158,7 @@ public class TenantServiceImplTest extends BaseDbUnitTest {
|
|
|
o.setWebsite("https://www.iocoder.cn");
|
|
|
o.setUsername("yunai");
|
|
|
o.setPassword("yuanma");
|
|
|
- });
|
|
|
+ }).setId(null); // 设置为 null,方便后面校验
|
|
|
|
|
|
// 调用
|
|
|
Long tenantId = tenantService.createTenant(reqVO);
|
|
@@ -168,7 +166,7 @@ public class TenantServiceImplTest extends BaseDbUnitTest {
|
|
|
assertNotNull(tenantId);
|
|
|
// 校验记录的属性是否正确
|
|
|
TenantDO tenant = tenantMapper.selectById(tenantId);
|
|
|
- assertPojoEquals(reqVO, tenant);
|
|
|
+ assertPojoEquals(reqVO, tenant, "id");
|
|
|
assertEquals(300L, tenant.getContactUserId());
|
|
|
// verify 分配权限
|
|
|
verify(permissionService).assignRoleMenu(eq(200L), same(tenantPackage.getMenuIds()));
|
|
@@ -182,7 +180,7 @@ public class TenantServiceImplTest extends BaseDbUnitTest {
|
|
|
TenantDO dbTenant = randomPojo(TenantDO.class, o -> o.setStatus(randomCommonStatus()));
|
|
|
tenantMapper.insert(dbTenant);// @Sql: 先插入出一条存在的数据
|
|
|
// 准备参数
|
|
|
- TenantUpdateReqVO reqVO = randomPojo(TenantUpdateReqVO.class, o -> {
|
|
|
+ TenantSaveReqVO reqVO = randomPojo(TenantSaveReqVO.class, o -> {
|
|
|
o.setId(dbTenant.getId()); // 设置更新的 ID
|
|
|
o.setStatus(randomCommonStatus());
|
|
|
o.setWebsite(randomString());
|
|
@@ -214,7 +212,7 @@ public class TenantServiceImplTest extends BaseDbUnitTest {
|
|
|
@Test
|
|
|
public void testUpdateTenant_notExists() {
|
|
|
// 准备参数
|
|
|
- TenantUpdateReqVO reqVO = randomPojo(TenantUpdateReqVO.class);
|
|
|
+ TenantSaveReqVO reqVO = randomPojo(TenantSaveReqVO.class);
|
|
|
|
|
|
// 调用, 并断言异常
|
|
|
assertServiceException(() -> tenantService.updateTenant(reqVO), TENANT_NOT_EXISTS);
|
|
@@ -226,7 +224,7 @@ public class TenantServiceImplTest extends BaseDbUnitTest {
|
|
|
TenantDO dbTenant = randomPojo(TenantDO.class, o -> o.setPackageId(PACKAGE_ID_SYSTEM));
|
|
|
tenantMapper.insert(dbTenant);// @Sql: 先插入出一条存在的数据
|
|
|
// 准备参数
|
|
|
- TenantUpdateReqVO reqVO = randomPojo(TenantUpdateReqVO.class, o -> {
|
|
|
+ TenantSaveReqVO reqVO = randomPojo(TenantSaveReqVO.class, o -> {
|
|
|
o.setId(dbTenant.getId()); // 设置更新的 ID
|
|
|
});
|
|
|
|
|
@@ -321,42 +319,6 @@ public class TenantServiceImplTest extends BaseDbUnitTest {
|
|
|
assertPojoEquals(dbTenant, pageResult.getList().get(0));
|
|
|
}
|
|
|
|
|
|
- @Test
|
|
|
- public void testGetTenantList() {
|
|
|
- // mock 数据
|
|
|
- TenantDO dbTenant = randomPojo(TenantDO.class, o -> { // 等会查询到
|
|
|
- o.setName("芋道源码");
|
|
|
- o.setContactName("芋艿");
|
|
|
- o.setContactMobile("15601691300");
|
|
|
- o.setStatus(CommonStatusEnum.ENABLE.getStatus());
|
|
|
- o.setCreateTime(buildTime(2020, 12, 12));
|
|
|
- });
|
|
|
- tenantMapper.insert(dbTenant);
|
|
|
- // 测试 name 不匹配
|
|
|
- tenantMapper.insert(cloneIgnoreId(dbTenant, o -> o.setName(randomString())));
|
|
|
- // 测试 contactName 不匹配
|
|
|
- tenantMapper.insert(cloneIgnoreId(dbTenant, o -> o.setContactName(randomString())));
|
|
|
- // 测试 contactMobile 不匹配
|
|
|
- tenantMapper.insert(cloneIgnoreId(dbTenant, o -> o.setContactMobile(randomString())));
|
|
|
- // 测试 status 不匹配
|
|
|
- tenantMapper.insert(cloneIgnoreId(dbTenant, o -> o.setStatus(CommonStatusEnum.DISABLE.getStatus())));
|
|
|
- // 测试 createTime 不匹配
|
|
|
- tenantMapper.insert(cloneIgnoreId(dbTenant, o -> o.setCreateTime(buildTime(2021, 12, 12))));
|
|
|
- // 准备参数
|
|
|
- TenantExportReqVO reqVO = new TenantExportReqVO();
|
|
|
- reqVO.setName("芋道");
|
|
|
- reqVO.setContactName("艿");
|
|
|
- reqVO.setContactMobile("1560");
|
|
|
- reqVO.setStatus(CommonStatusEnum.ENABLE.getStatus());
|
|
|
- reqVO.setCreateTime(buildBetweenTime(2020, 12, 1, 2020, 12, 24));
|
|
|
-
|
|
|
- // 调用
|
|
|
- List<TenantDO> list = tenantService.getTenantList(reqVO);
|
|
|
- // 断言
|
|
|
- assertEquals(1, list.size());
|
|
|
- assertPojoEquals(dbTenant, list.get(0));
|
|
|
- }
|
|
|
-
|
|
|
@Test
|
|
|
public void testGetTenantByName() {
|
|
|
// mock 数据
|