added test for handler chain order

open-telemetry · Apr 1, 2023 · 29b94bc · 29b94bc
1 parent 9e7c788
commit 29b94bc
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/instrumentation/jaxws/jaxws-2.0-cxf-3.0/javaagent/src/test/groovy/TestWsServlet.groovy b/instrumentation/jaxws/jaxws-2.0-cxf-3.0/javaagent/src/test/groovy/TestWsServlet.groovy
@@ -4,7 +4,13 @@
  */
 
 import hello.HelloServiceImpl
+import io.opentelemetry.api.trace.Span
+import io.opentelemetry.context.Context
+import io.opentelemetry.instrumentation.api.instrumenter.LocalRootSpan
 import org.apache.cxf.jaxws.EndpointImpl
+import org.apache.cxf.message.Message
+import org.apache.cxf.phase.AbstractPhaseInterceptor
+import org.apache.cxf.phase.Phase
 import org.apache.cxf.transport.servlet.CXFNonSpringServlet
 
 import javax.servlet.ServletConfig
@@ -18,5 +24,14 @@ class TestWsServlet extends CXFNonSpringServlet {
     Object implementor = new HelloServiceImpl()
     EndpointImpl endpoint = new EndpointImpl(bus, implementor)
     endpoint.publish("/HelloService")
+    endpoint.getOutInterceptors().add(new AbstractPhaseInterceptor<Message>(Phase.SETUP) {
+      @Override
+      void handleMessage(Message message) {
+        Context context = Context.current()
+        if (LocalRootSpan.fromContext(context) != Span.fromContext(context)) {
+          throw new IllegalStateException("handler span should not be ended before outgoing interceptors")
+        }
+      }
+    });
   }
 }