Fix comp nodes logging in case when it's disabled (#16837)

Author: lll-phill-lll

ydb/core/kqp/ut/runtime/kqp_scan_logging_ut.cpp

@@ -31,13 +31,15 @@ void FillTableWithData(NQuery::TQueryClient& db, ui64 numRows=300) {
     }
 }
 
-void RunTestForQuery(const std::string& query, const std::string& expectedLog) {
+void RunTestForQuery(const std::string& query, const std::string& expectedLog, bool enabledLogs) {
     TStringStream logsStream;
 
     Cerr << "cwd: " << NFs::CurrentWorkingDirectory() << Endl;
     TKikimrRunner kikimr(AppSettings(logsStream));
 
-    kikimr.GetTestServer().GetRuntime()->SetLogPriority(NKikimrServices::KQP_TASKS_RUNNER, NActors::NLog::PRI_DEBUG);
+    if (enabledLogs) {
+        kikimr.GetTestServer().GetRuntime()->SetLogPriority(NKikimrServices::KQP_TASKS_RUNNER, NActors::NLog::PRI_DEBUG);
+    }
 
     auto db = kikimr.GetQueryClient();
 
@@ -64,25 +66,27 @@ void RunTestForQuery(const std::string& query, const std::string& expectedLog) {
             break;
         }
     }
-    // TODO: Uncomment after: https://github.com/ydb-platform/ydb/issues/15597
-    Y_UNUSED(hasExpectedLog);
-    // UNIT_ASSERT(hasExpectedLog);
+
+    // TODO: remove this if after https://github.com/ydb-platform/ydb/issues/15597
+    if (!enabledLogs) {
+        UNIT_ASSERT(hasExpectedLog == enabledLogs);
+    }
 }
 
 } // anonymous namespace
 
 Y_UNIT_TEST_SUITE(KqpScanLogs) {
 
-Y_UNIT_TEST(WideCombine) {
+Y_UNIT_TEST_TWIN(WideCombine, EnabledLogs) {
     auto query = R"(
         --!syntax_v1
         select count(t.Key) from `/Root/KeyValue` as t group by t.Value
     )";
 
-    RunTestForQuery(query, "[WideCombine]");
+    RunTestForQuery(query, "[WideCombine]", EnabledLogs);
 }
 
-Y_UNIT_TEST(GraceJoin) {
+Y_UNIT_TEST_TWIN(GraceJoin, EnabledLogs) {
     auto query = R"(
         --!syntax_v1
         PRAGMA ydb.CostBasedOptimizationLevel='0';
@@ -92,7 +96,7 @@ Y_UNIT_TEST(GraceJoin) {
         order by t1.Value
     )";
 
-    RunTestForQuery(query, "[GraceJoin]");
+    RunTestForQuery(query, "[GraceJoin]", EnabledLogs);
 }
 
 

ydb/library/yql/dq/runtime/dq_tasks_runner.cpp

@@ -264,9 +264,8 @@ class TDqTaskRunner : public IDqTaskRunner {
             logProviderFunc = [log=LogFunc](const NUdf::TStringRef& component, NUdf::ELogLevel level, const NUdf::TStringRef& message) {
                 log(TStringBuilder() << "[" << component << "][" << level << "]: " << message << "\n");
             };
+            ComputationLogProvider = NUdf::MakeLogProvider(std::move(logProviderFunc), NUdf::ELogLevel::Debug);
         }
-
-        ComputationLogProvider = NUdf::MakeLogProvider(std::move(logProviderFunc), NUdf::ELogLevel::Debug);
     }
 
     ~TDqTaskRunner() {