msb-public
/
spring-cloud-tencent
mirror of https://github.com/Tencent/spring-cloud-tencent
3
1
Code Issues Projects Releases Wiki Activity

test:test for TransHeadersTransfer

Browse Source
pull/646/head
wulingxiao 3 years ago
parent 37ca6a740e
commit f53b8217d9
6 changed files with 45 additions and 253 deletions
Show Stats Download Patch File Download Diff File

6
spring-cloud-starter-tencent-metadata-transfer/pom.xml
Unescape View File

@ -50,12 +50,6 @@
<artifactId>spring-cloud-starter-loadbalancer</artifactId> <artifactId>spring-cloud-starter-loadbalancer</artifactId>
<scope>test</scope> <scope>test</scope>
</dependency> </dependency>
<dependency>
<groupId>com.github.stefanbirkner</groupId>
<artifactId>system-rules</artifactId>
<scope>test</scope>
</dependency>
</dependencies> </dependencies>
</project> </project>

129
spring-cloud-starter-tencent-metadata-transfer/src/test/java/com/tencent/cloud/metadata/core/TransHeadersTransferTest.java
Unescape View File

@ -18,108 +18,69 @@
package com.tencent.cloud.metadata.core; package com.tencent.cloud.metadata.core;
import java.util.Map;
import com.tencent.cloud.common.metadata.MetadataContext;
import com.tencent.cloud.common.metadata.MetadataContextHolder; import com.tencent.cloud.common.metadata.MetadataContextHolder;
import com.tencent.cloud.common.util.JacksonUtils; import com.tencent.cloud.common.util.JacksonUtils;
import com.tencent.cloud.metadata.support.DynamicEnvironmentVariable; import org.junit.AfterClass;
import com.tencent.cloud.metadata.support.DynamicEnvironmentVariablesSpringJUnit4ClassRunner;
import org.junit.Before;
import org.junit.Test; import org.junit.Test;
import org.junit.jupiter.api.Assertions;
import org.junit.runner.RunWith; import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.cloud.openfeign.EnableFeignClients;
import org.springframework.core.ParameterizedTypeReference;
import org.springframework.http.MediaType;
import org.springframework.mock.web.MockHttpServletResponse;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.web.reactive.server.FluxExchangeResult;
import org.springframework.test.web.reactive.server.WebTestClient;
import org.springframework.test.web.servlet.MockMvc;
import org.springframework.test.web.servlet.MvcResult;
import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;
import org.springframework.test.web.servlet.result.MockMvcResultHandlers;
import org.springframework.test.web.servlet.setup.MockMvcBuilders;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.context.WebApplicationContext;
import org.springframework.web.server.WebHandler;
import reactor.core.publisher.Flux;
import java.util.Map; import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.mock.http.server.reactive.MockServerHttpRequest;
import org.springframework.mock.web.MockHttpServletRequest;
import org.springframework.test.context.junit4.SpringRunner;
import static org.springframework.boot.test.context.SpringBootTest.WebEnvironment.DEFINED_PORT; import static org.springframework.boot.test.context.SpringBootTest.WebEnvironment.RANDOM_PORT;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
/** /**
* Test for {@link TransHeadersTransfer}. * Test for {@link TransHeadersTransfer}.
* *
* @author lingxiao.wlx * @author lingxiao.wlx
*/ */
@RunWith(DynamicEnvironmentVariablesSpringJUnit4ClassRunner.class) @RunWith(SpringRunner.class)
@DynamicEnvironmentVariable(name = "SCT_TRAFFIC_CONTENT_RAW_TRANSHEADERS", value = "header1,header2,header3") @SpringBootTest(webEnvironment = RANDOM_PORT,
@SpringBootTest(webEnvironment = DEFINED_PORT, classes = DecodeTransferMetadataServletFilterTest.TestApplication.class,
classes = TransHeadersTransferTest.TestApplication.class, properties = {"spring.config.location = classpath:application-test.yml"})
properties = {"server.port=8081", "spring.config.location = classpath:application-test.yml"})
public class TransHeadersTransferTest { public class TransHeadersTransferTest {
@AfterClass
@Autowired public static void afterClass() {
private WebApplicationContext webContext; MetadataContextHolder.remove();
@Autowired
private DecodeTransferMetadataServletFilter metadataServletFilter;
private MockMvc mockMvc;
private WebTestClient webTestClient;
@Before
public void setupMockMvc() throws Exception {
mockMvc = MockMvcBuilders.webAppContextSetup(webContext).addFilter(metadataServletFilter).build();
webTestClient = WebTestClient.bindToApplicationContext(webContext).build();
} }
@Test @Test
public void transferTest() throws Exception { public void transferServletTest() {
MvcResult mvcResult = mockMvc.perform(MockMvcRequestBuilders.get("/transHeaders") MetadataContext metadataContext = MetadataContextHolder.get();
.header("header1", "1") metadataContext.setTransHeaders("header1,header2,header3", "");
.header("header2", "2") MockHttpServletRequest request = new MockHttpServletRequest();
.header("header3", "3") request.addHeader("header1", "1");
.characterEncoding("UTF-8")) request.addHeader("header2", "2");
.andExpect(status().isOk()) request.addHeader("header3", "3");
.andDo(MockMvcResultHandlers.print(System.err)) TransHeadersTransfer.transfer(request);
.andReturn(); Map<String, String> transHeadersKV = MetadataContextHolder.get().getTransHeadersKV();
Assertions.assertEquals(transHeadersKV.get("header1"), "1");
MockHttpServletResponse response = mvcResult.getResponse(); Assertions.assertEquals(transHeadersKV.get("header2"), "2");
String contentAsString = response.getContentAsString(); Assertions.assertEquals(transHeadersKV.get("header3"), "3");
Map<String, String> map = JacksonUtils.deserialize2Map(contentAsString);
System.out.println(contentAsString);
} }
@Test @Test
public void transferReactiveTest(){ public void transferReactiveTest() {
FluxExchangeResult<Map<String, String>> result = webTestClient.get().uri("/transHeaders") MetadataContext metadataContext = MetadataContextHolder.get();
.header("header1", "1") metadataContext.setTransHeaders("header1,header2,header3", "");
.header("header2", "2") MockServerHttpRequest.BaseBuilder<?> builder = MockServerHttpRequest.get("");
.header("header3", "3") String[] header1 = {"1"};
.accept(MediaType.APPLICATION_JSON) String[] header2 = {"2"};
.exchange() String[] header3 = {"3"};
.expectStatus().isOk() builder.header("header1", header1);
.returnResult(new ParameterizedTypeReference<Map<String, String>>() { builder.header("header2", header2);
}); builder.header("header3", header3);
MockServerHttpRequest request = builder.build();
Flux<Map<String, String>> responseBody = result.getResponseBody(); TransHeadersTransfer.transfer(request);
Map<String, String> transHeadersKV = MetadataContextHolder.get().getTransHeadersKV();
Assertions.assertEquals(transHeadersKV.get("header1"), JacksonUtils.serialize2Json(header1));
} Assertions.assertEquals(transHeadersKV.get("header2"), JacksonUtils.serialize2Json(header2));
Assertions.assertEquals(transHeadersKV.get("header3"), JacksonUtils.serialize2Json(header3));
@SpringBootApplication
@EnableFeignClients
@RestController
protected static class TestApplication {
@GetMapping("/transHeaders")
public Map<String, String> test() {
return MetadataContextHolder.get().getTransHeadersKV();
}
} }
} }

50
spring-cloud-starter-tencent-metadata-transfer/src/test/java/com/tencent/cloud/metadata/support/DynamicEnvironmentVariable.java
Unescape View File

@ -1,50 +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.metadata.support;
import java.lang.annotation.ElementType;
import java.lang.annotation.Repeatable;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
/**
* DynamicEnvironmentVariable.
*
* @author lingxiao.wlx
*/
@Repeatable(DynamicEnvironmentVariables.class)
@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.TYPE})
public @interface DynamicEnvironmentVariable {
/**
* EnvironmentVariable name.
*
* @return EnvironmentVariable name.
*/
String name();
/**
* EnvironmentVariable value.
*
* @return EnvironmentVariable value.
*/
String value();
}

41
spring-cloud-starter-tencent-metadata-transfer/src/test/java/com/tencent/cloud/metadata/support/DynamicEnvironmentVariables.java
Unescape View File

@ -1,41 +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.metadata.support;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
/**
* DynamicEnvironmentVariables for multiDynamicEnvironmentVariable.
*
* @author lingxiao.wlx
*/
@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.TYPE})
public @interface DynamicEnvironmentVariables {
/**
* DynamicEnvironmentVariable array.
*
* @return dynamicEnvironmentVariable array.
*/
DynamicEnvironmentVariable[] value();
}

64
spring-cloud-starter-tencent-metadata-transfer/src/test/java/com/tencent/cloud/metadata/support/DynamicEnvironmentVariablesSpringJUnit4ClassRunner.java
Unescape View File

@ -1,64 +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.metadata.support;
import java.util.Arrays;
import java.util.Objects;
import org.junit.Rule;
import org.junit.contrib.java.lang.system.EnvironmentVariables;
import org.junit.runners.model.InitializationError;
import org.springframework.core.annotation.AnnotationUtils;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
/**
* Enhance {@link SpringJUnit4ClassRunner} to support add environment variables when Junit4 test run.
*
*
* @author lingxiao.wlx
* @see DynamicEnvironmentVariable
* @see DynamicEnvironmentVariables
*/
public class DynamicEnvironmentVariablesSpringJUnit4ClassRunner extends SpringJUnit4ClassRunner {
/**
* EnvironmentVariables.
*/
@Rule
public final EnvironmentVariables environmentVariables = new EnvironmentVariables();
public DynamicEnvironmentVariablesSpringJUnit4ClassRunner(Class<?> clazz) throws InitializationError {
super(clazz);
DynamicEnvironmentVariable dynamicEnvironmentVariable = AnnotationUtils.findAnnotation(clazz, DynamicEnvironmentVariable.class);
if (!Objects.isNull(dynamicEnvironmentVariable)) {
String key = dynamicEnvironmentVariable.name();
String value = dynamicEnvironmentVariable.value();
environmentVariables.set(key, value);
}
DynamicEnvironmentVariables dynamicEnvironmentVariables = AnnotationUtils.findAnnotation(clazz, DynamicEnvironmentVariables.class);
if (!Objects.isNull(dynamicEnvironmentVariables)) {
Arrays.stream(dynamicEnvironmentVariables.value()).forEach(
environmentVariable -> {
environmentVariables.set(environmentVariable.name(), environmentVariable.value());
}
);
}
}
}

8
spring-cloud-tencent-dependencies/pom.xml
Unescape View File

@ -80,7 +80,6 @@
<byte-buddy.version>1.12.10</byte-buddy.version> <byte-buddy.version>1.12.10</byte-buddy.version>
<protobuf-java.version>3.16.1</protobuf-java.version> <protobuf-java.version>3.16.1</protobuf-java.version>
<bcprov-jdk15on.version>1.69</bcprov-jdk15on.version> <bcprov-jdk15on.version>1.69</bcprov-jdk15on.version>
<system-rules.version>1.16.1</system-rules.version>
<!-- Maven Plugin Versions --> <!-- Maven Plugin Versions -->
<maven-source-plugin.version>3.2.0</maven-source-plugin.version> <maven-source-plugin.version>3.2.0</maven-source-plugin.version>
@ -249,13 +248,6 @@
<version>${mocktio.version}</version> <version>${mocktio.version}</version>
<scope>test</scope> <scope>test</scope>
</dependency> </dependency>
<dependency>
<groupId>com.github.stefanbirkner</groupId>
<artifactId>system-rules</artifactId>
<version>${system-rules.version}</version>
<scope>test</scope>
</dependency>
</dependencies> </dependencies>
</dependencyManagement> </dependencyManagement>

Write Preview
Loading…
Cancel
Save