YQ-3557 fixed unit test KqpLimits::ComputeActorMemoryAllocationFailur…

…eQueryService with resource pools (ydb-platform#7873)
qyryq · Aug 16, 2024 · d569f4b · d569f4b
1 parent 8796c8a
commit d569f4b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/ydb/core/kqp/rm_service/kqp_rm_service.cpp b/ydb/core/kqp/rm_service/kqp_rm_service.cpp
@@ -299,10 +299,9 @@ class TKqpResourceManager : public IKqpResourceManager {
                         auto it = MemoryNamedPools.find(tx->MakePoolId());
                         if (it != MemoryNamedPools.end()) {
                             it->second->Release(resources.Memory);
-                        }
-
-                        if (it->second->GetUsed() == 0) {
-                            MemoryNamedPools.erase(it);
+                            if (it->second->GetUsed() == 0) {
+                                MemoryNamedPools.erase(it);
+                            }
                         }
                     }
                 }

diff --git a/ydb/core/kqp/ut/query/kqp_limits_ut.cpp b/ydb/core/kqp/ut/query/kqp_limits_ut.cpp
@@ -192,6 +192,8 @@ Y_UNIT_TEST_SUITE(KqpLimits) {
 
         app.MutableResourceBrokerConfig()->CopyFrom(MakeResourceBrokerTestConfig(4));
 
+        app.MutableFeatureFlags()->SetEnableResourcePools(true);
+
         TKikimrRunner kikimr(app);
         CreateLargeTable(kikimr, 0, 0, 0);