|
31 | 31 | import static org.mockito.Mockito.mock; |
32 | 32 | import static org.mockito.Mockito.timeout; |
33 | 33 | import static org.mockito.Mockito.verify; |
34 | | -import static org.mockito.Mockito.when; |
35 | 34 |
|
36 | 35 | import com.google.common.base.Stopwatch; |
37 | 36 | import com.google.common.collect.ImmutableList; |
@@ -1929,8 +1928,9 @@ public void unaryRpc(SimpleRequest request, StreamObserver<SimpleResponse> respo |
1929 | 1928 | public void serverMetricsShouldRecordContextWithBaggage() { |
1930 | 1929 | // Mocks |
1931 | 1930 | DoubleHistogram serverCallDurationCounter = mock(DoubleHistogram.class); |
1932 | | - OpenTelemetryMetricsResource resource = mock(OpenTelemetryMetricsResource.class); |
1933 | | - when(resource.serverCallDurationCounter()).thenReturn(serverCallDurationCounter); |
| 1931 | + OpenTelemetryMetricsResource resource = OpenTelemetryMetricsResource.builder() |
| 1932 | + .serverCallDurationCounter(serverCallDurationCounter) |
| 1933 | + .build(); |
1934 | 1934 |
|
1935 | 1935 | // ContextPropagators with Baggage |
1936 | 1936 | ContextPropagators propagators = ContextPropagators.create( |
@@ -1974,13 +1974,14 @@ public void serverMetrics_withExternalExecutor_propagatesBaggage() throws Except |
1974 | 1974 | LongCounter serverCallCountCounter = mock(LongCounter.class); |
1975 | 1975 | LongHistogram serverTotalSentCompressedMessageSizeCounter = mock(LongHistogram.class); |
1976 | 1976 | LongHistogram serverTotalReceivedCompressedMessageSizeCounter = mock(LongHistogram.class); |
1977 | | - OpenTelemetryMetricsResource resource = mock(OpenTelemetryMetricsResource.class); |
1978 | | - when(resource.serverCallDurationCounter()).thenReturn(serverCallDurationCounter); |
1979 | | - when(resource.serverCallCountCounter()).thenReturn(serverCallCountCounter); |
1980 | | - when(resource.serverTotalSentCompressedMessageSizeCounter()) |
1981 | | - .thenReturn(serverTotalSentCompressedMessageSizeCounter); |
1982 | | - when(resource.serverTotalReceivedCompressedMessageSizeCounter()) |
1983 | | - .thenReturn(serverTotalReceivedCompressedMessageSizeCounter); |
| 1977 | + OpenTelemetryMetricsResource resource = OpenTelemetryMetricsResource.builder() |
| 1978 | + .serverCallDurationCounter(serverCallDurationCounter) |
| 1979 | + .serverCallCountCounter(serverCallCountCounter) |
| 1980 | + .serverTotalSentCompressedMessageSizeCounter( |
| 1981 | + serverTotalSentCompressedMessageSizeCounter) |
| 1982 | + .serverTotalReceivedCompressedMessageSizeCounter( |
| 1983 | + serverTotalReceivedCompressedMessageSizeCounter) |
| 1984 | + .build(); |
1984 | 1985 |
|
1985 | 1986 | // Setup Propagators |
1986 | 1987 | ContextPropagators propagators = ContextPropagators.create(W3CBaggagePropagator.getInstance()); |
|
0 commit comments