//package com.iailab.module.system.controller.admin.oauth2;
//
//import cn.hutool.core.collection.ListUtil;
//import cn.hutool.core.date.LocalDateTimeUtil;
//import cn.hutool.core.map.MapUtil;
//import com.iailab.framework.common.core.KeyValue;
//import com.iailab.framework.common.enums.UserTypeEnum;
//import com.iailab.framework.common.exception.ErrorCode;
//import com.iailab.framework.common.pojo.CommonResult;
//import com.iailab.framework.common.util.collection.SetUtils;
//import com.iailab.framework.common.util.object.ObjectUtils;
//import com.iailab.framework.test.core.ut.BaseMockitoUnitTest;
//import com.iailab.module.system.controller.admin.oauth2.vo.open.OAuth2OpenAccessTokenRespVO;
//import com.iailab.module.system.controller.admin.oauth2.vo.open.OAuth2OpenAuthorizeInfoRespVO;
//import com.iailab.module.system.controller.admin.oauth2.vo.open.OAuth2OpenCheckTokenRespVO;
//import com.iailab.module.system.dal.dataobject.oauth2.OAuth2AccessTokenDO;
//import com.iailab.module.system.dal.dataobject.oauth2.OAuth2ApproveDO;
//import com.iailab.module.system.dal.dataobject.oauth2.OAuth2ClientDO;
//import com.iailab.module.system.enums.oauth2.OAuth2GrantTypeEnum;
//import com.iailab.module.system.service.oauth2.OAuth2ApproveService;
//import com.iailab.module.system.service.oauth2.OAuth2ClientService;
//import com.iailab.module.system.service.oauth2.OAuth2GrantService;
//import com.iailab.module.system.service.oauth2.OAuth2TokenService;
//import org.assertj.core.util.Lists;
//import org.junit.jupiter.api.Test;
//import org.mockito.InjectMocks;
//import org.mockito.Mock;
//
//import javax.servlet.http.HttpServletRequest;
//import java.time.LocalDateTime;
//import java.time.temporal.ChronoUnit;
//import java.util.ArrayList;
//import java.util.LinkedHashMap;
//import java.util.List;
//
//import static com.iailab.framework.common.util.collection.SetUtils.asSet;
//import static com.iailab.framework.test.core.util.AssertUtils.assertPojoEquals;
//import static com.iailab.framework.test.core.util.AssertUtils.assertServiceException;
//import static com.iailab.framework.test.core.util.RandomUtils.randomPojo;
//import static com.iailab.framework.test.core.util.RandomUtils.randomString;
//import static java.util.Arrays.asList;
//import static org.hamcrest.CoreMatchers.anyOf;
//import static org.hamcrest.CoreMatchers.is;
//import static org.hamcrest.MatcherAssert.assertThat;
//import static org.junit.jupiter.api.Assertions.*;
//import static org.mockito.ArgumentMatchers.eq;
//import static org.mockito.ArgumentMatchers.isNull;
//import static org.mockito.Mockito.mock;
//import static org.mockito.Mockito.when;
//
///**
// * {@link OAuth2OpenController} 的单元测试
// *
// * @author iailab
// */
//public class OAuth2OpenControllerTest extends BaseMockitoUnitTest {
//
//    @InjectMocks
//    private OAuth2OpenController oauth2OpenController;
//
//    @Mock
//    private OAuth2GrantService oauth2GrantService;
//    @Mock
//    private OAuth2ClientService oauth2ClientService;
//    @Mock
//    private OAuth2ApproveService oauth2ApproveService;
//    @Mock
//    private OAuth2TokenService oauth2TokenService;
//
//    @Test
//    public void testPostAccessToken_authorizationCode() {
//        // 准备参数
//        String granType = OAuth2GrantTypeEnum.AUTHORIZATION_CODE.getGrantType();
//        String code = randomString();
//        String redirectUri = randomString();
//        String state = randomString();
//        HttpServletRequest request = mockRequest("test_client_id", "test_client_secret");
//        // mock 方法（client）
//        OAuth2ClientDO client = randomPojo(OAuth2ClientDO.class).setClientId("test_client_id");
//        when(oauth2ClientService.validOAuthClientFromCache(eq("test_client_id"), eq("test_client_secret"), eq(granType), eq(new ArrayList<>()), eq(redirectUri))).thenReturn(client);
//
//        // mock 方法（访问令牌）
//        OAuth2AccessTokenDO accessTokenDO = randomPojo(OAuth2AccessTokenDO.class)
//                .setExpiresTime(LocalDateTimeUtil.offset(LocalDateTime.now(), 30000L, ChronoUnit.MILLIS));
//        when(oauth2GrantService.grantAuthorizationCodeForAccessToken(eq("test_client_id"),
//                eq(code), eq(redirectUri), eq(state))).thenReturn(accessTokenDO);
//
//        // 调用
//        CommonResult<OAuth2OpenAccessTokenRespVO> result = oauth2OpenController.postAccessToken(request, granType,
//                code, redirectUri, state, null, null, null, null);
//        // 断言
//        assertEquals(0, result.getCode());
//        assertPojoEquals(accessTokenDO, result.getData());
//        assertTrue(ObjectUtils.equalsAny(result.getData().getExpiresIn(), 29L, 30L));  // 执行过程会过去几毫秒
//    }
//
//    @Test
//    public void testPostAccessToken_password() {
//        // 准备参数
//        String granType = OAuth2GrantTypeEnum.PASSWORD.getGrantType();
//        String username = randomString();
//        String password = randomString();
//        String scope = "write read";
//        HttpServletRequest request = mockRequest("test_client_id", "test_client_secret");
//        // mock 方法（client）
//        OAuth2ClientDO client = randomPojo(OAuth2ClientDO.class).setClientId("test_client_id");
//        when(oauth2ClientService.validOAuthClientFromCache(eq("test_client_id"), eq("test_client_secret"),
//                eq(granType), eq(Lists.newArrayList("write", "read")), isNull())).thenReturn(client);
//
//        // mock 方法（访问令牌）
//        OAuth2AccessTokenDO accessTokenDO = randomPojo(OAuth2AccessTokenDO.class)
//                .setExpiresTime(LocalDateTimeUtil.offset(LocalDateTime.now(), 30000L, ChronoUnit.MILLIS));
//        when(oauth2GrantService.grantPassword(eq(username), eq(password), eq("test_client_id"),
//                eq(Lists.newArrayList("write", "read")))).thenReturn(accessTokenDO);
//
//        // 调用
//        CommonResult<OAuth2OpenAccessTokenRespVO> result = oauth2OpenController.postAccessToken(request, granType,
//                null, null, null, username, password, scope, null);
//        // 断言
//        assertEquals(0, result.getCode());
//        assertPojoEquals(accessTokenDO, result.getData());
//        assertTrue(ObjectUtils.equalsAny(result.getData().getExpiresIn(), 29L, 30L));  // 执行过程会过去几毫秒
//    }
//
//    @Test
//    public void testPostAccessToken_refreshToken() {
//        // 准备参数
//        String granType = OAuth2GrantTypeEnum.REFRESH_TOKEN.getGrantType();
//        String refreshToken = randomString();
//        String password = randomString();
//        HttpServletRequest request = mockRequest("test_client_id", "test_client_secret");
//        // mock 方法（client）
//        OAuth2ClientDO client = randomPojo(OAuth2ClientDO.class).setClientId("test_client_id");
//        when(oauth2ClientService.validOAuthClientFromCache(eq("test_client_id"), eq("test_client_secret"),
//                eq(granType), eq(Lists.newArrayList()), isNull())).thenReturn(client);
//
//        // mock 方法（访问令牌）
//        OAuth2AccessTokenDO accessTokenDO = randomPojo(OAuth2AccessTokenDO.class)
//                .setExpiresTime(LocalDateTimeUtil.offset(LocalDateTime.now(), 30000L, ChronoUnit.MILLIS));
//        when(oauth2GrantService.grantRefreshToken(eq(refreshToken), eq("test_client_id"))).thenReturn(accessTokenDO);
//
//        // 调用
//        CommonResult<OAuth2OpenAccessTokenRespVO> result = oauth2OpenController.postAccessToken(request, granType,
//                null, null, null, null, password, null, refreshToken);
//        // 断言
//        assertEquals(0, result.getCode());
//        assertPojoEquals(accessTokenDO, result.getData());
//        assertTrue(ObjectUtils.equalsAny(result.getData().getExpiresIn(), 29L, 30L));  // 执行过程会过去几毫秒
//    }
//
//    @Test
//    public void testPostAccessToken_implicit() {
//        // 调用，并断言
//        assertServiceException(() -> oauth2OpenController.postAccessToken(null,
//                        OAuth2GrantTypeEnum.IMPLICIT.getGrantType(), null, null, null,
//                        null, null, null, null),
//                new ErrorCode(400, "Token 接口不支持 implicit 授权模式"));
//    }
//
//    @Test
//    public void testRevokeToken() {
//        // 准备参数
//        HttpServletRequest request = mockRequest("demo_client_id", "demo_client_secret");
//        String token = randomString();
//        // mock 方法（client）
//        OAuth2ClientDO client = randomPojo(OAuth2ClientDO.class).setClientId("demo_client_id");
//        when(oauth2ClientService.validOAuthClientFromCache(eq("demo_client_id"),
//                eq("demo_client_secret"), isNull(), isNull(), isNull())).thenReturn(client);
//        // mock 方法（移除）
//        when(oauth2GrantService.revokeToken(eq("demo_client_id"), eq(token))).thenReturn(true);
//
//        // 调用
//        CommonResult<Boolean> result = oauth2OpenController.revokeToken(request, token);
//        // 断言
//        assertEquals(0, result.getCode());
//        assertTrue(result.getData());
//    }
//
//    @Test
//    public void testCheckToken() {
//        // 准备参数
//        HttpServletRequest request = mockRequest("demo_client_id", "demo_client_secret");
//        String token = randomString();
//        // mock 方法
//        OAuth2AccessTokenDO accessTokenDO = randomPojo(OAuth2AccessTokenDO.class).setUserType(UserTypeEnum.ADMIN.getValue()).setExpiresTime(LocalDateTimeUtil.of(1653485731195L));
//        when(oauth2TokenService.checkAccessToken(eq(token))).thenReturn(accessTokenDO);
//
//        // 调用
//        CommonResult<OAuth2OpenCheckTokenRespVO> result = oauth2OpenController.checkToken(request, token);
//        // 断言
//        assertEquals(0, result.getCode());
//        assertPojoEquals(accessTokenDO, result.getData());
//        assertEquals(1653485731L, result.getData().getExp()); // 执行过程会过去几毫秒
//    }
//
//    @Test
//    public void testAuthorize() {
//        // 准备参数
//        String clientId = randomString();
//        // mock 方法（client）
//        OAuth2ClientDO client = randomPojo(OAuth2ClientDO.class).setClientId("demo_client_id").setScopes(ListUtil.toList("read", "write", "all"));
//        when(oauth2ClientService.validOAuthClientFromCache(eq(clientId))).thenReturn(client);
//        // mock 方法（approve）
//        List<OAuth2ApproveDO> approves = asList(
//                randomPojo(OAuth2ApproveDO.class).setScope("read").setApproved(true),
//                randomPojo(OAuth2ApproveDO.class).setScope("write").setApproved(false));
//        when(oauth2ApproveService.getApproveList(isNull(), eq(UserTypeEnum.ADMIN.getValue()), eq(clientId))).thenReturn(approves);
//
//        // 调用
//        CommonResult<OAuth2OpenAuthorizeInfoRespVO> result = oauth2OpenController.authorize(clientId);
//        // 断言
//        assertEquals(0, result.getCode());
//        assertPojoEquals(client, result.getData().getClient());
//        assertEquals(new KeyValue<>("read", true), result.getData().getScopes().get(0));
//        assertEquals(new KeyValue<>("write", false), result.getData().getScopes().get(1));
//        assertEquals(new KeyValue<>("all", false), result.getData().getScopes().get(2));
//    }
//
//    @Test
//    public void testApproveOrDeny_grantTypeError() {
//        // 调用，并断言
//        assertServiceException(() -> oauth2OpenController.approveOrDeny(randomString(), null,
//                        null, null, null, null),
//                new ErrorCode(400, "response_type 参数值只允许 code 和 token"));
//    }
//
//    @Test // autoApprove = true，但是不通过
//    public void testApproveOrDeny_autoApproveNo() {
//        // 准备参数
//        String responseType = "code";
//        String clientId = randomString();
//        String scope = "{\"read\": true, \"write\": false}";
//        String redirectUri = randomString();
//        String state = randomString();
//        // mock 方法
//        OAuth2ClientDO client = randomPojo(OAuth2ClientDO.class);
//        when(oauth2ClientService.validOAuthClientFromCache(eq(clientId), isNull(), eq("authorization_code"),
//                eq(asSet("read", "write")), eq(redirectUri))).thenReturn(client);
//
//        // 调用
//        CommonResult<String> result = oauth2OpenController.approveOrDeny(responseType, clientId,
//                scope, redirectUri, true, state);
//        // 断言
//        assertEquals(0, result.getCode());
//        assertNull(result.getData());
//    }
//
//    @Test // autoApprove = false，但是不通过
//    public void testApproveOrDeny_ApproveNo() {
//        // 准备参数
//        String responseType = "token";
//        String clientId = randomString();
//        String scope = "{\"read\": true, \"write\": false}";
//        String redirectUri = "https://www.baidu.com";
//        String state = "test";
//        // mock 方法
//        OAuth2ClientDO client = randomPojo(OAuth2ClientDO.class);
//        when(oauth2ClientService.validOAuthClientFromCache(eq(clientId), isNull(), eq("implicit"),
//                eq(asSet("read", "write")), eq(redirectUri))).thenReturn(client);
//
//        // 调用
//        CommonResult<String> result = oauth2OpenController.approveOrDeny(responseType, clientId,
//                scope, redirectUri, false, state);
//        // 断言
//        assertEquals(0, result.getCode());
//        assertEquals("https://www.baidu.com#error=access_denied&error_description=User%20denied%20access&state=test", result.getData());
//    }
//
//    @Test // autoApprove = true，通过 + token
//    public void testApproveOrDeny_autoApproveWithToken() {
//        // 准备参数
//        String responseType = "token";
//        String clientId = randomString();
//        String scope = "{\"read\": true, \"write\": false}";
//        String redirectUri = "https://www.baidu.com";
//        String state = "test";
//        // mock 方法（client)
//        OAuth2ClientDO client = randomPojo(OAuth2ClientDO.class).setClientId(clientId).setAdditionalInformation(null);
//        when(oauth2ClientService.validOAuthClientFromCache(eq(clientId), isNull(), eq("implicit"),
//                eq(asSet("read", "write")), eq(redirectUri))).thenReturn(client);
//        // mock 方法（场景一）
//        when(oauth2ApproveService.checkForPreApproval(isNull(), eq(UserTypeEnum.ADMIN.getValue()),
//                eq(clientId), eq(SetUtils.asSet("read", "write")))).thenReturn(true);
//        // mock 方法（访问令牌）
//        OAuth2AccessTokenDO accessTokenDO = randomPojo(OAuth2AccessTokenDO.class)
//                .setAccessToken("test_access_token").setExpiresTime(LocalDateTimeUtil.offset(LocalDateTime.now(), 30010L, ChronoUnit.MILLIS));
//        when(oauth2GrantService.grantImplicit(isNull(), eq(UserTypeEnum.ADMIN.getValue()),
//                eq(clientId), eq(ListUtil.toList("read")))).thenReturn(accessTokenDO);
//
//        // 调用
//        CommonResult<String> result = oauth2OpenController.approveOrDeny(responseType, clientId,
//                scope, redirectUri, true, state);
//        // 断言
//        assertEquals(0, result.getCode());
//        assertThat(result.getData(), anyOf( // 29 和 30 都有一定概率，主要是时间计算
//                is("https://www.baidu.com#access_token=test_access_token&token_type=bearer&state=test&expires_in=29&scope=read"),
//                is("https://www.baidu.com#access_token=test_access_token&token_type=bearer&state=test&expires_in=30&scope=read")
//        ));
//    }
//
//    @Test // autoApprove = false，通过 + code
//    public void testApproveOrDeny_approveWithCode() {
//        // 准备参数
//        String responseType = "code";
//        String clientId = randomString();
//        String scope = "{\"read\": true, \"write\": false}";
//        String redirectUri = "https://www.baidu.com";
//        String state = "test";
//        // mock 方法（client)
//        OAuth2ClientDO client = randomPojo(OAuth2ClientDO.class).setClientId(clientId).setAdditionalInformation(null);
//        when(oauth2ClientService.validOAuthClientFromCache(eq(clientId), isNull(), eq("authorization_code"),
//                eq(asSet("read", "write")), eq(redirectUri))).thenReturn(client);
//        // mock 方法（场景二）
//        when(oauth2ApproveService.updateAfterApproval(isNull(), eq(UserTypeEnum.ADMIN.getValue()), eq(clientId),
//                eq(MapUtil.builder(new LinkedHashMap<String, Boolean>()).put("read", true).put("write", false).build())))
//                .thenReturn(true);
//        // mock 方法（访问令牌）
//        String authorizationCode = "test_code";
//        when(oauth2GrantService.grantAuthorizationCodeForCode(isNull(), eq(UserTypeEnum.ADMIN.getValue()),
//                eq(clientId), eq(ListUtil.toList("read")), eq(redirectUri), eq(state))).thenReturn(authorizationCode);
//
//        // 调用
//        CommonResult<String> result = oauth2OpenController.approveOrDeny(responseType, clientId,
//                scope, redirectUri, false, state);
//        // 断言
//        assertEquals(0, result.getCode());
//        assertEquals("https://www.baidu.com?code=test_code&state=test", result.getData());
//    }
//
//    private HttpServletRequest mockRequest(String clientId, String secret) {
//        HttpServletRequest request = mock(HttpServletRequest.class);
//        when(request.getParameter(eq("client_id"))).thenReturn(clientId);
//        when(request.getParameter(eq("client_secret"))).thenReturn(secret);
//        return request;
//    }
//
//}