提交 44976020 编写于 作者: A ascrutae

fix test failed

上级 4be6d23e
...@@ -21,6 +21,7 @@ package org.skywalking.apm.plugin.grpc.v1; ...@@ -21,6 +21,7 @@ package org.skywalking.apm.plugin.grpc.v1;
import io.grpc.Metadata; import io.grpc.Metadata;
import io.grpc.MethodDescriptor; import io.grpc.MethodDescriptor;
import io.grpc.Status; import io.grpc.Status;
import io.grpc.StatusRuntimeException;
import org.junit.Before; import org.junit.Before;
import org.junit.Rule; import org.junit.Rule;
import org.junit.Test; import org.junit.Test;
...@@ -99,7 +100,7 @@ public class ClientCallStartInterceptorTest { ...@@ -99,7 +100,7 @@ public class ClientCallStartInterceptorTest {
clientCallStartInterceptor.beforeMethod(clientCallImpl, null, arguments, argumentTypes, null); clientCallStartInterceptor.beforeMethod(clientCallImpl, null, arguments, argumentTypes, null);
clientCallStartInterceptor.afterMethod(clientCallImpl, null, arguments, argumentTypes, null); clientCallStartInterceptor.afterMethod(clientCallImpl, null, arguments, argumentTypes, null);
unaryClientOnCloseInterceptor.afterMethod(null, null, new Object[] {Status.OK}, null, null); unaryClientOnCloseInterceptor.afterMethod(null, null, new Object[] {Status.OK, new Metadata()}, null, null);
assertThat(segmentStorage.getTraceSegments().size(), is(1)); assertThat(segmentStorage.getTraceSegments().size(), is(1));
TraceSegment traceSegment = segmentStorage.getTraceSegments().get(0); TraceSegment traceSegment = segmentStorage.getTraceSegments().get(0);
...@@ -116,7 +117,7 @@ public class ClientCallStartInterceptorTest { ...@@ -116,7 +117,7 @@ public class ClientCallStartInterceptorTest {
clientCallStartInterceptor.beforeMethod(clientCallImpl, null, arguments, argumentTypes, null); clientCallStartInterceptor.beforeMethod(clientCallImpl, null, arguments, argumentTypes, null);
clientCallStartInterceptor.afterMethod(clientCallImpl, null, arguments, argumentTypes, null); clientCallStartInterceptor.afterMethod(clientCallImpl, null, arguments, argumentTypes, null);
unaryClientOnCloseInterceptor.afterMethod(null, null, new Object[] {exceptionStatus}, null, null); unaryClientOnCloseInterceptor.afterMethod(null, null, new Object[] {exceptionStatus, new Metadata()}, null, null);
assertThat(segmentStorage.getTraceSegments().size(), is(1)); assertThat(segmentStorage.getTraceSegments().size(), is(1));
TraceSegment traceSegment = segmentStorage.getTraceSegments().get(0); TraceSegment traceSegment = segmentStorage.getTraceSegments().get(0);
...@@ -125,7 +126,7 @@ public class ClientCallStartInterceptorTest { ...@@ -125,7 +126,7 @@ public class ClientCallStartInterceptorTest {
SpanAssert.assertComponent(abstractTracingSpan, ComponentsDefine.GRPC); SpanAssert.assertComponent(abstractTracingSpan, ComponentsDefine.GRPC);
SpanAssert.assertLayer(abstractTracingSpan, SpanLayer.RPC_FRAMEWORK); SpanAssert.assertLayer(abstractTracingSpan, SpanLayer.RPC_FRAMEWORK);
SpanAssert.assertOccurException(abstractTracingSpan, true); SpanAssert.assertOccurException(abstractTracingSpan, true);
SpanAssert.assertException(SpanHelper.getLogs(abstractTracingSpan).get(0), RuntimeException.class); SpanAssert.assertException(SpanHelper.getLogs(abstractTracingSpan).get(0), StatusRuntimeException.class, "NOT_FOUND");
} }
@Test @Test
......
...@@ -26,6 +26,7 @@ import org.junit.Test; ...@@ -26,6 +26,7 @@ import org.junit.Test;
import org.junit.runner.RunWith; import org.junit.runner.RunWith;
import org.mockito.Matchers; import org.mockito.Matchers;
import org.mockito.Mock; import org.mockito.Mock;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner; import org.powermock.modules.junit4.PowerMockRunner;
import org.skywalking.apm.agent.core.plugin.interceptor.enhance.EnhancedInstance; import org.skywalking.apm.agent.core.plugin.interceptor.enhance.EnhancedInstance;
...@@ -34,6 +35,7 @@ import static org.mockito.Mockito.verify; ...@@ -34,6 +35,7 @@ import static org.mockito.Mockito.verify;
import static org.powermock.api.mockito.PowerMockito.when; import static org.powermock.api.mockito.PowerMockito.when;
@RunWith(PowerMockRunner.class) @RunWith(PowerMockRunner.class)
@PrepareForTest(MethodDescriptor.class)
public class ServerCallHandlerInterceptorTest { public class ServerCallHandlerInterceptorTest {
@Mock @Mock
private EnhancedInstance enhancedInstance; private EnhancedInstance enhancedInstance;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册