🎨 #4047 【企业微信】通讯录同步相关接口补齐“独立 secret + 独立 access token”的调用通道
This commit is contained in:
@@ -106,6 +106,11 @@ public class BaseWxCpServiceImplTest {
|
||||
return "mock_msg_audit_access_token";
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getContactAccessToken(boolean forceRefresh) throws WxErrorException {
|
||||
return "mock_contact_access_token";
|
||||
}
|
||||
|
||||
@Override
|
||||
public void initHttp() {
|
||||
|
||||
|
||||
@@ -0,0 +1,264 @@
|
||||
package me.chanjar.weixin.cp.api.impl;
|
||||
|
||||
import me.chanjar.weixin.common.error.WxErrorException;
|
||||
import me.chanjar.weixin.common.util.http.HttpClientType;
|
||||
import me.chanjar.weixin.cp.config.WxCpConfigStorage;
|
||||
import me.chanjar.weixin.cp.config.impl.WxCpDefaultConfigImpl;
|
||||
import org.testng.annotations.BeforeMethod;
|
||||
import org.testng.annotations.Test;
|
||||
|
||||
import java.util.concurrent.locks.Lock;
|
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat;
|
||||
import static org.assertj.core.api.Assertions.assertThatThrownBy;
|
||||
|
||||
/**
|
||||
* 测试 getContactAccessToken 方法在各个实现类中的正确性
|
||||
*
|
||||
* @author Binary Wang
|
||||
*/
|
||||
@Test
|
||||
public class WxCpServiceGetContactAccessTokenTest {
|
||||
|
||||
private WxCpDefaultConfigImpl config;
|
||||
|
||||
@BeforeMethod
|
||||
public void setUp() {
|
||||
config = new WxCpDefaultConfigImpl();
|
||||
config.setCorpId("testCorpId");
|
||||
config.setCorpSecret("testCorpSecret");
|
||||
config.setContactSecret("testContactSecret");
|
||||
}
|
||||
|
||||
/**
|
||||
* 测试通讯录同步access token的缓存机制
|
||||
* 验证当token未过期时,直接从配置中返回缓存的token
|
||||
*/
|
||||
@Test
|
||||
public void testGetContactAccessToken_Cache() throws WxErrorException {
|
||||
// 预先设置一个有效的token
|
||||
config.updateContactAccessToken("cached_token", 7200);
|
||||
|
||||
BaseWxCpServiceImpl service = createTestService(config);
|
||||
|
||||
// 不强制刷新时应该返回缓存的token
|
||||
String token = service.getContactAccessToken(false);
|
||||
assertThat(token).isEqualTo("cached_token");
|
||||
}
|
||||
|
||||
/**
|
||||
* 测试强制刷新通讯录同步access token
|
||||
* 验证forceRefresh=true时会重新获取token
|
||||
*/
|
||||
@Test
|
||||
public void testGetContactAccessToken_ForceRefresh() throws WxErrorException {
|
||||
// 预先设置一个有效的token
|
||||
config.updateContactAccessToken("old_token", 7200);
|
||||
|
||||
BaseWxCpServiceImpl service = createTestServiceWithMockToken(config, "new_token");
|
||||
|
||||
// 强制刷新应该获取新token
|
||||
String token = service.getContactAccessToken(true);
|
||||
assertThat(token).isEqualTo("new_token");
|
||||
}
|
||||
|
||||
/**
|
||||
* 测试token过期时自动刷新
|
||||
* 验证当token已过期时,会自动重新获取
|
||||
*/
|
||||
@Test
|
||||
public void testGetContactAccessToken_Expired() throws WxErrorException {
|
||||
// 设置一个已过期的token(过期时间为负数,确保立即过期)
|
||||
config.updateContactAccessToken("expired_token", -1);
|
||||
|
||||
BaseWxCpServiceImpl service = createTestServiceWithMockToken(config, "refreshed_token");
|
||||
|
||||
// 过期的token应该被自动刷新
|
||||
String token = service.getContactAccessToken(false);
|
||||
assertThat(token).isEqualTo("refreshed_token");
|
||||
}
|
||||
|
||||
/**
|
||||
* 测试获取锁机制
|
||||
* 验证配置中的锁可以正常获取和使用
|
||||
*/
|
||||
@Test
|
||||
public void testGetContactAccessToken_Lock() {
|
||||
// 验证配置提供的锁不为null
|
||||
assertThat(config.getContactAccessTokenLock()).isNotNull();
|
||||
|
||||
// 验证锁可以正常使用
|
||||
config.getContactAccessTokenLock().lock();
|
||||
try {
|
||||
assertThat(config.getContactAccessToken()).isNull();
|
||||
} finally {
|
||||
config.getContactAccessTokenLock().unlock();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 检查token是否需要刷新的公共逻辑
|
||||
*/
|
||||
private boolean shouldRefreshToken(WxCpConfigStorage storage, boolean forceRefresh) {
|
||||
return storage.isContactAccessTokenExpired() || forceRefresh;
|
||||
}
|
||||
|
||||
/**
|
||||
* 验证通讯录同步secret是否已配置的公共逻辑
|
||||
*/
|
||||
private void validateContactSecret(String contactSecret) throws WxErrorException {
|
||||
if (contactSecret == null || contactSecret.trim().isEmpty()) {
|
||||
throw new WxErrorException("通讯录同步secret未配置");
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建一个用于测试的BaseWxCpServiceImpl实现,
|
||||
* 用于测试缓存和过期逻辑
|
||||
*/
|
||||
private BaseWxCpServiceImpl createTestService(WxCpConfigStorage config) {
|
||||
return new BaseWxCpServiceImpl() {
|
||||
@Override
|
||||
public Object getRequestHttpClient() {
|
||||
return null;
|
||||
}
|
||||
|
||||
@Override
|
||||
public Object getRequestHttpProxy() {
|
||||
return null;
|
||||
}
|
||||
|
||||
@Override
|
||||
public HttpClientType getRequestType() {
|
||||
return null;
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getAccessToken(boolean forceRefresh) throws WxErrorException {
|
||||
return "test_access_token";
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getContactAccessToken(boolean forceRefresh) throws WxErrorException {
|
||||
// 检查是否需要刷新
|
||||
if (!shouldRefreshToken(getWxCpConfigStorage(), forceRefresh)) {
|
||||
return getWxCpConfigStorage().getContactAccessToken();
|
||||
}
|
||||
|
||||
// 使用通讯录同步secret获取access_token
|
||||
String contactSecret = getWxCpConfigStorage().getContactSecret();
|
||||
validateContactSecret(contactSecret);
|
||||
|
||||
// 返回缓存的token(用于测试缓存机制)
|
||||
return getWxCpConfigStorage().getContactAccessToken();
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getMsgAuditAccessToken(boolean forceRefresh) throws WxErrorException {
|
||||
return "test_msg_audit_token";
|
||||
}
|
||||
|
||||
@Override
|
||||
public void initHttp() {
|
||||
}
|
||||
|
||||
@Override
|
||||
public WxCpConfigStorage getWxCpConfigStorage() {
|
||||
return config;
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建一个用于测试的BaseWxCpServiceImpl实现,
|
||||
* 模拟返回指定的token(用于测试刷新逻辑)
|
||||
*/
|
||||
private BaseWxCpServiceImpl createTestServiceWithMockToken(WxCpConfigStorage config, String mockToken) {
|
||||
return new BaseWxCpServiceImpl() {
|
||||
@Override
|
||||
public Object getRequestHttpClient() {
|
||||
return null;
|
||||
}
|
||||
|
||||
@Override
|
||||
public Object getRequestHttpProxy() {
|
||||
return null;
|
||||
}
|
||||
|
||||
@Override
|
||||
public HttpClientType getRequestType() {
|
||||
return null;
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getAccessToken(boolean forceRefresh) throws WxErrorException {
|
||||
return "test_access_token";
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getContactAccessToken(boolean forceRefresh) throws WxErrorException {
|
||||
// 使用锁机制
|
||||
Lock lock = getWxCpConfigStorage().getContactAccessTokenLock();
|
||||
lock.lock();
|
||||
try {
|
||||
// 检查是否需要刷新
|
||||
if (!shouldRefreshToken(getWxCpConfigStorage(), forceRefresh)) {
|
||||
return getWxCpConfigStorage().getContactAccessToken();
|
||||
}
|
||||
|
||||
// 使用通讯录同步secret获取access_token
|
||||
String contactSecret = getWxCpConfigStorage().getContactSecret();
|
||||
validateContactSecret(contactSecret);
|
||||
|
||||
// 模拟获取新token并更新配置
|
||||
getWxCpConfigStorage().updateContactAccessToken(mockToken, 7200);
|
||||
return mockToken;
|
||||
} finally {
|
||||
lock.unlock();
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getMsgAuditAccessToken(boolean forceRefresh) throws WxErrorException {
|
||||
return "test_msg_audit_token";
|
||||
}
|
||||
|
||||
@Override
|
||||
public void initHttp() {
|
||||
}
|
||||
|
||||
@Override
|
||||
public WxCpConfigStorage getWxCpConfigStorage() {
|
||||
return config;
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* 测试当 ContactSecret 未配置时应该抛出异常
|
||||
*/
|
||||
@Test
|
||||
public void testGetContactAccessToken_WithoutSecret() {
|
||||
config.setContactSecret(null);
|
||||
BaseWxCpServiceImpl service = createTestService(config);
|
||||
|
||||
// 验证当 secret 为 null 时抛出异常
|
||||
assertThatThrownBy(() -> service.getContactAccessToken(true))
|
||||
.isInstanceOf(WxErrorException.class)
|
||||
.hasMessageContaining("通讯录同步secret未配置");
|
||||
}
|
||||
|
||||
/**
|
||||
* 测试当 ContactSecret 为空字符串时应该抛出异常
|
||||
*/
|
||||
@Test
|
||||
public void testGetContactAccessToken_WithEmptySecret() {
|
||||
config.setContactSecret(" ");
|
||||
BaseWxCpServiceImpl service = createTestService(config);
|
||||
|
||||
// 验证当 secret 为空字符串时抛出异常
|
||||
assertThatThrownBy(() -> service.getContactAccessToken(true))
|
||||
.isInstanceOf(WxErrorException.class)
|
||||
.hasMessageContaining("通讯录同步secret未配置");
|
||||
}
|
||||
}
|
||||
@@ -153,6 +153,11 @@ public class WxCpServiceGetMsgAuditAccessTokenTest {
|
||||
return getWxCpConfigStorage().getMsgAuditAccessToken();
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getContactAccessToken(boolean forceRefresh) throws WxErrorException {
|
||||
return "mock_contact_access_token";
|
||||
}
|
||||
|
||||
@Override
|
||||
public void initHttp() {
|
||||
}
|
||||
@@ -213,6 +218,11 @@ public class WxCpServiceGetMsgAuditAccessTokenTest {
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getContactAccessToken(boolean forceRefresh) throws WxErrorException {
|
||||
return "mock_contact_access_token";
|
||||
}
|
||||
|
||||
@Override
|
||||
public void initHttp() {
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user