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
|
# Copyright (c) 2023-2025 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.input_models.system.py."""
from __future__ import annotations
from typing import TYPE_CHECKING
import pytest
from pydantic import ValidationError
from anta.tests.system import VerifyNTPAssociations
if TYPE_CHECKING:
from anta.input_models.system import NTPPool, NTPServer
class TestVerifyNTPAssociationsInput:
"""Test anta.tests.system.VerifyNTPAssociations.Input."""
@pytest.mark.parametrize(
("ntp_servers", "ntp_pool"),
[
pytest.param([{"server_address": "1.1.1.1", "preferred": True, "stratum": 1}], None, id="valid-ntp-server"),
pytest.param(None, {"server_addresses": ["1.1.1.1"], "preferred_stratum_range": [1, 3]}, id="valid-ntp-pool"),
],
)
def test_valid(self, ntp_servers: list[NTPServer], ntp_pool: NTPPool) -> None:
"""Test VerifyNTPAssociations.Input valid inputs."""
VerifyNTPAssociations.Input(ntp_servers=ntp_servers, ntp_pool=ntp_pool)
@pytest.mark.parametrize(
("ntp_servers", "ntp_pool"),
[
pytest.param(
[{"server_address": "1.1.1.1", "preferred": True, "stratum": 1}],
{"server_addresses": ["1.1.1.1"], "preferred_stratum_range": [1, 3]},
id="invalid-both-server-pool",
),
pytest.param(None, {"server_addresses": ["1.1.1.1"], "preferred_stratum_range": [1, 3, 6]}, id="invalid-ntp-pool-stratum"),
pytest.param(None, None, id="invalid-both-none"),
],
)
def test_invalid(self, ntp_servers: list[NTPServer], ntp_pool: NTPPool) -> None:
"""Test VerifyNTPAssociations.Input invalid inputs."""
with pytest.raises(ValidationError):
VerifyNTPAssociations.Input(ntp_servers=ntp_servers, ntp_pool=ntp_pool)
|
