From 54e963aa2a3ab83c84152427143de7df87e212fd Mon Sep 17 00:00:00 2001
From: George Sedov <george.sedov@desy.de>
Date: Wed, 12 Jan 2022 21:38:28 +0100
Subject: [PATCH] fix kafka unit test
---
.../test_request_handler_kafka_notify.cpp | 9 +++++++--
1 file changed, 7 insertions(+), 2 deletions(-)
diff --git a/receiver/unittests/request_handler/test_request_handler_kafka_notify.cpp b/receiver/unittests/request_handler/test_request_handler_kafka_notify.cpp
index 4b9c68d89..458214f86 100644
--- a/receiver/unittests/request_handler/test_request_handler_kafka_notify.cpp
+++ b/receiver/unittests/request_handler/test_request_handler_kafka_notify.cpp
@@ -15,13 +15,18 @@ class KafkaNotifyHandlerTests : public Test {
NiceMock<MockKafkaClient> kafka_client;
RequestHandlerKafkaNotify handler{&kafka_client};
std::unique_ptr<NiceMock<MockRequest>> mock_request;
- const std::string expected_filename = "filename";
+ std::string expected_filename = std::string("processed") + asapo::kPathSeparator + "filename";
+ std::string expected_offline_path = std::string("offline") + asapo::kPathSeparator + "path";
+ CustomRequestData expected_custom_data {kDefaultIngestMode, 0, 0};
const std::string expected_topic = "asapo";
void SetUp() override {
GenericRequestHeader request_header;
mock_request.reset(new NiceMock<MockRequest> {request_header, 1, "", nullptr});
- EXPECT_CALL(*mock_request, GetFileName()).WillOnce(Return(expected_filename));
+ EXPECT_CALL(*mock_request, GetFileName()).Times(2).WillRepeatedly(Return(expected_filename));
+ EXPECT_CALL(*mock_request, GetSourceType()).Times(2).WillRepeatedly(Return(SourceType::kProcessed));
+ EXPECT_CALL(*mock_request, GetCustomData_t()).WillOnce(Return(expected_custom_data));
+ EXPECT_CALL(*mock_request, GetOfflinePath()).WillOnce(ReturnRef(expected_offline_path));
EXPECT_CALL(kafka_client, Send_t(HasSubstr(expected_filename), expected_topic)).WillOnce(Return(nullptr));
}
--
GitLab