explicitly stop driver in these tests (ydb-platform#11290)

shnikd · Nov 5, 2024 · 9b41ac2 · 9b41ac2
1 parent f8cbfb3
commit 9b41ac2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/ydb/core/kqp/ut/common/kqp_ut_common.h b/ydb/core/kqp/ut/common/kqp_ut_common.h
@@ -145,6 +145,8 @@ class TKikimrRunner {
     TKikimrRunner(const NFake::TStorage& storage);
 
     ~TKikimrRunner() {
+        Server->GetRuntime()->SetObserverFunc(TTestActorRuntime::DefaultObserverFunc);
+
         RunCall([&] { Driver->Stop(true); return false; });
         if (ThreadPoolStarted_) {
             ThreadPool.Stop();
@@ -156,6 +158,7 @@ class TKikimrRunner {
         Client.Reset();
     }
 
+    NYdb::TDriver* GetDriverMut() { return Driver.Get(); }
     const TString& GetEndpoint() const { return Endpoint; }
     const NYdb::TDriver& GetDriver() const { return *Driver; }
     NYdb::NScheme::TSchemeClient GetSchemeClient() const { return NYdb::NScheme::TSchemeClient(*Driver); }