From 2a00311b58515e3c5f0c57e26eeabb77837930e0 Mon Sep 17 00:00:00 2001 From: Alexander Popov Date: Mon, 18 Sep 2023 08:58:44 +0300 Subject: [PATCH] Refactor the assertion in colorize_result() to improve test coverage --- kernel_hardening_checker/engine.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/kernel_hardening_checker/engine.py b/kernel_hardening_checker/engine.py index 4fdc222..64be204 100644 --- a/kernel_hardening_checker/engine.py +++ b/kernel_hardening_checker/engine.py @@ -20,10 +20,9 @@ def colorize_result(input_text): return input_text if input_text.startswith('OK'): color = GREEN_COLOR - elif input_text.startswith('FAIL:'): - color = RED_COLOR else: - assert(False), f'unexpected result "{input_text}"' + assert(input_text.startswith('FAIL:')), f'unexpected result "{input_text}"' + color = RED_COLOR return f'{color}{input_text}{COLOR_END}' -- 2.31.1