39 lines
998 B
Python
39 lines
998 B
Python
|
# Copyright (c) 2023-2024 Arista Networks, Inc.
|
||
|
|
# Use of this source code is governed by the Apache License 2.0
|
||
|
|
# that can be found in the LICENSE file.
|
||
|
|
"""
|
||
|
|
Tests for anta.tools.misc
|
||
|
|
"""
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from anta.tools.misc import exc_to_str, tb_to_str
|
||
|
|
|
||
|
|
|
||
|
|
def my_raising_function(exception: Exception) -> None:
|
||
|
|
"""
|
||
|
|
dummy function to raise Exception
|
||
|
|
"""
|
||
|
|
raise exception
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("exception, expected_output", [(ValueError("test"), "ValueError (test)"), (ValueError(), "ValueError")])
|
||
|
|
def test_exc_to_str(exception: Exception, expected_output: str) -> None:
|
||
|
|
"""
|
||
|
|
Test exc_to_str
|
||
|
|
"""
|
||
|
|
assert exc_to_str(exception) == expected_output
|
||
|
|
|
||
|
|
|
||
|
|
def test_tb_to_str() -> None:
|
||
|
|
"""
|
||
|
|
Test tb_to_str
|
||
|
|
"""
|
||
|
|
try:
|
||
|
|
my_raising_function(ValueError("test"))
|
||
|
|
except ValueError as e:
|
||
|
|
output = tb_to_str(e)
|
||
|
|
assert "Traceback" in output
|
||
|
|
assert 'my_raising_function(ValueError("test"))' in output
|