PowerMockito Error

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 spark kafka库

主代码如下

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，你应该返回一个相同类的对象