NO-JIRA fixing openTracingPluginTest

This commit is contained in:
Clebert Suconic 2022-04-27 16:03:30 -04:00
parent 72251dc947
commit 4bfba9fe96
1 changed files with 3 additions and 1 deletions

View File

@ -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);