Skip to content

Commit 7e10adf

Browse files
committed
style: format and use convenience constructor in tests
1 parent 8422318 commit 7e10adf

3 files changed

Lines changed: 13 additions & 8 deletions

File tree

sdk-platform-java/gax-java/gax/src/main/java/com/google/api/gax/tracing/TracedUnaryCallable.java

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -80,6 +80,13 @@ public TracedUnaryCallable(
8080
this.resourceNameExtractor = resourceNameExtractor;
8181
}
8282

83+
TracedUnaryCallable(
84+
UnaryCallable<RequestT, ResponseT> innerCallable,
85+
ApiTracerFactory tracerFactory,
86+
ApiTracerContext apiTracerContext) {
87+
this(innerCallable, tracerFactory, apiTracerContext, null);
88+
}
89+
8390
/**
8491
* Calls the wrapped {@link UnaryCallable} within the context of a new trace.
8592
*

sdk-platform-java/gax-java/gax/src/test/java/com/google/api/gax/tracing/TracedCallableTest.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@
5656

5757
@ExtendWith(MockitoExtension.class)
5858
class TracedCallableTest {
59-
private static final SpanName SPAN_NAME = SpanName.of("FakeClient", "FakeRpc");
59+
6060
private static final ApiTracerContext TRACER_CONTEXT =
6161
ApiTracerContext.newBuilder()
6262
.setFullMethodName("FakeClient/FakeRpc")
@@ -94,7 +94,7 @@ public UnaryCallable<String, String> setupTracedUnaryCallable(
9494
UnaryCallSettings<Object, Object> callSettings) {
9595
UnaryCallable<String, String> callable =
9696
Callables.retrying(innerCallable, callSettings, clientContext);
97-
return new TracedUnaryCallable<>(callable, tracerFactory, TRACER_CONTEXT, null);
97+
return new TracedUnaryCallable<>(callable, tracerFactory, TRACER_CONTEXT);
9898
}
9999

100100
@Test

sdk-platform-java/gax-java/gax/src/test/java/com/google/api/gax/tracing/TracedUnaryCallableTest.java

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@
5353

5454
@ExtendWith(MockitoExtension.class)
5555
class TracedUnaryCallableTest {
56-
private static final SpanName SPAN_NAME = SpanName.of("FakeClient", "FakeRpc");
56+
5757
private static final ApiTracerContext TRACER_CONTEXT =
5858
ApiTracerContext.newBuilder()
5959
.setFullMethodName("FakeClient/FakeRpc")
@@ -77,8 +77,7 @@ void init() {
7777
// Wire the mock tracer factory
7878
when(tracerFactory.newTracer(any(ApiTracer.class), any(ApiTracerContext.class)))
7979
.thenReturn(tracer);
80-
tracedUnaryCallable =
81-
new TracedUnaryCallable<>(innerCallable, tracerFactory, TRACER_CONTEXT, null);
80+
tracedUnaryCallable = new TracedUnaryCallable<>(innerCallable, tracerFactory, TRACER_CONTEXT);
8281

8382
// Wire the mock inner callable
8483
innerResult = SettableApiFuture.create();
@@ -98,13 +97,12 @@ void testTracerCreated() {
9897
void testOperationTypeIsSet() {
9998
when(tracerFactory.newTracer(any(ApiTracer.class), any(ApiTracerContext.class)))
10099
.thenReturn(tracer);
101-
tracedUnaryCallable =
102-
new TracedUnaryCallable<>(innerCallable, tracerFactory, TRACER_CONTEXT, null);
100+
tracedUnaryCallable = new TracedUnaryCallable<>(innerCallable, tracerFactory, TRACER_CONTEXT);
103101
ApiTracerContext contextWithWrongType =
104102
TRACER_CONTEXT.toBuilder().setOperationType(OperationType.BidiStreaming).build();
105103

106104
tracedUnaryCallable =
107-
new TracedUnaryCallable<>(innerCallable, tracerFactory, contextWithWrongType, null);
105+
new TracedUnaryCallable<>(innerCallable, tracerFactory, contextWithWrongType);
108106

109107
innerResult = SettableApiFuture.create();
110108
when(innerCallable.futureCall(anyString(), any(ApiCallContext.class))).thenReturn(innerResult);

0 commit comments

Comments
 (0)