Mockito ArgumentCaptor正在捕获大量重复记录

Question

我正在尝试使用Mockito和JUnit来测试多线程应用程序。以下是一些有问题的代码：

ArgumentCaptor<MessageCreator> messageCaptor = ArgumentCaptor.forClass(MessageCreator.class);
jmsHandler.put(message);
Mockito.verify(mockJmsTemplate, Mockito.after(10000).times(4)).send(messageCaptor.capture());

jmsHandler.put(message)行将String放入应用程序中，该应用程序进入BlockingQueue并从多线程部分开始。然后，我等待该方法在接下来的 10 秒内执行 4 次并捕获结果。

应用程序应该吐出 4 个MessageCreator实例，就我而言，这只是一个包含String的对象，然后我将它与预期的输出进行比较。测试期间的日志记录确认正在创建 4 条消息。

但是，当我尝试遍历ArgumentCaptor的getAllValues()方法来检查结果时，我注意到List正在复制我的程序输出数百万次。它应该有 4 个对象，但最后一次运行有 6,984,988。

这个数字似乎在某种程度上是可变的，但是当我在调试模式下使用测试计时时，它会发生巨大变化。例如，如果我在jmsHandler行设置断点，单步执行该断点，然后等待应用程序完成处理，然后才开始Mockito.verify(...)步骤，则List大小将骤降至"仅"158,636 个对象。

以前有没有人遇到过这种问题？如果我能提供更多细节，请告诉我。

编辑：这是测试和程序结构的独立示例：

import static org.junit.Assert.assertEquals;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.BlockingQueue;
import javax.jms.JMSException;
import javax.jms.Message;
import javax.jms.Session;
import javax.jms.TextMessage;
import org.junit.Before;
import org.junit.Test;
import org.mockito.ArgumentCaptor;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.mockito.MockitoAnnotations;
import org.springframework.jms.core.JmsTemplate;
import org.springframework.jms.core.MessageCreator;
public class MultiThreadTest {
private Input inputHandler;
@Mock
private JmsTemplate mockJmsTemplate;
@Before
public void setup() {
MockitoAnnotations.initMocks(this);
BlockingQueue<String> queue = new ArrayBlockingQueue<String>(10);
inputHandler = new Input();
inputHandler.setQueue(queue);
Output outputHandler = new Output();
outputHandler.setQueue(queue);
outputHandler.setJmsTemplate(mockJmsTemplate);
new Thread(outputHandler).start();
}
@Test
public void testMessage() {
ArgumentCaptor<OutputMessageCreator> messageCaptor = ArgumentCaptor.forClass(OutputMessageCreator.class);
String inMessage = "testMessage";
List<String> expectedMessages = new ArrayList<String>(4);
inputHandler.put(inMessage);
Mockito.verify(mockJmsTemplate, Mockito.after(10000).times(4)).send(messageCaptor.capture());
System.out.println("Number: " + messageCaptor.getAllValues().size());
System.out.println("Equal: " + messageCaptor.getAllValues().get(0).equals(messageCaptor.getAllValues().get(0)));
System.out.println("Equal: " + messageCaptor.getAllValues().get(0).equals(messageCaptor.getAllValues().get(1)));
System.out.println("Equal: " + messageCaptor.getAllValues().get(0).equals(messageCaptor.getAllValues().get(2)));
System.out.println("Equal: " + messageCaptor.getAllValues().get(0).equals(messageCaptor.getAllValues().get(3)));
System.out.println("Equal: " + messageCaptor.getAllValues().get(0).equals(messageCaptor.getAllValues().get(4)));
System.out.println("Equal: " + messageCaptor.getAllValues().get(0).equals(messageCaptor.getAllValues().get(5)));
System.out.println("Equal: " + messageCaptor.getAllValues().get(0).equals(messageCaptor.getAllValues().get(6)));
System.out.println("Equal: " + messageCaptor.getAllValues().get(0).equals(messageCaptor.getAllValues().get(7)));
System.out.println("Equal: " + messageCaptor.getAllValues().get(0).equals(messageCaptor.getAllValues().get(8)));
List<String> outMsgs = new ArrayList<String>();
for (OutputMessageCreator creator : messageCaptor.getAllValues()) {
outMsgs.add(creator.getMsg());
}
assertEquals(expectedMessages, outMsgs);
}
private class Input {
private BlockingQueue<String> queue;
public void put(String msg) {
try {
queue.put(msg);
} catch (InterruptedException e) {
e.printStackTrace();
}
}
public void setQueue(BlockingQueue<String> queue) {
this.queue = queue;
}
}
private class Output implements Runnable {
private BlockingQueue<String> queue;
private JmsTemplate jmsTemplate;
private int counter = 1;
@Override
public void run() {
while (true) {
String msg = null;
try {
msg = queue.take();
String[] messagesOut = new String[4];
for (int i = 0; i < 4; i++) {
messagesOut[i] = msg + "-" + counter++;
}
for (String messageOut : messagesOut) {
System.out.println(messageOut);
jmsTemplate.send(new OutputMessageCreator(messageOut));
Thread.sleep(2000);
}
} catch (InterruptedException e) {
e.printStackTrace();
}
}
}
public void setQueue(BlockingQueue<String> queue) {
this.queue = queue;
}
public void setJmsTemplate(JmsTemplate jmsTemplate) {
this.jmsTemplate = jmsTemplate;
}
}
private class OutputMessageCreator implements MessageCreator {
private String msg;
public OutputMessageCreator(String msg) {
this.msg = msg;
}
@Override
public Message createMessage(Session session) throws JMSException {
TextMessage message = session.createTextMessage();
message.setText(msg);
return message;
}
public String getMsg() {
return msg;
}
}
}

下面是运行该测试的输出：

testMessage-1
testMessage-2
testMessage-3
testMessage-4
Number: 5392168
Equal: true
Equal: false
Equal: false
Equal: false
Equal: true
Equal: false
Equal: false
Equal: false
Equal: true

Answer 1

看起来这是Mockito中的某个错误：

https://github.com/mockito/mockito/issues/345

https://github.com/mockito/mockito/issues/379

这些问题似乎暗示这已在 Mockito 2 或 2.1 中修复，但我使用的是 2.0.41-beta 并尝试了 2.4.0 和 2.8.9 并得到了相同的结果。在github问题的一些评论中，似乎有各种各样的解决方法。

这行代码：

Mockito.verify(mockJmsTemplate, Mockito.after(10000).times(4)).send(messageCaptor.capture());

可以替换为：

try {
Thread.sleep(10000);
} catch (InterruptedException e) {
e.printStackTrace();
}
Mockito.verify(mockJmsTemplate, Mockito.times(4)).send(messageCaptor.capture());

我已经对问题 379 发表了评论，以查看修复程序是否实际发布，但似乎这个解决方法现在可以解决问题。