|
@@ -1,6 +1,6 @@
|
|
|
package cn.iocoder.yudao.module.system.service.oauth2;
|
|
|
|
|
|
-import cn.hutool.core.map.MapUtil;
|
|
|
+import cn.hutool.extra.spring.SpringUtil;
|
|
|
import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
|
|
|
import cn.iocoder.yudao.framework.common.pojo.PageResult;
|
|
|
import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest;
|
|
@@ -9,14 +9,12 @@ import cn.iocoder.yudao.module.system.controller.admin.oauth2.vo.client.OAuth2Cl
|
|
|
import cn.iocoder.yudao.module.system.controller.admin.oauth2.vo.client.OAuth2ClientUpdateReqVO;
|
|
|
import cn.iocoder.yudao.module.system.dal.dataobject.oauth2.OAuth2ClientDO;
|
|
|
import cn.iocoder.yudao.module.system.dal.mysql.oauth2.OAuth2ClientMapper;
|
|
|
-import cn.iocoder.yudao.module.system.mq.producer.auth.OAuth2ClientProducer;
|
|
|
import org.junit.jupiter.api.Test;
|
|
|
-import org.springframework.boot.test.mock.mockito.MockBean;
|
|
|
+import org.mockito.MockedStatic;
|
|
|
import org.springframework.context.annotation.Import;
|
|
|
|
|
|
import javax.annotation.Resource;
|
|
|
import java.util.Collections;
|
|
|
-import java.util.Map;
|
|
|
|
|
|
import static cn.iocoder.yudao.framework.common.util.object.ObjectUtils.cloneIgnoreId;
|
|
|
import static cn.iocoder.yudao.framework.test.core.util.AssertUtils.assertPojoEquals;
|
|
@@ -24,7 +22,8 @@ import static cn.iocoder.yudao.framework.test.core.util.AssertUtils.assertServic
|
|
|
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.verify;
|
|
|
+import static org.mockito.ArgumentMatchers.eq;
|
|
|
+import static org.mockito.Mockito.mockStatic;
|
|
|
|
|
|
/**
|
|
|
* {@link OAuth2ClientServiceImpl} 的单元测试类
|
|
@@ -40,26 +39,6 @@ public class OAuth2ClientServiceImplTest extends BaseDbUnitTest {
|
|
|
@Resource
|
|
|
private OAuth2ClientMapper oauth2ClientMapper;
|
|
|
|
|
|
- @MockBean
|
|
|
- private OAuth2ClientProducer oauth2ClientProducer;
|
|
|
-
|
|
|
- @Test
|
|
|
- public void testInitLocalCache() {
|
|
|
- // mock 数据
|
|
|
- OAuth2ClientDO clientDO1 = randomPojo(OAuth2ClientDO.class);
|
|
|
- oauth2ClientMapper.insert(clientDO1);
|
|
|
- OAuth2ClientDO clientDO2 = randomPojo(OAuth2ClientDO.class);
|
|
|
- oauth2ClientMapper.insert(clientDO2);
|
|
|
-
|
|
|
- // 调用
|
|
|
- oauth2ClientService.initLocalCache();
|
|
|
- // 断言 clientCache 缓存
|
|
|
- Map<String, OAuth2ClientDO> clientCache = oauth2ClientService.getClientCache();
|
|
|
- assertEquals(2, clientCache.size());
|
|
|
- assertPojoEquals(clientDO1, clientCache.get(clientDO1.getClientId()));
|
|
|
- assertPojoEquals(clientDO2, clientCache.get(clientDO2.getClientId()));
|
|
|
- }
|
|
|
-
|
|
|
@Test
|
|
|
public void testCreateOAuth2Client_success() {
|
|
|
// 准备参数
|
|
@@ -73,7 +52,6 @@ public class OAuth2ClientServiceImplTest extends BaseDbUnitTest {
|
|
|
// 校验记录的属性是否正确
|
|
|
OAuth2ClientDO oAuth2Client = oauth2ClientMapper.selectById(oauth2ClientId);
|
|
|
assertPojoEquals(reqVO, oAuth2Client);
|
|
|
- verify(oauth2ClientProducer).sendOAuth2ClientRefreshMessage();
|
|
|
}
|
|
|
|
|
|
@Test
|
|
@@ -92,7 +70,6 @@ public class OAuth2ClientServiceImplTest extends BaseDbUnitTest {
|
|
|
// 校验是否更新正确
|
|
|
OAuth2ClientDO oAuth2Client = oauth2ClientMapper.selectById(reqVO.getId()); // 获取最新的
|
|
|
assertPojoEquals(reqVO, oAuth2Client);
|
|
|
- verify(oauth2ClientProducer).sendOAuth2ClientRefreshMessage();
|
|
|
}
|
|
|
|
|
|
@Test
|
|
@@ -116,7 +93,6 @@ public class OAuth2ClientServiceImplTest extends BaseDbUnitTest {
|
|
|
oauth2ClientService.deleteOAuth2Client(id);
|
|
|
// 校验数据不存在了
|
|
|
assertNull(oauth2ClientMapper.selectById(id));
|
|
|
- verify(oauth2ClientProducer).sendOAuth2ClientRefreshMessage();
|
|
|
}
|
|
|
|
|
|
@Test
|
|
@@ -166,6 +142,19 @@ public class OAuth2ClientServiceImplTest extends BaseDbUnitTest {
|
|
|
assertPojoEquals(clientDO, dbClientDO);
|
|
|
}
|
|
|
|
|
|
+ @Test
|
|
|
+ public void testGetOAuth2ClientFromCache() {
|
|
|
+ // mock 数据
|
|
|
+ OAuth2ClientDO clientDO = randomPojo(OAuth2ClientDO.class);
|
|
|
+ oauth2ClientMapper.insert(clientDO);
|
|
|
+ // 准备参数
|
|
|
+ String clientId = clientDO.getClientId();
|
|
|
+
|
|
|
+ // 调用,并断言
|
|
|
+ OAuth2ClientDO dbClientDO = oauth2ClientService.getOAuth2ClientFromCache(clientId);
|
|
|
+ assertPojoEquals(clientDO, dbClientDO);
|
|
|
+ }
|
|
|
+
|
|
|
@Test
|
|
|
public void testGetOAuth2ClientPage() {
|
|
|
// mock 数据
|
|
@@ -193,36 +182,39 @@ public class OAuth2ClientServiceImplTest extends BaseDbUnitTest {
|
|
|
|
|
|
@Test
|
|
|
public void testValidOAuthClientFromCache() {
|
|
|
- // mock 方法
|
|
|
- OAuth2ClientDO client = randomPojo(OAuth2ClientDO.class).setClientId("default")
|
|
|
- .setStatus(CommonStatusEnum.ENABLE.getStatus());
|
|
|
- OAuth2ClientDO client02 = randomPojo(OAuth2ClientDO.class).setClientId("disable")
|
|
|
- .setStatus(CommonStatusEnum.DISABLE.getStatus());
|
|
|
- Map<String, OAuth2ClientDO> clientCache = MapUtil.<String, OAuth2ClientDO>builder()
|
|
|
- .put(client.getClientId(), client)
|
|
|
- .put(client02.getClientId(), client02).build();
|
|
|
- oauth2ClientService.setClientCache(clientCache);
|
|
|
-
|
|
|
- // 调用,并断言
|
|
|
- assertServiceException(() -> oauth2ClientService.validOAuthClientFromCache(randomString(),
|
|
|
- null, null, null, null), OAUTH2_CLIENT_NOT_EXISTS);
|
|
|
- assertServiceException(() -> oauth2ClientService.validOAuthClientFromCache("disable",
|
|
|
- null, null, null, null), OAUTH2_CLIENT_DISABLE);
|
|
|
- assertServiceException(() -> oauth2ClientService.validOAuthClientFromCache("default",
|
|
|
- randomString(), null, null, null), OAUTH2_CLIENT_CLIENT_SECRET_ERROR);
|
|
|
- assertServiceException(() -> oauth2ClientService.validOAuthClientFromCache("default",
|
|
|
- null, randomString(), null, null), OAUTH2_CLIENT_AUTHORIZED_GRANT_TYPE_NOT_EXISTS);
|
|
|
- assertServiceException(() -> oauth2ClientService.validOAuthClientFromCache("default",
|
|
|
- null, null, Collections.singleton(randomString()), null), OAUTH2_CLIENT_SCOPE_OVER);
|
|
|
- assertServiceException(() -> oauth2ClientService.validOAuthClientFromCache("default",
|
|
|
- null, null, null, "test"), OAUTH2_CLIENT_REDIRECT_URI_NOT_MATCH, "test");
|
|
|
- // 成功调用(1:参数完整)
|
|
|
- OAuth2ClientDO result = oauth2ClientService.validOAuthClientFromCache(client.getClientId(), client.getSecret(),
|
|
|
- client.getAuthorizedGrantTypes().get(0), client.getScopes(), client.getRedirectUris().get(0));
|
|
|
- assertPojoEquals(client, result);
|
|
|
- // 成功调用(2:只有 clientId 参数)
|
|
|
- result = oauth2ClientService.validOAuthClientFromCache(client.getClientId());
|
|
|
- assertPojoEquals(client, result);
|
|
|
+ try (MockedStatic<SpringUtil> springUtilMockedStatic = mockStatic(SpringUtil.class)) {
|
|
|
+ springUtilMockedStatic.when(() -> SpringUtil.getBean(eq(OAuth2ClientServiceImpl.class)))
|
|
|
+ .thenReturn(oauth2ClientService);
|
|
|
+
|
|
|
+ // mock 方法
|
|
|
+ OAuth2ClientDO client = randomPojo(OAuth2ClientDO.class).setClientId("default")
|
|
|
+ .setStatus(CommonStatusEnum.ENABLE.getStatus());
|
|
|
+ oauth2ClientMapper.insert(client);
|
|
|
+ OAuth2ClientDO client02 = randomPojo(OAuth2ClientDO.class).setClientId("disable")
|
|
|
+ .setStatus(CommonStatusEnum.DISABLE.getStatus());
|
|
|
+ oauth2ClientMapper.insert(client02);
|
|
|
+
|
|
|
+ // 调用,并断言
|
|
|
+ assertServiceException(() -> oauth2ClientService.validOAuthClientFromCache(randomString(),
|
|
|
+ null, null, null, null), OAUTH2_CLIENT_NOT_EXISTS);
|
|
|
+ assertServiceException(() -> oauth2ClientService.validOAuthClientFromCache("disable",
|
|
|
+ null, null, null, null), OAUTH2_CLIENT_DISABLE);
|
|
|
+ assertServiceException(() -> oauth2ClientService.validOAuthClientFromCache("default",
|
|
|
+ randomString(), null, null, null), OAUTH2_CLIENT_CLIENT_SECRET_ERROR);
|
|
|
+ assertServiceException(() -> oauth2ClientService.validOAuthClientFromCache("default",
|
|
|
+ null, randomString(), null, null), OAUTH2_CLIENT_AUTHORIZED_GRANT_TYPE_NOT_EXISTS);
|
|
|
+ assertServiceException(() -> oauth2ClientService.validOAuthClientFromCache("default",
|
|
|
+ null, null, Collections.singleton(randomString()), null), OAUTH2_CLIENT_SCOPE_OVER);
|
|
|
+ assertServiceException(() -> oauth2ClientService.validOAuthClientFromCache("default",
|
|
|
+ null, null, null, "test"), OAUTH2_CLIENT_REDIRECT_URI_NOT_MATCH, "test");
|
|
|
+ // 成功调用(1:参数完整)
|
|
|
+ OAuth2ClientDO result = oauth2ClientService.validOAuthClientFromCache(client.getClientId(), client.getSecret(),
|
|
|
+ client.getAuthorizedGrantTypes().get(0), client.getScopes(), client.getRedirectUris().get(0));
|
|
|
+ assertPojoEquals(client, result);
|
|
|
+ // 成功调用(2:只有 clientId 参数)
|
|
|
+ result = oauth2ClientService.validOAuthClientFromCache(client.getClientId());
|
|
|
+ assertPojoEquals(client, result);
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
}
|