package cn.iocoder.yudao.module.system.service.social;
|
|
import cn.hutool.core.util.ReflectUtil;
|
import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
|
import cn.iocoder.yudao.framework.common.enums.UserTypeEnum;
|
import cn.iocoder.yudao.framework.common.pojo.PageResult;
|
import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest;
|
import cn.iocoder.yudao.module.system.controller.admin.socail.vo.client.SocialClientPageReqVO;
|
import cn.iocoder.yudao.module.system.controller.admin.socail.vo.client.SocialClientSaveReqVO;
|
import cn.iocoder.yudao.module.system.dal.dataobject.social.SocialClientDO;
|
import cn.iocoder.yudao.module.system.dal.mysql.social.SocialClientMapper;
|
import cn.iocoder.yudao.module.system.enums.social.SocialTypeEnum;
|
import cn.iocoder.yudao.module.system.framework.justauth.core.AuthRequestFactory;
|
import jakarta.annotation.Resource;
|
import me.zhyd.oauth.config.AuthConfig;
|
import me.zhyd.oauth.model.AuthResponse;
|
import me.zhyd.oauth.model.AuthUser;
|
import me.zhyd.oauth.request.AuthDefaultRequest;
|
import me.zhyd.oauth.request.AuthRequest;
|
import me.zhyd.oauth.utils.AuthStateUtils;
|
import org.junit.jupiter.api.Test;
|
import org.mockito.MockedStatic;
|
import org.springframework.context.annotation.Import;
|
import org.springframework.test.context.bean.override.mockito.MockitoBean;
|
|
import static cn.hutool.core.util.RandomUtil.randomEle;
|
import static cn.iocoder.yudao.framework.common.util.object.ObjectUtils.cloneIgnoreId;
|
import static cn.iocoder.yudao.framework.test.core.util.AssertUtils.assertPojoEquals;
|
import static cn.iocoder.yudao.framework.test.core.util.AssertUtils.assertServiceException;
|
import static cn.iocoder.yudao.framework.test.core.util.RandomUtils.*;
|
import static cn.iocoder.yudao.module.system.enums.ErrorCodeConstants.*;
|
import static org.junit.jupiter.api.Assertions.*;
|
import static org.mockito.Mockito.*;
|
|
/**
|
* {@link SocialClientServiceImpl} 的单元测试类
|
*
|
* @author 超级管理员
|
*/
|
@Import(SocialClientServiceImpl.class)
|
public class SocialClientServiceImplTest extends BaseDbUnitTest {
|
|
@Resource
|
private SocialClientServiceImpl socialClientService;
|
|
@Resource
|
private SocialClientMapper socialClientMapper;
|
|
@MockitoBean
|
private AuthRequestFactory authRequestFactory;
|
|
@Test
|
public void testGetAuthorizeUrl() {
|
try (MockedStatic<AuthStateUtils> authStateUtilsMock = mockStatic(AuthStateUtils.class)) {
|
// 准备参数
|
Integer socialType = SocialTypeEnum.WECHAT_MP.getType();
|
Integer userType = randomPojo(UserTypeEnum.class).getValue();
|
String redirectUri = "sss";
|
// mock 获得对应的 AuthRequest 实现
|
AuthRequest authRequest = mock(AuthRequest.class);
|
when(authRequestFactory.get(eq("WECHAT_MP"))).thenReturn(authRequest);
|
// mock 方法
|
authStateUtilsMock.when(AuthStateUtils::createState).thenReturn("aoteman");
|
when(authRequest.authorize(eq("aoteman"))).thenReturn("https://www.iocoder.cn?redirect_uri=yyy");
|
|
// 调用
|
String url = socialClientService.getAuthorizeUrl(socialType, userType, redirectUri);
|
// 断言
|
assertEquals("https://www.iocoder.cn?redirect_uri=sss", url);
|
}
|
}
|
|
@Test
|
public void testAuthSocialUser_success() {
|
// 准备参数
|
Integer socialType = SocialTypeEnum.WECHAT_MP.getType();
|
Integer userType = randomPojo(UserTypeEnum.class).getValue();
|
String code = randomString();
|
String state = randomString();
|
// mock 方法（AuthRequest）
|
AuthRequest authRequest = mock(AuthRequest.class);
|
when(authRequestFactory.get(eq("WECHAT_MP"))).thenReturn(authRequest);
|
// mock 方法（AuthResponse）
|
AuthUser authUser = randomPojo(AuthUser.class);
|
AuthResponse<AuthUser> authResponse = new AuthResponse<>(2000, null, authUser);
|
when(authRequest.login(argThat(authCallback -> {
|
assertEquals(code, authCallback.getCode());
|
assertEquals(state, authCallback.getState());
|
return true;
|
}))).thenReturn(authResponse);
|
|
// 调用
|
AuthUser result = socialClientService.getAuthUser(socialType, userType, code, state);
|
// 断言
|
assertSame(authUser, result);
|
}
|
|
@Test
|
public void testAuthSocialUser_fail() {
|
// 准备参数
|
Integer socialType = SocialTypeEnum.WECHAT_MP.getType();
|
Integer userType = randomPojo(UserTypeEnum.class).getValue();
|
String code = randomString();
|
String state = randomString();
|
// mock 方法（AuthRequest）
|
AuthRequest authRequest = mock(AuthRequest.class);
|
when(authRequestFactory.get(eq("WECHAT_MP"))).thenReturn(authRequest);
|
// mock 方法（AuthResponse）
|
AuthResponse<AuthUser> authResponse = new AuthResponse<>(0, "模拟失败", null);
|
when(authRequest.login(argThat(authCallback -> {
|
assertEquals(code, authCallback.getCode());
|
assertEquals(state, authCallback.getState());
|
return true;
|
}))).thenReturn(authResponse);
|
|
// 调用并断言
|
assertServiceException(
|
() -> socialClientService.getAuthUser(socialType, userType, code, state),
|
SOCIAL_USER_AUTH_FAILURE, "模拟失败");
|
}
|
|
@Test
|
public void testBuildAuthRequest_clientNull() {
|
// 准备参数
|
Integer socialType = SocialTypeEnum.WECHAT_MP.getType();
|
Integer userType = randomPojo(SocialTypeEnum.class).getType();
|
// mock 获得对应的 AuthRequest 实现
|
AuthRequest authRequest = mock(AuthDefaultRequest.class);
|
AuthConfig authConfig = (AuthConfig) ReflectUtil.getFieldValue(authRequest, "config");
|
when(authRequestFactory.get(eq("WECHAT_MP"))).thenReturn(authRequest);
|
|
// 调用
|
AuthRequest result = socialClientService.buildAuthRequest(socialType, userType);
|
// 断言
|
assertSame(authRequest, result);
|
assertSame(authConfig, ReflectUtil.getFieldValue(authConfig, "config"));
|
}
|
|
@Test
|
public void testBuildAuthRequest_clientDisable() {
|
// 准备参数
|
Integer socialType = SocialTypeEnum.WECHAT_MP.getType();
|
Integer userType = randomPojo(SocialTypeEnum.class).getType();
|
// mock 获得对应的 AuthRequest 实现
|
AuthRequest authRequest = mock(AuthDefaultRequest.class);
|
AuthConfig authConfig = (AuthConfig) ReflectUtil.getFieldValue(authRequest, "config");
|
when(authRequestFactory.get(eq("WECHAT_MP"))).thenReturn(authRequest);
|
// mock 数据
|
SocialClientDO client = randomPojo(SocialClientDO.class, o -> o.setStatus(CommonStatusEnum.DISABLE.getStatus())
|
.setUserType(userType).setSocialType(socialType));
|
socialClientMapper.insert(client);
|
|
// 调用
|
AuthRequest result = socialClientService.buildAuthRequest(socialType, userType);
|
// 断言
|
assertSame(authRequest, result);
|
assertSame(authConfig, ReflectUtil.getFieldValue(authConfig, "config"));
|
}
|
|
@Test
|
public void testBuildAuthRequest_clientEnable() {
|
// 准备参数
|
Integer socialType = SocialTypeEnum.WECHAT_MP.getType();
|
Integer userType = randomPojo(SocialTypeEnum.class).getType();
|
// mock 获得对应的 AuthRequest 实现
|
AuthConfig authConfig = mock(AuthConfig.class);
|
AuthRequest authRequest = mock(AuthDefaultRequest.class);
|
ReflectUtil.setFieldValue(authRequest, "config", authConfig);
|
when(authRequestFactory.get(eq("WECHAT_MP"))).thenReturn(authRequest);
|
// mock 数据
|
SocialClientDO client = randomPojo(SocialClientDO.class, o -> o.setStatus(CommonStatusEnum.ENABLE.getStatus())
|
.setUserType(userType).setSocialType(socialType));
|
socialClientMapper.insert(client);
|
|
// 调用
|
AuthRequest result = socialClientService.buildAuthRequest(socialType, userType);
|
// 断言
|
assertSame(authRequest, result);
|
assertNotSame(authConfig, ReflectUtil.getFieldValue(authRequest, "config"));
|
}
|
|
// =================== 客户端管理 ===================
|
|
@Test
|
public void testCreateSocialClient_success() {
|
// 准备参数
|
SocialClientSaveReqVO reqVO = randomPojo(SocialClientSaveReqVO.class,
|
o -> o.setSocialType(randomEle(SocialTypeEnum.values()).getType())
|
.setUserType(randomEle(UserTypeEnum.values()).getValue())
|
.setStatus(randomCommonStatus()))
|
.setId(null); // 防止 id 被赋值
|
|
// 调用
|
Long socialClientId = socialClientService.createSocialClient(reqVO);
|
// 断言
|
assertNotNull(socialClientId);
|
// 校验记录的属性是否正确
|
SocialClientDO socialClient = socialClientMapper.selectById(socialClientId);
|
assertPojoEquals(reqVO, socialClient, "id");
|
}
|
|
@Test
|
public void testUpdateSocialClient_success() {
|
// mock 数据
|
SocialClientDO dbSocialClient = randomPojo(SocialClientDO.class);
|
socialClientMapper.insert(dbSocialClient);// @Sql: 先插入出一条存在的数据
|
// 准备参数
|
SocialClientSaveReqVO reqVO = randomPojo(SocialClientSaveReqVO.class, o -> {
|
o.setId(dbSocialClient.getId()); // 设置更新的 ID
|
o.setSocialType(randomEle(SocialTypeEnum.values()).getType())
|
.setUserType(randomEle(UserTypeEnum.values()).getValue())
|
.setStatus(randomCommonStatus());
|
});
|
|
// 调用
|
socialClientService.updateSocialClient(reqVO);
|
// 校验是否更新正确
|
SocialClientDO socialClient = socialClientMapper.selectById(reqVO.getId()); // 获取最新的
|
assertPojoEquals(reqVO, socialClient);
|
}
|
|
@Test
|
public void testUpdateSocialClient_notExists() {
|
// 准备参数
|
SocialClientSaveReqVO reqVO = randomPojo(SocialClientSaveReqVO.class);
|
|
// 调用, 并断言异常
|
assertServiceException(() -> socialClientService.updateSocialClient(reqVO), SOCIAL_CLIENT_NOT_EXISTS);
|
}
|
|
@Test
|
public void testDeleteSocialClient_success() {
|
// mock 数据
|
SocialClientDO dbSocialClient = randomPojo(SocialClientDO.class);
|
socialClientMapper.insert(dbSocialClient);// @Sql: 先插入出一条存在的数据
|
// 准备参数
|
Long id = dbSocialClient.getId();
|
|
// 调用
|
socialClientService.deleteSocialClient(id);
|
// 校验数据不存在了
|
assertNull(socialClientMapper.selectById(id));
|
}
|
|
@Test
|
public void testDeleteSocialClient_notExists() {
|
// 准备参数
|
Long id = randomLongId();
|
|
// 调用, 并断言异常
|
assertServiceException(() -> socialClientService.deleteSocialClient(id), SOCIAL_CLIENT_NOT_EXISTS);
|
}
|
|
@Test
|
public void testGetSocialClient() {
|
// mock 数据
|
SocialClientDO dbSocialClient = randomPojo(SocialClientDO.class);
|
socialClientMapper.insert(dbSocialClient);// @Sql: 先插入出一条存在的数据
|
// 准备参数
|
Long id = dbSocialClient.getId();
|
|
// 调用
|
SocialClientDO socialClient = socialClientService.getSocialClient(id);
|
// 校验数据正确
|
assertPojoEquals(dbSocialClient, socialClient);
|
}
|
|
@Test
|
public void testGetSocialClientPage() {
|
// mock 数据
|
SocialClientDO dbSocialClient = randomPojo(SocialClientDO.class, o -> { // 等会查询到
|
o.setName("芋头");
|
o.setSocialType(SocialTypeEnum.GITEE.getType());
|
o.setUserType(UserTypeEnum.ADMIN.getValue());
|
o.setClientId("yudao");
|
o.setStatus(CommonStatusEnum.ENABLE.getStatus());
|
});
|
socialClientMapper.insert(dbSocialClient);
|
// 测试 name 不匹配
|
socialClientMapper.insert(cloneIgnoreId(dbSocialClient, o -> o.setName(randomString())));
|
// 测试 socialType 不匹配
|
socialClientMapper.insert(cloneIgnoreId(dbSocialClient, o -> o.setSocialType(SocialTypeEnum.DINGTALK.getType())));
|
// 测试 userType 不匹配
|
socialClientMapper.insert(cloneIgnoreId(dbSocialClient, o -> o.setUserType(UserTypeEnum.MEMBER.getValue())));
|
// 测试 clientId 不匹配
|
socialClientMapper.insert(cloneIgnoreId(dbSocialClient, o -> o.setClientId("dao")));
|
// 测试 status 不匹配
|
socialClientMapper.insert(cloneIgnoreId(dbSocialClient, o -> o.setStatus(CommonStatusEnum.DISABLE.getStatus())));
|
// 准备参数
|
SocialClientPageReqVO reqVO = new SocialClientPageReqVO();
|
reqVO.setName("芋");
|
reqVO.setSocialType(SocialTypeEnum.GITEE.getType());
|
reqVO.setUserType(UserTypeEnum.ADMIN.getValue());
|
reqVO.setClientId("yu");
|
reqVO.setStatus(CommonStatusEnum.ENABLE.getStatus());
|
|
// 调用
|
PageResult<SocialClientDO> pageResult = socialClientService.getSocialClientPage(reqVO);
|
// 断言
|
assertEquals(1, pageResult.getTotal());
|
assertEquals(1, pageResult.getList().size());
|
assertPojoEquals(dbSocialClient, pageResult.getList().get(0));
|
}
|
|
}
|
