48 lines
1.9 KiB
Python
48 lines
1.9 KiB
Python
# 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)
|