add test

spring-cloud-starter-tencent-polaris-circuitbreaker/src/main/java/com/tencent/cloud/polaris/circuitbreaker/PolarisRestTemplateResponseErrorHandler.java

@@ -74,7 +74,7 @@ public class PolarisRestTemplateResponseErrorHandler implements ResponseErrorHan
 	public void handleError(URI url, HttpMethod method, ClientHttpResponse response) {
 		ServiceCallResult resultRequest = null;
 		try {
-			resultRequest = builderServiceCallResult(url, method, response);
+			resultRequest = builderServiceCallResult(url, response);
 		} catch (IOException e) {
 			LOG.error("Will report response of {} url {}", response, url, e);
 			throw new RuntimeException(e);
@@ -83,12 +83,12 @@ public class PolarisRestTemplateResponseErrorHandler implements ResponseErrorHan
 		}
 	}
 
-	private ServiceCallResult builderServiceCallResult(URI url, HttpMethod method, ClientHttpResponse response) throws IOException {
+	private ServiceCallResult builderServiceCallResult(URI uri, ClientHttpResponse response) throws IOException {
 		ServiceCallResult resultRequest = new ServiceCallResult();
-		String serviceName = url.getHost();
+		String serviceName = uri.getHost();
 		resultRequest.setService(serviceName);
 		resultRequest.setNamespace(MetadataContext.LOCAL_NAMESPACE);
-		resultRequest.setMethod(url.getPath());
+		resultRequest.setMethod(uri.getPath());
 		resultRequest.setRetStatus(RetStatus.RetSuccess);
 		String sourceNamespace = MetadataContext.LOCAL_NAMESPACE;
 		String sourceService = MetadataContext.LOCAL_SERVICE;
@@ -96,9 +96,9 @@ public class PolarisRestTemplateResponseErrorHandler implements ResponseErrorHan
 			resultRequest.setCallerService(new ServiceKey(sourceNamespace, sourceService));
 		}
 		HttpURLConnection connection = (HttpURLConnection) ReflectionUtils.getFieldValue(response, FileName);
-		URL url1 = connection.getURL();
+		URL url = connection.getURL();
-		resultRequest.setHost(url1.getHost());
+		resultRequest.setHost(url.getHost());
-		resultRequest.setPort(url1.getPort());
+		resultRequest.setPort(url.getPort());
 		if (response.getStatusCode().is5xxServerError()) {
 			resultRequest.setRetStatus(RetStatus.RetFail);
 		}

spring-cloud-starter-tencent-polaris-circuitbreaker/src/test/java/com/tencent/cloud/polaris/circuitbreaker/PolarisRestTemplateResponseErrorHandlerTest.java

@@ -0,0 +1,78 @@
+/*
+ * 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.polaris.circuitbreaker;
+
+/**
+ * @author : wh
+ * @date : 2022/6/22 08:48
+ * @description:
+ */
+
+import com.tencent.polaris.api.core.ConsumerAPI;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.http.HttpMethod;
+import org.springframework.test.context.junit4.SpringRunner;
+
+import java.net.HttpURLConnection;
+import java.net.URI;
+import java.net.URL;
+
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+
+/**
+ * @author : wh
+ * @date : 2022/6/22 09:00
+ * @description: Test for {@link PolarisRestTemplateResponseErrorHandler}.
+ */
+@RunWith(SpringRunner.class)
+@SpringBootTest(classes = PolarisRestTemplateResponseErrorHandlerTest.TestApplication.class,
+		properties = {"spring.cloud.polaris.namespace=Test", "spring.cloud.polaris.service=TestApp"})
+public class PolarisRestTemplateResponseErrorHandlerTest {
+
+	
+
+
+	
+	@Test
+	public void handleError() throws Exception{
+		ConsumerAPI consumerAPI = mock(ConsumerAPI.class);
+		PolarisRestTemplateResponseErrorHandler polarisRestTemplateResponseErrorHandler = new PolarisRestTemplateResponseErrorHandler(consumerAPI, null);
+		URI uri = mock(URI.class);
+		when(uri.getPath()).thenReturn("/test");
+		when(uri.getHost()).thenReturn("host");
+		HttpURLConnection httpURLConnection = mock(HttpURLConnection.class);
+		URL url = mock(URL.class);
+		when(httpURLConnection.getURL()).thenReturn(url);
+		when(url.getHost()).thenReturn("127.0.0.1");
+		when(url.getPort()).thenReturn(8080);
+		when(httpURLConnection.getResponseCode()).thenReturn(200);
+		SimpleClientHttpResponseTest clientHttpResponse = new SimpleClientHttpResponseTest(httpURLConnection);
+		polarisRestTemplateResponseErrorHandler.handleError(uri, HttpMethod.GET, clientHttpResponse);
+		when(consumerAPI.unWatchService(null)).thenReturn(true);
+	}
+
+
+	@SpringBootApplication
+	protected static class TestApplication {
+
+	}
+}

spring-cloud-starter-tencent-polaris-circuitbreaker/src/test/java/com/tencent/cloud/polaris/circuitbreaker/SimpleClientHttpResponseTest.java

@@ -0,0 +1,105 @@
+/*
+ * 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.polaris.circuitbreaker;
+
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.client.AbstractClientHttpResponse;
+import org.springframework.lang.Nullable;
+import org.springframework.util.StreamUtils;
+import org.springframework.util.StringUtils;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.net.HttpURLConnection;
+
+/**
+ * @author : wh
+ * @date : 2022/6/22 09:00
+ * @description: mock {@link org.springframework.http.client.SimpleClientHttpResponse}
+ */
+public class SimpleClientHttpResponseTest extends AbstractClientHttpResponse {
+
+	private final HttpURLConnection connection;
+
+	@Nullable
+	private HttpHeaders headers;
+
+	@Nullable
+	private InputStream responseStream;
+
+
+	SimpleClientHttpResponseTest(HttpURLConnection connection) {
+		this.connection = connection;
+	}
+
+
+	@Override
+	public int getRawStatusCode() throws IOException {
+		return this.connection.getResponseCode();
+	}
+
+	@Override
+	public String getStatusText() throws IOException {
+		String result = this.connection.getResponseMessage();
+		return (result != null) ? result : "";
+	}
+
+	@Override
+	public HttpHeaders getHeaders() {
+		if (this.headers == null) {
+			this.headers = new HttpHeaders();
+			// Header field 0 is the status line for most HttpURLConnections, but not on GAE
+			String name = this.connection.getHeaderFieldKey(0);
+			if (StringUtils.hasLength(name)) {
+				this.headers.add(name, this.connection.getHeaderField(0));
+			}
+			int i = 1;
+			while (true) {
+				name = this.connection.getHeaderFieldKey(i);
+				if (!StringUtils.hasLength(name)) {
+					break;
+				}
+				this.headers.add(name, this.connection.getHeaderField(i));
+				i++;
+			}
+		}
+		return this.headers;
+	}
+
+	@Override
+	public InputStream getBody() throws IOException {
+		InputStream errorStream = this.connection.getErrorStream();
+		this.responseStream = (errorStream != null ? errorStream : this.connection.getInputStream());
+		return this.responseStream;
+	}
+
+	@Override
+	public void close() {
+		try {
+			if (this.responseStream == null) {
+				getBody();
+			}
+			StreamUtils.drain(this.responseStream);
+			this.responseStream.close();
+		}
+		catch (Exception ex) {
+			// ignore
+		}
+	}
+
+}