Fix death process in case of IncreaseSessions

1 files changed, 5 insertions, 1 deletions

diff --git a/ydb/core/load_test/kqp.cpp b/ydb/core/load_test/kqp.cpp
index 12a788c7f1..f1ac0c8a93 100644
--- a/ydb/core/load_test/kqp.cpp
+++ b/ydb/core/load_test/kqp.cpp
@@ -337,6 +337,10 @@ public:
     }
 
     void HandleWakeup(const TActorContext& ctx) {
+        if (ResultsReceived) {
+            // if death process is started, then brake wakeup circuit
+            return;
+        }
         size_t targetSessions;
         if (IncreaseSessions) {
             targetSessions = 1 + NumOfSessions * (TInstant::Now() - Start).Seconds() / DurationSeconds;
@@ -458,7 +462,7 @@ private:
 
         Total->Add(response->Data);
         ++ResultsReceived;
-        if (ResultsReceived == NumOfSessions) {
+        if (ResultsReceived == Workers.size()) {
             StartDeathProcess(ctx);
         }
     }