diff options
| -rw-r--r-- | ydb/core/kqp/session_actor/kqp_session_actor.cpp | 5 | ||||
| -rw-r--r-- | ydb/public/sdk/cpp/client/ydb_topic/ut/topic_to_table_ut.cpp | 10 |
2 files changed, 10 insertions, 5 deletions
diff --git a/ydb/core/kqp/session_actor/kqp_session_actor.cpp b/ydb/core/kqp/session_actor/kqp_session_actor.cpp index 9e6db5c5330..5fd19f0ca9b 100644 --- a/ydb/core/kqp/session_actor/kqp_session_actor.cpp +++ b/ydb/core/kqp/session_actor/kqp_session_actor.cpp @@ -2433,6 +2433,10 @@ public: PassAway(); } + void HandleFinalCleanup(TEvKqp::TEvQueryRequest::TPtr& ev) { + ReplyProcessError(ev, Ydb::StatusIds::BAD_SESSION, "Session is under shutdown"); + } + STFUNC(ReadyState) { try { switch (ev->GetTypeRewrite()) { @@ -2567,6 +2571,7 @@ public: hFunc(TEvents::TEvUndelivered, HandleNoop); hFunc(TEvKqpSnapshot::TEvCreateSnapshotResponse, Handle); hFunc(NWorkload::TEvContinueRequest, HandleNoop); + hFunc(TEvKqp::TEvQueryRequest, HandleFinalCleanup); } } catch (const yexception& ex) { InternalError(ex.what()); diff --git a/ydb/public/sdk/cpp/client/ydb_topic/ut/topic_to_table_ut.cpp b/ydb/public/sdk/cpp/client/ydb_topic/ut/topic_to_table_ut.cpp index 9410ebd992f..a18e160149c 100644 --- a/ydb/public/sdk/cpp/client/ydb_topic/ut/topic_to_table_ut.cpp +++ b/ydb/public/sdk/cpp/client/ydb_topic/ut/topic_to_table_ut.cpp @@ -605,11 +605,11 @@ Y_UNIT_TEST_F(Offsets_Cannot_Be_Promoted_When_Reading_In_A_Transaction, TFixture UNIT_ASSERT_EXCEPTION(ReadMessage(reader, {.Tx = tx, .CommitOffsets = true}), yexception); } -//Y_UNIT_TEST_F(WriteToTopic_Invalid_Session, TFixture) -//{ -// WriteToTopicWithInvalidTxId(false); -//} -// +Y_UNIT_TEST_F(WriteToTopic_Invalid_Session, TFixture) +{ + WriteToTopicWithInvalidTxId(false); +} + //Y_UNIT_TEST_F(WriteToTopic_Invalid_Tx, TFixture) //{ // WriteToTopicWithInvalidTxId(true); |