From d0eb392acc5e91d85d0fda33d2ae5a0bddca23a0 Mon Sep 17 00:00:00 2001
From: Tim Pillinger <26465611+wxtim@users.noreply.github.com>
Date: Fri, 12 Jan 2024 10:57:41 +0000
Subject: [PATCH] ensure logger set for all caplog.set_level

---
 tests/unit/test_flow_mgr.py | 3 ++-
 tests/unit/test_id_cli.py   | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/tests/unit/test_flow_mgr.py b/tests/unit/test_flow_mgr.py
index 765fdf0c522..5fea08c1b97 100644
--- a/tests/unit/test_flow_mgr.py
+++ b/tests/unit/test_flow_mgr.py
@@ -22,6 +22,7 @@
 
 from cylc.flow.flow_mgr import FlowMgr
 from cylc.flow.workflow_db_mgr import WorkflowDatabaseManager
+from cylc.flow import CYLC_LOG
 
 
 FAKE_NOW = datetime.datetime(2020, 12, 25, 17, 5, 55)
@@ -44,7 +45,7 @@ def test_all(
 ):
     db_mgr = WorkflowDatabaseManager()
     flow_mgr = FlowMgr(db_mgr)
-    caplog.set_level(logging.INFO)
+    caplog.set_level(logging.INFO, CYLC_LOG)
 
     count = 1
     meta = "the quick brown fox"
diff --git a/tests/unit/test_id_cli.py b/tests/unit/test_id_cli.py
index 7649e9fa0f2..3db329436bf 100644
--- a/tests/unit/test_id_cli.py
+++ b/tests/unit/test_id_cli.py
@@ -19,6 +19,7 @@
 from pathlib import Path
 import pytest
 
+from cylc.flow import CYLC_LOG
 from cylc.flow.async_util import pipe
 from cylc.flow.exceptions import InputError, WorkflowFilesError
 from cylc.flow.id import detokenise, tokenise, Tokens
@@ -537,7 +538,7 @@ def test_validate_workflow_ids_basic(tmp_run_dir):
 
 def test_validate_workflow_ids_warning(caplog):
     """It should warn when the run number is provided as a cycle point."""
-    caplog.set_level(logging.WARN)
+    caplog.set_level(logging.WARN, CYLC_LOG)
     _validate_workflow_ids(Tokens('workflow/run1//cycle/task'), src_path='')
     assert caplog.messages == []