summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authoralexbogo <[email protected]>2023-06-16 11:58:12 +0300
committeralexbogo <[email protected]>2023-06-16 11:58:12 +0300
commit1ea35d0010fa056401642b5555cac149b3ef4690 (patch)
treeb39f82b5671e0b6e47c634442ba6ec0a09b82640
parentc6aa0a6ce4faced28d83bb9d13821f1abb1cfe30 (diff)
[mirrorer] add custom session meta to test
init
-rw-r--r--ydb/core/persqueue/ut/mirrorer_ut.cpp8
-rw-r--r--ydb/public/sdk/cpp/client/ydb_persqueue_core/ut/ut_utils/data_plane_helpers.cpp4
-rw-r--r--ydb/public/sdk/cpp/client/ydb_persqueue_core/ut/ut_utils/data_plane_helpers.h3
3 files changed, 12 insertions, 3 deletions
diff --git a/ydb/core/persqueue/ut/mirrorer_ut.cpp b/ydb/core/persqueue/ut/mirrorer_ut.cpp
index 4da514cbe7e..4dc3cd36d5d 100644
--- a/ydb/core/persqueue/ut/mirrorer_ut.cpp
+++ b/ydb/core/persqueue/ut/mirrorer_ut.cpp
@@ -66,7 +66,13 @@ Y_UNIT_TEST_SUITE(TPersQueueMirrorer) {
TVector<ui32> messagesPerPartition(partitionsCount, 0);
for (ui32 partition = 0; partition < partitionsCount; ++partition) {
TString sourceId = "some_sourceid_" + ToString(partition);
- auto writer = CreateSimpleWriter(*driver, srcTopic, sourceId, partition + 1);
+ THashMap<TString, TString> sessionMeta = {
+ {"some_extra_field", "some_value"},
+ {"some_extra_field2", "another_value" + ToString(partition)},
+ {"file", "/home/user/log" + ToString(partition)}
+ };
+ auto writer = CreateSimpleWriter(*driver, srcTopic, sourceId, partition + 1, std::nullopt, std::nullopt, sessionMeta);
+
ui64 seqNo = writer->GetInitSeqNo();
for (ui32 i = 1; i <= 11; ++i) {
diff --git a/ydb/public/sdk/cpp/client/ydb_persqueue_core/ut/ut_utils/data_plane_helpers.cpp b/ydb/public/sdk/cpp/client/ydb_persqueue_core/ut/ut_utils/data_plane_helpers.cpp
index 4c5eb570f5e..6d29aa27c19 100644
--- a/ydb/public/sdk/cpp/client/ydb_persqueue_core/ut/ut_utils/data_plane_helpers.cpp
+++ b/ydb/public/sdk/cpp/client/ydb_persqueue_core/ut/ut_utils/data_plane_helpers.cpp
@@ -50,7 +50,8 @@ namespace NKikimr::NPersQueueTests {
const TString& sourceId,
std::optional<ui32> partitionGroup,
std::optional<TString> codec,
- std::optional<bool> reconnectOnFailure
+ std::optional<bool> reconnectOnFailure,
+ THashMap<TString, TString> sessionMeta
) {
auto settings = TWriteSessionSettings().Path(topic).MessageGroupId(sourceId);
if (partitionGroup) settings.PartitionGroupId(*partitionGroup);
@@ -64,6 +65,7 @@ namespace NKikimr::NPersQueueTests {
settings.Codec(ECodec::LZOP);
}
settings.MaxMemoryUsage(1024*1024*1024*1024ll);
+ settings.Meta_.Fields = sessionMeta;
return CreateSimpleWriter(driver, settings);
}
diff --git a/ydb/public/sdk/cpp/client/ydb_persqueue_core/ut/ut_utils/data_plane_helpers.h b/ydb/public/sdk/cpp/client/ydb_persqueue_core/ut/ut_utils/data_plane_helpers.h
index 8868ce64b23..90996a89f99 100644
--- a/ydb/public/sdk/cpp/client/ydb_persqueue_core/ut/ut_utils/data_plane_helpers.h
+++ b/ydb/public/sdk/cpp/client/ydb_persqueue_core/ut/ut_utils/data_plane_helpers.h
@@ -33,7 +33,8 @@ namespace NKikimr::NPersQueueTests {
const TString& sourceId,
std::optional<ui32> partitionGroup = {},
std::optional<TString> codec = {},
- std::optional<bool> reconnectOnFailure = {}
+ std::optional<bool> reconnectOnFailure = {},
+ THashMap<TString, TString> sessionMeta = {}
);
std::shared_ptr<NYdb::NPersQueue::IReadSession> CreateReader(