Wait for TCommitOffsetAcknowledgementEvent

1 files changed, 26 insertions, 8 deletions

diff --git a/ydb/public/sdk/cpp/client/ydb_topic/ut/describe_topic_ut.cpp b/ydb/public/sdk/cpp/client/ydb_topic/ut/describe_topic_ut.cpp
index a89ed6f171..0e172f64aa 100644
--- a/ydb/public/sdk/cpp/client/ydb_topic/ut/describe_topic_ut.cpp
+++ b/ydb/public/sdk/cpp/client/ydb_topic/ut/describe_topic_ut.cpp
@@ -290,17 +290,35 @@ namespace NYdb::NTopic::NTests {
                                         .AppendTopics(setup.GetTestTopicPath());
                 auto readSession = client.CreateReadSession(readSettings);
 
-                auto event = readSession->GetEvent(true);
-                UNIT_ASSERT(event.Defined());
+                // Event 1: start partition session
+                {
+                    TMaybe<TReadSessionEvent::TEvent> event = readSession->GetEvent(true);
+                    UNIT_ASSERT(event);
+                    auto startPartitionSession = std::get_if<TReadSessionEvent::TStartPartitionSessionEvent>(event.Get());
+                    UNIT_ASSERT_C(startPartitionSession, DebugString(*event));
+
+                    startPartitionSession->Confirm();
+                }
+
+                // Event 2: data received
+                {
+                    TMaybe<TReadSessionEvent::TEvent> event = readSession->GetEvent(true);
+                    UNIT_ASSERT(event);
+                    auto dataReceived = std::get_if<TReadSessionEvent::TDataReceivedEvent>(event.Get());
+                    UNIT_ASSERT_C(dataReceived, DebugString(*event));
 
-                auto& startPartitionSession = std::get<TReadSessionEvent::TStartPartitionSessionEvent>(*event);
-                startPartitionSession.Confirm();
+                    dataReceived->Commit();
+                }
 
-                event = readSession->GetEvent(true);
-                UNIT_ASSERT(event.Defined());
+                // Event 3: commit acknowledgement
+                {
+                    TMaybe<TReadSessionEvent::TEvent> event = readSession->GetEvent(true);
+                    UNIT_ASSERT(event);
+                    auto commitOffsetAck = std::get_if<TReadSessionEvent::TCommitOffsetAcknowledgementEvent>(event.Get());
+                    UNIT_ASSERT_C(commitOffsetAck, DebugString(*event));
 
-                auto& dataReceived = std::get<TReadSessionEvent::TDataReceivedEvent>(*event);
-                dataReceived.Commit();
+                    UNIT_ASSERT_VALUES_EQUAL(commitOffsetAck->GetCommittedOffset(), 1);
+                }
             }
 
             // Get non-empty description