Telecominfraproject/wlan-cloud-lib-cppkafka
1
0
Fork 0
mirror of https://github.com/Telecominfraproject/wlan-cloud-lib-cppkafka.git synced 2026-01-27 02:22:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix Consumer::poll_batch test

Browse Source
This commit is contained in:
Matias Fontanini
2017-10-14 08:28:51 -07:00
parent 5889c322c2
commit 853396acab
1 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
tests/consumer_test.cpp
View File

@@ -3,6 +3,7 @@
#include <set>
#include <mutex>
#include <chrono>
#include <iterator>
#include <condition_variable>
#include <gtest/gtest.h>
#include "cppkafka/consumer.h"
@@ -21,6 +22,7 @@ using std::tie;
using std::condition_variable;
using std::lock_guard;
using std::unique_lock;
using std::make_move_iterator;
using std::chrono::seconds;
using std::chrono::milliseconds;
using std::chrono::system_clock;
@@ -232,8 +234,15 @@ TEST_F(ConsumerTest, ConsumeBatch) {
producer.produce(MessageBuilder(KAFKA_TOPIC).partition(partition).payload(payload));
producer.flush();
vector<Message> messages = consumer.poll_batch(2);
ASSERT_EQ(2, messages.size());
EXPECT_EQ(payload, messages[0].get_payload());
EXPECT_EQ(payload, messages[1].get_payload());
vector<Message> all_messages;
int i = 0;
while (i < 5 && all_messages.size() != 2) {
vector<Message> messages = consumer.poll_batch(2);
all_messages.insert(all_messages.end(), make_move_iterator(messages.begin()),
make_move_iterator(messages.end()));
++i;
}
ASSERT_EQ(2, all_messages.size());
EXPECT_EQ(payload, all_messages[0].get_payload());
EXPECT_EQ(payload, all_messages[1].get_payload());
}