diff options
author | alexbogo <[email protected]> | 2023-06-16 11:58:12 +0300 |
---|---|---|
committer | alexbogo <[email protected]> | 2023-06-16 11:58:12 +0300 |
commit | 1ea35d0010fa056401642b5555cac149b3ef4690 (patch) | |
tree | b39f82b5671e0b6e47c634442ba6ec0a09b82640 | |
parent | c6aa0a6ce4faced28d83bb9d13821f1abb1cfe30 (diff) |
[mirrorer] add custom session meta to test
init
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( |