tursom/memgraph
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix Kafka flaky unit test (#1409)

Browse Source
This commit is contained in:
Aidar Samerkhanov 2023-11-05 22:51:56 +03:00 committed by GitHub
parent 48631d1e37
commit 16b8c7b27c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/integrations/kafka/consumer.cpp
View File

@ -1,4 +1,4 @@
// Copyright 2022 Memgraph Ltd.
// Copyright 2023 Memgraph Ltd.
//
// Use of this software is governed by the Business Source License
// included in the file licenses/BSL.txt; by using this file, you agree to be bound by the terms of the Business Source

8
tests/unit/integrations_kafka_consumer.cpp
View File

@ -1,4 +1,4 @@
// Copyright 2022 Memgraph Ltd.
// Copyright 2023 Memgraph Ltd.
//
// Use of this software is governed by the Business Source License
// included in the file licenses/BSL.txt; by using this file, you agree to be bound by the terms of the Business Source
@ -147,6 +147,8 @@ TEST_F(ConsumerTest, BatchInterval) {
cluster.SeedTopic(kTopicName, kMessage);
std::this_thread::sleep_for(kBatchInterval * 0.5);
}
// Wait for all messages to be delivered
std::this_thread::sleep_for(kBatchInterval);
consumer->Stop();
EXPECT_TRUE(expected_messages_received) << "Some unexpected message has been received";
@ -170,9 +172,13 @@ TEST_F(ConsumerTest, BatchInterval) {
EXPECT_TRUE(1 <= received_timestamps[0].first && received_timestamps[0].first <= 2);
EXPECT_LE(3, received_timestamps.size());
int msgsCnt = received_timestamps[0].first;
for (auto i = 1; i < received_timestamps.size(); ++i) {
msgsCnt += received_timestamps[i].first;
check_received_timestamp(i);
}
EXPECT_EQ(kMessageCount, msgsCnt);
}
TEST_F(ConsumerTest, StartStop) {