✨ Introducing new features. close #3498 @WithMockOAuth2User注释在4.2版本下报错

This commit is contained in:
lbw committed 2021-12-29 12:33:00 +08:00
1 parent 3790f7d583
commit 86472c9a62
1 file changed
+12 -6
@@ -1,14 +1,11 @@
package com.pig4cloud.pigx.test;
import java.nio.charset.StandardCharsets;
import cn.hutool.core.util.StrUtil;
import cn.hutool.http.HttpUtil;
import cn.hutool.json.JSONUtil;
import com.pig4cloud.pigx.common.core.util.SpringContextHolder;
import com.pig4cloud.pigx.common.data.tenant.TenantContextHolder;
import com.pig4cloud.pigx.test.annotation.WithMockOAuth2User;
import org.springframework.http.HttpEntity;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpMethod;
@@ -24,6 +21,8 @@ import org.springframework.util.LinkedMultiValueMap;
import org.springframework.util.MultiValueMap;
import org.springframework.web.client.RestTemplate;
import java.nio.charset.StandardCharsets;
/**
* @author lengleng
* @date 2020/9/22
@@ -80,9 +79,16 @@ public class WithMockSecurityContextFactory implements WithSecurityContextFactor
RestTemplate restTemplate = SpringContextHolder.getBean(RestTemplate.class);
// 优先获取配置文件URI 配置
String url = StrUtil.isBlank(clientProperties.getAccessTokenUri()) ? TOKEN_URL
: clientProperties.getAccessTokenUri();
String result = restTemplate.exchange(url, HttpMethod.POST, requestEntity, String.class).getBody();
String result;
if (StrUtil.isBlank(clientProperties.getAccessTokenUri())) {
result = restTemplate.exchange(TOKEN_URL, HttpMethod.POST, requestEntity, String.class).getBody();
}
else {
result = new RestTemplate()
.exchange(clientProperties.getAccessTokenUri(), HttpMethod.POST, requestEntity, String.class)
.getBody();
}
return JSONUtil.parseObj(result).getStr("access_token");
}