properly stop ycsb kqp upsert actor and info collector

2 files changed, 5 insertions, 2 deletions

diff --git a/ydb/core/load_test/ycsb/kqp_upsert.cpp b/ydb/core/load_test/ycsb/kqp_upsert.cpp
index 5fa970b116..ecfd3c28a0 100644
--- a/ydb/core/load_test/ycsb/kqp_upsert.cpp
+++ b/ydb/core/load_test/ycsb/kqp_upsert.cpp
@@ -398,8 +398,7 @@ private:
     void HandlePoison(const TActorContext& ctx) {
         LOG_DEBUG_S(ctx, NKikimrServices::DS_LOAD_TEST, "TKqpUpsertActorMultiSession# " << Tag
             << " tablet recieved PoisonPill, going to die");
-        // TODO: stop subactors?
-        Die(ctx);
+        Stop(ctx);
     }
 
     void StopWithError(const TActorContext& ctx, const TString& reason) {
diff --git a/ydb/core/load_test/ycsb/test_load_actor.cpp b/ydb/core/load_test/ycsb/test_load_actor.cpp
index d40c8c9415..1bc0d52491 100644
--- a/ydb/core/load_test/ycsb/test_load_actor.cpp
+++ b/ydb/core/load_test/ycsb/test_load_actor.cpp
@@ -552,6 +552,10 @@ public:
             ctx.Send(actorId, new TEvents::TEvPoison());
         }
 
+        if (HttpInfoCollector) {
+            ctx.Send(HttpInfoCollector, new TEvents::TEvPoison());
+        }
+
         CloseSession(ctx);
 
         Die(ctx);