java - 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 个。

这个数字似乎在某种程度上是可变的，但是当我在 Debug模式下测试测试时序时，它会发生巨大的变化。例如，如果我在 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 发表了评论，以查看该修复程序是否确实已发布，但目前看来此解决方案可以解决问题。