在構造函數中初始化的模擬依賴項

Question

我想模擬在構造函數中初始化的依賴關系，在我的情況下，我想模擬Kafka生產者，以便可以模擬通過kafka發送消息，我的代碼如下所示：

private Producer<String, String> producer;

private int messageTimeOut;
private String topicName;

@Autowired
public classConstructor(@Value("${bootstrap.servers}") String bootstrapServers,
        @Value("${topic.name}") String topicName, @Value("${message.send.timeout}") int messageTimeOut) {
    this.messageTimeOut = messageTimeOut;
    this.topicName = topicName;
    Properties props = new Properties();
    props.put("bootstrap.servers", bootstrapServers);
    props.put("key.serializer", StringSerializer.class.getName());
    props.put("value.serializer", StringSerializer.class.getName());
    props.put("acks", "all");
    producer = new KafkaProducer<>(props);
}

任何人都可以建議如何實現這一目標。

Answer 1

無法模擬構造函數創建的對象。 理想情況下，創建類時，不應實例化要模擬的類。 無論如何，有幾種解決方法可以實現此目的。

1. 將KafkaProducer作為構造函數中的參數傳遞。
2. 添加一個setKafkaProducer方法以僅用於單元測試，並將模擬對象設置為該類。
3. 使用Reflection設置私有字段http://www.java2s.com/Code/Java/Reflection/Setprivatefieldvalue.htm

Answer 2

我想您可以使用PowerMock。 它允許在實例化時返回自定義對象。 像這樣：

@RunWith(PowerMockRunner.class)
@PrepareForTest({TargetClass.class})
public class Test {
    @Before
    public void setUp() throws Exception {
        PowerMockito.whenNew(TargetClass.class).withAnyArguments().thenReturn(/* instance with you target constructor arguments */);
    }
}