From 4bfba9fe96f6b311b60e0674a68baefe0aebfd96 Mon Sep 17 00:00:00 2001
From: Clebert Suconic <clebertsuconic@apache.org>
Date: Wed, 27 Apr 2022 16:03:30 -0400
Subject: [PATCH] NO-JIRA fixing openTracingPluginTest

---
 .../activemq/artemis/jms/example/OpenTracingPluginTest.java   | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/examples/features/standard/opentracing/src/test/java/org/apache/activemq/artemis/jms/example/OpenTracingPluginTest.java b/examples/features/standard/opentracing/src/test/java/org/apache/activemq/artemis/jms/example/OpenTracingPluginTest.java
index c80e3162ee..653aba695e 100644
--- a/examples/features/standard/opentracing/src/test/java/org/apache/activemq/artemis/jms/example/OpenTracingPluginTest.java
+++ b/examples/features/standard/opentracing/src/test/java/org/apache/activemq/artemis/jms/example/OpenTracingPluginTest.java
@@ -115,8 +115,10 @@ public class OpenTracingPluginTest {
    public void assertAfterDeliver() throws ActiveMQException {
       ServerConsumer consumer = mock(ServerConsumer.class);
       MessageReference reference = mock(MessageReference.class);
+      Message mockMessage = mock(Message.class);
 
-      when(reference.getMessage().getUserContext(Span.class)).thenReturn(span);
+      when(reference.getMessage()).thenReturn(mockMessage);
+      when(mockMessage.getUserContext(Span.class)).thenReturn(span);
 
       plugin.afterDeliver(consumer, reference);