seanyu
2 years ago
parent
bd1297b07e
commit
2411fd1c82
@ -0,0 +1,155 @@
|
|||||||
|
/*
|
||||||
|
* Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
|
||||||
|
*
|
||||||
|
* Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
|
||||||
|
*
|
||||||
|
* Licensed under the BSD 3-Clause License (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* https://opensource.org/licenses/BSD-3-Clause
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software distributed
|
||||||
|
* under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
|
||||||
|
* CONDITIONS OF ANY KIND, either express or implied. See the License for the
|
||||||
|
* specific language governing permissions and limitations under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package com.tencent.cloud.rpc.enhancement.scg;
|
||||||
|
|
||||||
|
import java.net.URI;
|
||||||
|
import java.net.URISyntaxException;
|
||||||
|
|
||||||
|
import com.tencent.cloud.common.metadata.MetadataContext;
|
||||||
|
import com.tencent.cloud.common.metadata.StaticMetadataManager;
|
||||||
|
import com.tencent.cloud.common.metadata.config.MetadataLocalProperties;
|
||||||
|
import com.tencent.cloud.common.util.ApplicationContextAwareUtils;
|
||||||
|
import com.tencent.cloud.rpc.enhancement.config.RpcEnhancementReporterProperties;
|
||||||
|
import com.tencent.polaris.api.config.Configuration;
|
||||||
|
import com.tencent.polaris.api.config.global.APIConfig;
|
||||||
|
import com.tencent.polaris.api.config.global.GlobalConfig;
|
||||||
|
import com.tencent.polaris.api.core.ConsumerAPI;
|
||||||
|
import com.tencent.polaris.circuitbreak.api.CircuitBreakAPI;
|
||||||
|
import com.tencent.polaris.client.api.SDKContext;
|
||||||
|
import org.junit.jupiter.api.AfterAll;
|
||||||
|
import org.junit.jupiter.api.BeforeAll;
|
||||||
|
import org.junit.jupiter.api.BeforeEach;
|
||||||
|
import org.junit.jupiter.api.Test;
|
||||||
|
import org.junit.jupiter.api.extension.ExtendWith;
|
||||||
|
import org.mockito.Mock;
|
||||||
|
import org.mockito.MockedStatic;
|
||||||
|
import org.mockito.Mockito;
|
||||||
|
import org.mockito.junit.jupiter.MockitoExtension;
|
||||||
|
import reactor.core.publisher.Mono;
|
||||||
|
|
||||||
|
import org.springframework.cloud.client.ServiceInstance;
|
||||||
|
import org.springframework.cloud.client.loadbalancer.Response;
|
||||||
|
import org.springframework.cloud.gateway.filter.GatewayFilterChain;
|
||||||
|
import org.springframework.context.ApplicationContext;
|
||||||
|
import org.springframework.http.HttpHeaders;
|
||||||
|
import org.springframework.http.HttpMethod;
|
||||||
|
import org.springframework.http.server.reactive.ServerHttpRequest;
|
||||||
|
import org.springframework.http.server.reactive.ServerHttpResponse;
|
||||||
|
import org.springframework.web.server.ServerWebExchange;
|
||||||
|
|
||||||
|
import static com.tencent.polaris.test.common.Consts.NAMESPACE_TEST;
|
||||||
|
import static com.tencent.polaris.test.common.Consts.SERVICE_PROVIDER;
|
||||||
|
import static org.mockito.ArgumentMatchers.anyString;
|
||||||
|
import static org.mockito.Mockito.doReturn;
|
||||||
|
import static org.mockito.Mockito.mock;
|
||||||
|
import static org.springframework.cloud.gateway.support.ServerWebExchangeUtils.GATEWAY_LOADBALANCER_RESPONSE_ATTR;
|
||||||
|
|
||||||
|
@ExtendWith(MockitoExtension.class)
|
||||||
|
public class EnhancedPolarisGatewayReporterTest {
|
||||||
|
|
||||||
|
private static MockedStatic<ApplicationContextAwareUtils> mockedApplicationContextAwareUtils;
|
||||||
|
@Mock
|
||||||
|
private RpcEnhancementReporterProperties reporterProperties;
|
||||||
|
@Mock
|
||||||
|
private SDKContext sdkContext;
|
||||||
|
@Mock
|
||||||
|
private ConsumerAPI consumerAPI;
|
||||||
|
@Mock
|
||||||
|
private CircuitBreakAPI circuitBreakAPI;
|
||||||
|
@Mock
|
||||||
|
ServerWebExchange exchange;
|
||||||
|
@Mock
|
||||||
|
GatewayFilterChain chain;
|
||||||
|
@Mock
|
||||||
|
ServerHttpResponse response;
|
||||||
|
@Mock
|
||||||
|
ServerHttpRequest request;
|
||||||
|
|
||||||
|
@BeforeAll
|
||||||
|
static void beforeAll() {
|
||||||
|
mockedApplicationContextAwareUtils = Mockito.mockStatic(ApplicationContextAwareUtils.class);
|
||||||
|
mockedApplicationContextAwareUtils.when(() -> ApplicationContextAwareUtils.getProperties(anyString()))
|
||||||
|
.thenReturn("unit-test");
|
||||||
|
ApplicationContext applicationContext = mock(ApplicationContext.class);
|
||||||
|
MetadataLocalProperties metadataLocalProperties = mock(MetadataLocalProperties.class);
|
||||||
|
StaticMetadataManager staticMetadataManager = mock(StaticMetadataManager.class);
|
||||||
|
doReturn(metadataLocalProperties).when(applicationContext).getBean(MetadataLocalProperties.class);
|
||||||
|
doReturn(staticMetadataManager).when(applicationContext).getBean(StaticMetadataManager.class);
|
||||||
|
mockedApplicationContextAwareUtils.when(ApplicationContextAwareUtils::getApplicationContext).thenReturn(applicationContext);
|
||||||
|
}
|
||||||
|
|
||||||
|
@AfterAll
|
||||||
|
static void afterAll() {
|
||||||
|
mockedApplicationContextAwareUtils.close();
|
||||||
|
}
|
||||||
|
|
||||||
|
@BeforeEach
|
||||||
|
void setUp() {
|
||||||
|
MetadataContext.LOCAL_NAMESPACE = NAMESPACE_TEST;
|
||||||
|
MetadataContext.LOCAL_SERVICE = SERVICE_PROVIDER;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testRun() throws URISyntaxException {
|
||||||
|
APIConfig apiConfig = mock(APIConfig.class);
|
||||||
|
doReturn("0.0.0.0").when(apiConfig).getBindIP();
|
||||||
|
|
||||||
|
GlobalConfig globalConfig = mock(GlobalConfig.class);
|
||||||
|
doReturn(apiConfig).when(globalConfig).getAPI();
|
||||||
|
|
||||||
|
Configuration configuration = mock(Configuration.class);
|
||||||
|
doReturn(globalConfig).when(configuration).getGlobal();
|
||||||
|
|
||||||
|
doReturn(configuration).when(sdkContext).getConfig();
|
||||||
|
|
||||||
|
doReturn(new URI("http://0.0.0.0/")).when(request).getURI();
|
||||||
|
doReturn(new HttpHeaders()).when(request).getHeaders();
|
||||||
|
doReturn(HttpMethod.GET).when(request).getMethod();
|
||||||
|
doReturn(new HttpHeaders()).when(response).getHeaders();
|
||||||
|
doReturn(Mono.empty()).when(chain).filter(exchange);
|
||||||
|
|
||||||
|
ServiceInstance serviceInstance = mock(ServiceInstance.class);
|
||||||
|
doReturn("test").when(serviceInstance).getServiceId();
|
||||||
|
doReturn("0.0.0.0").when(serviceInstance).getHost();
|
||||||
|
doReturn(80).when(serviceInstance).getPort();
|
||||||
|
Response<ServiceInstance> serviceInstanceResponse = new Response<ServiceInstance>() {
|
||||||
|
@Override
|
||||||
|
public boolean hasServer() {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public ServiceInstance getServer() {
|
||||||
|
return serviceInstance;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
doReturn(serviceInstanceResponse).when(exchange).getAttribute(GATEWAY_LOADBALANCER_RESPONSE_ATTR);
|
||||||
|
doReturn(request).when(exchange).getRequest();
|
||||||
|
doReturn(response).when(exchange).getResponse();
|
||||||
|
|
||||||
|
EnhancedPolarisGatewayReporter reporter = new EnhancedPolarisGatewayReporter(reporterProperties, sdkContext, consumerAPI, circuitBreakAPI);
|
||||||
|
reporter.getOrder();
|
||||||
|
|
||||||
|
reporter.filter(exchange, chain).block();
|
||||||
|
|
||||||
|
doReturn(true).when(reporterProperties).isEnabled();
|
||||||
|
|
||||||
|
reporter.filter(exchange, chain).block();
|
||||||
|
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -1,49 +0,0 @@
|
|||||||
///*
|
|
||||||
// * Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
|
|
||||||
// *
|
|
||||||
// * Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
|
|
||||||
// *
|
|
||||||
// * Licensed under the BSD 3-Clause License (the "License");
|
|
||||||
// * you may not use this file except in compliance with the License.
|
|
||||||
// * You may obtain a copy of the License at
|
|
||||||
// *
|
|
||||||
// * https://opensource.org/licenses/BSD-3-Clause
|
|
||||||
// *
|
|
||||||
// * Unless required by applicable law or agreed to in writing, software distributed
|
|
||||||
// * under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
|
|
||||||
// * CONDITIONS OF ANY KIND, either express or implied. See the License for the
|
|
||||||
// * specific language governing permissions and limitations under the License.
|
|
||||||
// */
|
|
||||||
//
|
|
||||||
//package com.tencent.cloud.rpc.enhancement.scg;
|
|
||||||
//
|
|
||||||
//import com.tencent.cloud.rpc.enhancement.config.RpcEnhancementReporterProperties;
|
|
||||||
//import com.tencent.polaris.api.core.ConsumerAPI;
|
|
||||||
//import com.tencent.polaris.client.api.SDKContext;
|
|
||||||
//import org.junit.jupiter.api.Assertions;
|
|
||||||
//import org.junit.jupiter.api.Test;
|
|
||||||
//import org.mockito.Mockito;
|
|
||||||
//import reactor.netty.http.client.HttpClient;
|
|
||||||
//
|
|
||||||
//public class EnhancedPolarisHttpClientCustomizerTest {
|
|
||||||
//
|
|
||||||
// @Test
|
|
||||||
// public void testCustomize() {
|
|
||||||
// RpcEnhancementReporterProperties properties = new RpcEnhancementReporterProperties();
|
|
||||||
// properties.setEnabled(true);
|
|
||||||
// properties.getStatuses().clear();
|
|
||||||
// properties.getSeries().clear();
|
|
||||||
//
|
|
||||||
// SDKContext context = Mockito.mock(SDKContext.class);
|
|
||||||
// ConsumerAPI consumerAPI = Mockito.mock(ConsumerAPI.class);
|
|
||||||
//
|
|
||||||
// EnhancedPolarisHttpClientCustomizer clientCustomizer = new EnhancedPolarisHttpClientCustomizer(properties, context, consumerAPI);
|
|
||||||
//
|
|
||||||
// HttpClient client = HttpClient.create();
|
|
||||||
// HttpClient proxyClient = clientCustomizer.customize(client);
|
|
||||||
//
|
|
||||||
// Assertions.assertNotNull(proxyClient);
|
|
||||||
// Assertions.assertEquals(EnhancedPolarisHttpClient.class.getName(), proxyClient.getClass().getName());
|
|
||||||
// }
|
|
||||||
//
|
|
||||||
//}
|
|
||||||
@ -1,52 +0,0 @@
|
|||||||
///*
|
|
||||||
// * Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
|
|
||||||
// *
|
|
||||||
// * Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
|
|
||||||
// *
|
|
||||||
// * Licensed under the BSD 3-Clause License (the "License");
|
|
||||||
// * you may not use this file except in compliance with the License.
|
|
||||||
// * You may obtain a copy of the License at
|
|
||||||
// *
|
|
||||||
// * https://opensource.org/licenses/BSD-3-Clause
|
|
||||||
// *
|
|
||||||
// * Unless required by applicable law or agreed to in writing, software distributed
|
|
||||||
// * under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
|
|
||||||
// * CONDITIONS OF ANY KIND, either express or implied. See the License for the
|
|
||||||
// * specific language governing permissions and limitations under the License.
|
|
||||||
// */
|
|
||||||
//
|
|
||||||
//package com.tencent.cloud.rpc.enhancement.scg;
|
|
||||||
//
|
|
||||||
//import java.util.Collections;
|
|
||||||
//
|
|
||||||
//import com.tencent.cloud.common.constant.HeaderConstant;
|
|
||||||
//import org.junit.jupiter.api.Assertions;
|
|
||||||
//import org.junit.jupiter.api.Test;
|
|
||||||
//import org.mockito.Mockito;
|
|
||||||
//
|
|
||||||
//import org.springframework.cloud.client.ServiceInstance;
|
|
||||||
//import org.springframework.cloud.client.loadbalancer.DefaultResponse;
|
|
||||||
//import org.springframework.http.HttpHeaders;
|
|
||||||
//import org.springframework.web.server.ServerWebExchange;
|
|
||||||
//
|
|
||||||
//import static org.springframework.cloud.gateway.support.ServerWebExchangeUtils.GATEWAY_LOADBALANCER_RESPONSE_ATTR;
|
|
||||||
//
|
|
||||||
//public class EnhancedPolarisHttpHeadersFilterTest {
|
|
||||||
//
|
|
||||||
// @Test
|
|
||||||
// public void testFilter() {
|
|
||||||
// EnhancedPolarisHttpHeadersFilter filter = new EnhancedPolarisHttpHeadersFilter();
|
|
||||||
// ServiceInstance instance = Mockito.mock(ServiceInstance.class);
|
|
||||||
// Mockito.doReturn("mock_service").when(instance).getServiceId();
|
|
||||||
// Mockito.doReturn("127.0.0.1").when(instance).getHost();
|
|
||||||
// Mockito.doReturn(8080).when(instance).getPort();
|
|
||||||
// DefaultResponse response = new DefaultResponse(instance);
|
|
||||||
// ServerWebExchange exchange = Mockito.mock(ServerWebExchange.class);
|
|
||||||
// Mockito.doReturn(response).when(exchange).getAttribute(GATEWAY_LOADBALANCER_RESPONSE_ATTR);
|
|
||||||
// HttpHeaders input = new HttpHeaders();
|
|
||||||
// HttpHeaders headers = filter.filter(input, exchange);
|
|
||||||
// Assertions.assertEquals(Collections.singletonList("mock_service"), headers.get(HeaderConstant.INTERNAL_CALLEE_SERVICE_ID));
|
|
||||||
// Assertions.assertEquals(Collections.singletonList("127.0.0.1"), headers.get(HeaderConstant.INTERNAL_CALLEE_INSTANCE_HOST));
|
|
||||||
// Assertions.assertEquals(Collections.singletonList("8080"), headers.get(HeaderConstant.INTERNAL_CALLEE_INSTANCE_PORT));
|
|
||||||
// }
|
|
||||||
//}
|
|
||||||
Loading…
Reference in new issue