PowerMockito错误

Question

需要快速的帮助。我正在尝试编写一个测试类，并得到以下错误：“无法解决方法.thenreturn(org.apache.kafka.clients.producer)”

@Test
public void testPublishData_Success() throws java.lang.Exception {
    when(GetPropValues.getPropValue(PublisherConstants.ATMID)).thenReturn("ATM");
    when(GetPropValues.getPropValue(PublisherConstants.DATA_SOURCE)).thenReturn("PCE");

    ReadAndWriteFiles mockFiles = Mockito.mock(ReadAndWriteFiles.class);
    PowerMockito.whenNew(ReadAndWriteFiles.class).withNoArguments().thenReturn(mockFiles);
    Mockito.when(mockFiles.getAllFiles()).thenReturn("someValue");

    KafkaProducer mockProducer = Mockito.mock(KafkaProducer.class);
    PowerMockito.whenNew(KafkaProducer.class).withAnyArguments().thenReturn(mockProducer);

    producer.publishData(null, "Test", "Data1");
}

Powermockito可以返回ReadAndWriteFiles.class对象，但它正在为KafkaProducer.class抛出一个错误。在线

PowerMockito.whenNew(KafkaProducer.class).withAnyArguments().thenReturn(mockProducer);

还有其他方法可以解决这个问题吗？如有任何建议，将不胜感激。

注意: KafkaProducer.class不是自定义类，而是apache库的内部类。

主要代码如下所示

KafkaProducer<String, String> producer = new KafkaProducer<String, String>(props);
            InputData inputMessage;
            try {
                inputMessage = populateData(timeStamp, dataCategory, data, atmId, topic);
                ReadAndWriteFiles readerWriter = new ReadAndWriteFiles();
                File[] directory = readerWriter.getAllFiles();
                if (directory != null && directory.length > 0) {
                    if (connectionSet && !publishingData) {
                        sendDataFromFiles(producer, directory);
                        publishingData = false;
                    }
                } else {
                    producer.send(keyedMsg, new KafkaResponseHandler(inputMessage));
                }

            } catch (IOException e) {

            }

Answer 1

我认为错误是

KafkaProducer mockProducer = Mockito.mock(KafkaProducer.class);
PowerMockito.whenNew(ReadAndWriteFiles.class).withAnyArguments().thenReturn(mockProducer)

我认为返回的值应该是ReadAndWriteFiles类的模拟，而不是KafkaProducer

ReadAndWriteFiles readMock = Mockito.mock(ReadAndWriteFiles.class)
PowerMockito.whenNew(ReadAndWriteFiles.class).withAnyArguments().thenReturn(readMock)

Mockito.when(readMock.getAllFiles()).thenReturn(anArrayOfFiles);

thenReturn方法的签名如下

OngoingStubbing<T>  [More ...] thenReturn(T value);

因此，您是用来返回一个ReadAndWriteFiles的，您应该返回同一个类的一个对象