NO-JIRA fixing openTracingPluginTest

apache · Apr 27, 2022 · 4bfba9fe96f6b311b60e0674a68baefe0aebfd96 · 4bfba9f
1 parent 72251dc
commit 4bfba9fe96f6b311b60e0674a68baefe0aebfd96
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/...ntracing/src/test/java/org/apache/activemq/artemis/jms/example/OpenTracingPluginTest.java b/...ntracing/src/test/java/org/apache/activemq/artemis/jms/example/OpenTracingPluginTest.java
@@ -115,8 +115,10 @@ public void assertOnSendException() throws ActiveMQException {
    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);