|
19 | 19 | import java.util.List;
|
20 | 20 |
|
21 | 21 | import org.junit.Test;
|
22 |
| -import org.junit.runner.RunWith; |
23 | 22 |
|
24 |
| -import org.springframework.beans.factory.annotation.Autowired; |
25 |
| -import org.springframework.boot.test.context.SpringBootTest; |
| 23 | +import org.springframework.boot.autoconfigure.AutoConfigurations; |
| 24 | +import org.springframework.boot.test.context.runner.WebApplicationContextRunner; |
26 | 25 | import org.springframework.cloud.gcp.logging.CompositeTraceIdExtractor;
|
27 | 26 | import org.springframework.cloud.gcp.logging.TraceIdExtractor;
|
28 | 27 | import org.springframework.cloud.gcp.logging.TraceIdLoggingWebMvcInterceptor;
|
29 | 28 | import org.springframework.cloud.gcp.logging.XCloudTraceIdExtractor;
|
30 | 29 | import org.springframework.cloud.gcp.logging.ZipkinTraceIdExtractor;
|
31 |
| -import org.springframework.test.context.TestPropertySource; |
32 |
| -import org.springframework.test.context.junit4.SpringRunner; |
33 | 30 |
|
34 |
| -import static org.hamcrest.Matchers.is; |
35 |
| -import static org.junit.Assert.assertEquals; |
36 |
| -import static org.junit.Assert.assertThat; |
37 |
| -import static org.junit.Assert.assertTrue; |
| 31 | +import static org.assertj.core.api.Assertions.assertThat; |
38 | 32 |
|
39 | 33 | /**
|
40 | 34 | * @author Mike Eltsufin
|
| 35 | + * @author João André Martins |
41 | 36 | */
|
42 |
| -@RunWith(SpringRunner.class) |
43 |
| -@SpringBootTest(classes = { StackdriverLoggingAutoConfiguration.class }, |
44 |
| - properties = { |
45 |
| - "debug", |
46 |
| - "spring.cloud.gcp.config.enabled=false" |
47 |
| - }, |
48 |
| - webEnvironment = SpringBootTest.WebEnvironment.MOCK) |
49 |
| -public abstract class StackdriverLoggingAutoConfigurationTests { |
50 |
| - |
51 |
| - @Autowired |
52 |
| - private Object[] interceptors; |
53 |
| - |
54 |
| - public abstract void test(); |
55 |
| - |
56 |
| - public int countTraceIdInterceptors() { |
57 |
| - int count = 0; |
58 |
| - for (int i = 0; i < this.interceptors.length; i++) { |
59 |
| - if (this.interceptors[i] instanceof TraceIdLoggingWebMvcInterceptor) { |
60 |
| - count++; |
61 |
| - } |
62 |
| - } |
63 |
| - return count; |
| 37 | +public class StackdriverLoggingAutoConfigurationTests { |
| 38 | + |
| 39 | + private WebApplicationContextRunner contextRunner = |
| 40 | + new WebApplicationContextRunner() |
| 41 | + .withConfiguration( |
| 42 | + AutoConfigurations.of(StackdriverLoggingAutoConfiguration.class)); |
| 43 | + |
| 44 | + @Test |
| 45 | + public void testDisabledConfiguration() { |
| 46 | + this.contextRunner.withPropertyValues("spring.cloud.gcp.logging.enabled=false") |
| 47 | + .run(context -> { |
| 48 | + assertThat(context.getBeansOfType(TraceIdLoggingWebMvcInterceptor.class).size()) |
| 49 | + .isEqualTo(0); |
| 50 | + }); |
| 51 | + } |
| 52 | + |
| 53 | + @Test |
| 54 | + public void testRegularConfiguration() { |
| 55 | + this.contextRunner.run(context -> { |
| 56 | + assertThat(context.getBeansOfType(TraceIdLoggingWebMvcInterceptor.class).size()) |
| 57 | + .isEqualTo(1); |
| 58 | + }); |
64 | 59 | }
|
65 | 60 |
|
66 | 61 | @Test
|
67 | 62 | public void testGetTraceIdExtractorsDefault() {
|
68 |
| - StackdriverLoggingProperties propertires = new StackdriverLoggingProperties(); |
69 |
| - propertires.setTraceIdExtractor(null); |
70 |
| - List<TraceIdExtractor> extractors = ((CompositeTraceIdExtractor) new StackdriverLoggingAutoConfiguration() |
71 |
| - .traceIdExtractor(propertires)).getExtractors(); |
72 |
| - |
73 |
| - assertEquals(2, extractors.size()); |
74 |
| - assertTrue(extractors.get(0) instanceof XCloudTraceIdExtractor); |
75 |
| - assertTrue(extractors.get(1) instanceof ZipkinTraceIdExtractor); |
| 63 | + StackdriverLoggingProperties properties = new StackdriverLoggingProperties(); |
| 64 | + properties.setTraceIdExtractor(null); |
| 65 | + List<TraceIdExtractor> extractors = |
| 66 | + ((CompositeTraceIdExtractor) new StackdriverLoggingAutoConfiguration() |
| 67 | + .traceIdExtractor(properties)).getExtractors(); |
| 68 | + |
| 69 | + assertThat(extractors.size()).isEqualTo(2); |
| 70 | + assertThat(extractors.get(0)).isInstanceOf(XCloudTraceIdExtractor.class); |
| 71 | + assertThat(extractors.get(1)).isInstanceOf(ZipkinTraceIdExtractor.class); |
76 | 72 | }
|
77 | 73 |
|
78 | 74 | @Test
|
79 | 75 | public void testGetTraceIdExtractorsPrioritizeXCloudTrace() {
|
80 |
| - StackdriverLoggingProperties propertires = new StackdriverLoggingProperties(); |
81 |
| - propertires.setTraceIdExtractor(TraceIdExtractorType.XCLOUD_ZIPKIN); |
82 |
| - List<TraceIdExtractor> extractors = ((CompositeTraceIdExtractor) new StackdriverLoggingAutoConfiguration() |
83 |
| - .traceIdExtractor(propertires)).getExtractors(); |
84 |
| - |
85 |
| - assertEquals(2, extractors.size()); |
86 |
| - assertTrue(extractors.get(0) instanceof XCloudTraceIdExtractor); |
87 |
| - assertTrue(extractors.get(1) instanceof ZipkinTraceIdExtractor); |
| 76 | + StackdriverLoggingProperties properties = new StackdriverLoggingProperties(); |
| 77 | + properties.setTraceIdExtractor(TraceIdExtractorType.XCLOUD_ZIPKIN); |
| 78 | + List<TraceIdExtractor> extractors = |
| 79 | + ((CompositeTraceIdExtractor) new StackdriverLoggingAutoConfiguration() |
| 80 | + .traceIdExtractor(properties)).getExtractors(); |
| 81 | + |
| 82 | + assertThat(extractors.size()).isEqualTo(2); |
| 83 | + assertThat(extractors.get(0)).isInstanceOf(XCloudTraceIdExtractor.class); |
| 84 | + assertThat(extractors.get(1)).isInstanceOf(ZipkinTraceIdExtractor.class); |
88 | 85 | }
|
89 | 86 |
|
90 | 87 | @Test
|
91 | 88 | public void testGetTraceIdExtractorsPrioritizeZipkinTrace() {
|
92 |
| - StackdriverLoggingProperties propertires = new StackdriverLoggingProperties(); |
93 |
| - propertires.setTraceIdExtractor(TraceIdExtractorType.ZIPKIN_XCLOUD); |
94 |
| - List<TraceIdExtractor> extractors = ((CompositeTraceIdExtractor) new StackdriverLoggingAutoConfiguration() |
95 |
| - .traceIdExtractor(propertires)).getExtractors(); |
96 |
| - |
97 |
| - assertEquals(2, extractors.size()); |
98 |
| - assertTrue(extractors.get(0) instanceof ZipkinTraceIdExtractor); |
99 |
| - assertTrue(extractors.get(1) instanceof XCloudTraceIdExtractor); |
| 89 | + StackdriverLoggingProperties properties = new StackdriverLoggingProperties(); |
| 90 | + properties.setTraceIdExtractor(TraceIdExtractorType.ZIPKIN_XCLOUD); |
| 91 | + List<TraceIdExtractor> extractors = |
| 92 | + ((CompositeTraceIdExtractor) new StackdriverLoggingAutoConfiguration() |
| 93 | + .traceIdExtractor(properties)).getExtractors(); |
| 94 | + |
| 95 | + assertThat(extractors.size()).isEqualTo(2); |
| 96 | + assertThat(extractors.get(0)).isInstanceOf(ZipkinTraceIdExtractor.class); |
| 97 | + assertThat(extractors.get(1)).isInstanceOf(XCloudTraceIdExtractor.class); |
100 | 98 | }
|
101 | 99 |
|
102 | 100 | @Test
|
103 | 101 | public void testGetTraceIdExtractorsOnlyXCloud() {
|
104 |
| - StackdriverLoggingProperties propertires = new StackdriverLoggingProperties(); |
105 |
| - propertires.setTraceIdExtractor(TraceIdExtractorType.XCLOUD); |
| 102 | + StackdriverLoggingProperties properties = new StackdriverLoggingProperties(); |
| 103 | + properties.setTraceIdExtractor(TraceIdExtractorType.XCLOUD); |
106 | 104 |
|
107 |
| - assertTrue(new StackdriverLoggingAutoConfiguration() |
108 |
| - .traceIdExtractor(propertires) instanceof XCloudTraceIdExtractor); |
| 105 | + assertThat(new StackdriverLoggingAutoConfiguration().traceIdExtractor(properties)) |
| 106 | + .isInstanceOf(XCloudTraceIdExtractor.class); |
109 | 107 | }
|
110 | 108 |
|
111 | 109 | @Test
|
112 | 110 | public void testGetTraceIdExtractorsOnlyZipkin() {
|
113 |
| - StackdriverLoggingProperties propertires = new StackdriverLoggingProperties(); |
114 |
| - propertires.setTraceIdExtractor(TraceIdExtractorType.ZIPKIN); |
115 |
| - |
116 |
| - assertTrue(new StackdriverLoggingAutoConfiguration() |
117 |
| - .traceIdExtractor(propertires) instanceof ZipkinTraceIdExtractor); |
118 |
| - } |
119 |
| - |
120 |
| - @TestPropertySource(properties = { "spring.cloud.gcp.logging.enabled=false" }) |
121 |
| - public static class StackdriverLoggingAutoConfigurationDisabledTests |
122 |
| - extends StackdriverLoggingAutoConfigurationTests { |
123 |
| - |
124 |
| - @Test |
125 |
| - public void test() { |
126 |
| - assertThat(countTraceIdInterceptors(), is(0)); |
127 |
| - } |
128 |
| - } |
129 |
| - |
130 |
| - public static class StackdriverLoggingAutoConfigurationDefaultTests |
131 |
| - extends StackdriverLoggingAutoConfigurationTests { |
| 111 | + StackdriverLoggingProperties properties = new StackdriverLoggingProperties(); |
| 112 | + properties.setTraceIdExtractor(TraceIdExtractorType.ZIPKIN); |
132 | 113 |
|
133 |
| - @Test |
134 |
| - public void test() { |
135 |
| - assertThat(countTraceIdInterceptors(), is(1)); |
136 |
| - } |
| 114 | + assertThat(new StackdriverLoggingAutoConfiguration().traceIdExtractor(properties)) |
| 115 | + .isInstanceOf(ZipkinTraceIdExtractor.class); |
137 | 116 | }
|
138 | 117 | }
|
0 commit comments