1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
# 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.
"""ANTA Result Manager models unit tests."""
from __future__ import annotations
from typing import TYPE_CHECKING, Callable
import pytest
from anta.result_manager.models import AntaTestStatus
from tests.units.conftest import DEVICE_NAME
if TYPE_CHECKING:
from _pytest.mark.structures import ParameterSet
# Import as Result to avoid pytest collection
from anta.result_manager.models import TestResult as Result
TEST_RESULT_SET_STATUS: list[ParameterSet] = [
pytest.param(AntaTestStatus.SUCCESS, "test success message", id="set_success"),
pytest.param(AntaTestStatus.ERROR, "test error message", id="set_error"),
pytest.param(AntaTestStatus.FAILURE, "test failure message", id="set_failure"),
pytest.param(AntaTestStatus.SKIPPED, "test skipped message", id="set_skipped"),
pytest.param(AntaTestStatus.UNSET, "test unset message", id="set_unset"),
]
class TestTestResultModels:
"""Test components of anta.result_manager.models."""
@pytest.mark.parametrize(("target", "message"), TEST_RESULT_SET_STATUS)
def test__is_status_foo(self, test_result_factory: Callable[[int], Result], target: AntaTestStatus, message: str) -> None:
"""Test TestResult.is_foo methods."""
testresult = test_result_factory(1)
assert testresult.result == AntaTestStatus.UNSET
assert len(testresult.messages) == 0
if target == AntaTestStatus.SUCCESS:
testresult.is_success(message)
assert testresult.result == "success"
assert message in testresult.messages
if target == AntaTestStatus.FAILURE:
testresult.is_failure(message)
assert testresult.result == "failure"
assert message in testresult.messages
if target == AntaTestStatus.ERROR:
testresult.is_error(message)
assert testresult.result == "error"
assert message in testresult.messages
if target == AntaTestStatus.SKIPPED:
testresult.is_skipped(message)
assert testresult.result == "skipped"
assert message in testresult.messages
if target == AntaTestStatus.UNSET:
# no helper for unset, testing _set_status
testresult._set_status(AntaTestStatus.UNSET, message)
assert testresult.result == "unset"
assert message in testresult.messages
@pytest.mark.parametrize(("target", "message"), TEST_RESULT_SET_STATUS)
def test____str__(self, test_result_factory: Callable[[int], Result], target: AntaTestStatus, message: str) -> None:
"""Test TestResult.__str__."""
testresult = test_result_factory(1)
assert testresult.result == AntaTestStatus.UNSET
assert len(testresult.messages) == 0
testresult._set_status(target, message)
assert testresult.result == target
assert str(testresult) == f"Test 'VerifyTest1' (on '{DEVICE_NAME}'): Result '{target}'\nMessages: {[message]}"
|
