From c5e84a8f127b91130cc1d2c1648240021136b989 Mon Sep 17 00:00:00 2001 From: Tim Pillinger <26465611+wxtim@users.noreply.github.com> Date: Thu, 14 Nov 2024 15:15:34 +0000 Subject: [PATCH] Update tests/unit/run_modes/test_skip_units.py Co-authored-by: Ronnie Dutta <61982285+MetRonnie@users.noreply.github.com> --- tests/unit/run_modes/test_skip_units.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/tests/unit/run_modes/test_skip_units.py b/tests/unit/run_modes/test_skip_units.py index 54f7d96969..0b5316662a 100644 --- a/tests/unit/run_modes/test_skip_units.py +++ b/tests/unit/run_modes/test_skip_units.py @@ -129,8 +129,12 @@ def test_skip_mode_validate(caplog, log_filter): skip_mode_validate(taskdefs) - message = caplog.messages[0] - - assert 'skip mode:\n * skip_task' in message - assert ' live mode' not in message # Avoid matching "non-live mode" - assert 'workflow mode' not in message + assert len(caplog.records) == 1 + assert log_filter( + level=logging.WARNING, + exact_match=( + "The following tasks are set to run in skip mode:\n" + " * skip_task" + ), + log=caplog + )