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

无法模拟构造函数创建的对象。理想情况下，在创建类时，不应实例化要模拟的类。无论如何，有几种解决方法可以解决这个问题。

将KafkaProducer作为参数传递给构造函数。
添加setKafkaProducer方法以仅用于单元测试，并将模拟对象设置为类。
使用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 */);
    }
}

模拟依赖项在构造函数中初始化

2 个答案: