Add tests for KafkaProducer