Add MessageBuilder and use it on producer classes

This commit is contained in:
Matias Fontanini
2017-04-15 16:46:37 -07:00
parent c8154ac6cb
commit e26d7e7db2
10 changed files with 224 additions and 127 deletions

View File

@@ -118,7 +118,7 @@ TEST_F(ConsumerTest, AssignmentCallback) {
Producer producer(make_producer_config());
Topic topic = producer.get_topic(KAFKA_TOPIC);
string payload = "Hello world!";
producer.produce(topic, partition, payload);
producer.produce(MessageBuilder(topic).partition(partition).payload(payload));
runner.try_join();
// All 3 partitions should be ours
@@ -173,7 +173,7 @@ TEST_F(ConsumerTest, Rebalance) {
Producer producer(make_producer_config());
Topic topic = producer.get_topic(KAFKA_TOPIC);
string payload = "Hello world!";
producer.produce(topic, partition, payload);
producer.produce(MessageBuilder(topic).partition(partition).payload(payload));
runner1.try_join();
runner2.try_join();
@@ -215,7 +215,7 @@ TEST_F(ConsumerTest, OffsetCommit) {
Producer producer(make_producer_config());
Topic topic = producer.get_topic(KAFKA_TOPIC);
string payload = "Hello world!";
producer.produce(topic, partition, payload);
producer.produce(MessageBuilder(topic).partition(partition).payload(payload));
runner.try_join();
ASSERT_EQ(1, runner.get_messages().size());