tongtong/youlai-boot
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: 动态数据源单元测试优化

Browse Source
This commit is contained in:
haoxr
2023-04-29 17:49:07 +08:00
parent cbd1621935
commit 1b1888c36d
4 changed files with 5 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/main/java/com/youlai/system/sample/DynamicDataSourceSample.java
View File

@@ -24,7 +24,7 @@ public class DynamicDataSourceSample {
@Transactional
public boolean updateDictTypeCode(Long dictTypeId,String newTypeCode) {
public boolean updateMaster(Long dictTypeId,String newTypeCode) {
SysDictType dictType = dictTypeService.getById(dictTypeId);
String originalTypeCode = dictType.getCode();
@@ -32,7 +32,7 @@ public class DynamicDataSourceSample {
boolean result = dictTypeService.updateById(dictType);
if (result) {
result = dictService.updateDictTypeCode(originalTypeCode, newTypeCode);
result = dictService.updateSlave(originalTypeCode, newTypeCode);
}
return result;
}

2
src/main/java/com/youlai/system/service/SysDictService.java
View File

@@ -73,5 +73,5 @@ public interface SysDictService extends IService<SysDict> {
* @param newTypeCode
* @return
*/
boolean updateDictTypeCode(String originalTypeCode, String newTypeCode);
boolean updateSlave(String originalTypeCode, String newTypeCode);
}

2
src/main/java/com/youlai/system/service/impl/SysDictServiceImpl.java
View File

@@ -170,7 +170,7 @@ public class SysDictServiceImpl extends ServiceImpl<SysDictMapper, SysDict> impl
*/
@DS("slave")
@Override
public boolean updateDictTypeCode(String originalTypeCode, String newTypeCode) {
public boolean updateSlave(String originalTypeCode, String newTypeCode) {
boolean result = this.update(new LambdaUpdateWrapper<SysDict>()
.eq(SysDict::getTypeCode, originalTypeCode)
.set(SysDict::getTypeCode, newTypeCode)

8
src/test/java/com/youlai/system/DynamicDataSourceTest.java
View File

@@ -1,17 +1,11 @@
package com.youlai.system;
import cn.hutool.core.util.RandomUtil;
import com.baomidou.dynamic.datasource.annotation.DS;
import com.youlai.system.pojo.form.RoleForm;
import com.youlai.system.pojo.form.UserForm;
import com.youlai.system.sample.DynamicDataSourceSample;
import com.youlai.system.service.SysRoleService;
import com.youlai.system.service.SysUserService;
import lombok.extern.slf4j.Slf4j;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.transaction.annotation.Transactional;
@SpringBootTest
@Slf4j
@@ -24,7 +18,7 @@ class DynamicDataSourceTest {
@Test
void testDynamicDataSourceWithTransactional() {
String newTypeCod = RandomUtil.randomString(RandomUtil.BASE_CHAR, 6).toUpperCase();
boolean result = dynamicDataSourceSample.updateDictTypeCode(2L, newTypeCod);
boolean result = dynamicDataSourceSample.updateMaster(2L, newTypeCod);
log.info("testDynamicDataSourceWithTransactional result:{}", result);
}
}