diff --git a/smoke_testing/test_version.py b/smoke_testing/test_version.py index bf075027..22600aa7 100644 --- a/smoke_testing/test_version.py +++ b/smoke_testing/test_version.py @@ -8,7 +8,9 @@ def run(kubescape_exec: str): ver = os.getenv("RELEASE") msg = smoke_utils.run_command(command=[kubescape_exec, "version"]) - assert ver in msg, f"expected version: {ver}, found: {msg}" + if isinstance(msg, bytes): + msg = msg.decode('utf-8') + assert (ver and ver in msg) or (ver and ver.lstrip('v') in msg), f"expected version: {ver}, found: {msg}" print("Done testing version")