arista/anta
1
0
Fork 0
Code Issues Activity

Merging upstream version 1.3.0.

Browse source 
Signed-off-by: Daniel Baumann <daniel@debian.org>
This commit is contained in:
Daniel Baumann 2025-03-17 07:33:51 +01:00
parent 5b922100c9
commit 8a6a3342fc
Signed by: daniel
GPG key ID: FBB4F0E80A80222F
337 changed files with 16571 additions and 4891 deletions
Download patch file Download diff file Expand all files Collapse all files

192
tests/units/input_models/test_snmp.py Normal file
View file

@ -0,0 +1,192 @@
# 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.snmp.py."""
# pylint: disable=C0302
from __future__ import annotations
from typing import TYPE_CHECKING
import pytest
from pydantic import ValidationError
from anta.input_models.snmp import SnmpGroup
from anta.tests.snmp import VerifySnmpNotificationHost, VerifySnmpUser
if TYPE_CHECKING:
from anta.custom_types import SnmpVersion, SnmpVersionV3AuthType
from anta.input_models.snmp import SnmpHost, SnmpUser
class TestVerifySnmpUserInput:
"""Test anta.tests.snmp.VerifySnmpUser.Input."""
@pytest.mark.parametrize(
("snmp_users"),
[
pytest.param([{"username": "test", "group_name": "abc", "version": "v1", "auth_type": None, "priv_type": None}], id="valid-v1"),
pytest.param([{"username": "test", "group_name": "abc", "version": "v2c", "auth_type": None, "priv_type": None}], id="valid-v2c"),
pytest.param([{"username": "test", "group_name": "abc", "version": "v3", "auth_type": "SHA", "priv_type": "AES-128"}], id="valid-v3"),
],
)
def test_valid(self, snmp_users: list[SnmpUser]) -> None:
"""Test VerifySnmpUser.Input valid inputs."""
VerifySnmpUser.Input(snmp_users=snmp_users)
@pytest.mark.parametrize(
("snmp_users"),
[
pytest.param([{"username": "test", "group_name": "abc", "version": "v3", "auth_type": None, "priv_type": None}], id="invalid-v3"),
],
)
def test_invalid(self, snmp_users: list[SnmpUser]) -> None:
"""Test VerifySnmpUser.Input invalid inputs."""
with pytest.raises(ValidationError):
VerifySnmpUser.Input(snmp_users=snmp_users)
class TestSnmpHost:
"""Test anta.input_models.snmp.SnmpHost."""
@pytest.mark.parametrize(
("notification_hosts"),
[
pytest.param(
[
{
"hostname": "192.168.1.100",
"vrf": "test",
"notification_type": "trap",
"version": "v1",
"udp_port": 162,
"community_string": "public",
"user": None,
}
],
id="valid-v1",
),
pytest.param(
[
{
"hostname": "192.168.1.100",
"vrf": "test",
"notification_type": "trap",
"version": "v2c",
"udp_port": 162,
"community_string": "public",
"user": None,
}
],
id="valid-v2c",
),
pytest.param(
[
{
"hostname": "192.168.1.100",
"vrf": "test",
"notification_type": "trap",
"version": "v3",
"udp_port": 162,
"community_string": None,
"user": "public",
}
],
id="valid-v3",
),
],
)
def test_valid(self, notification_hosts: list[SnmpHost]) -> None:
"""Test VerifySnmpNotificationHost.Input valid inputs."""
VerifySnmpNotificationHost.Input(notification_hosts=notification_hosts)
@pytest.mark.parametrize(
("notification_hosts"),
[
pytest.param(
[
{
"hostname": "192.168.1.100",
"vrf": "test",
"notification_type": "trap",
"version": None,
"udp_port": 162,
"community_string": None,
"user": None,
}
],
id="invalid-version",
),
pytest.param(
[
{
"hostname": "192.168.1.100",
"vrf": "test",
"notification_type": "trap",
"version": "v1",
"udp_port": 162,
"community_string": None,
"user": None,
}
],
id="invalid-community-string-version-v1",
),
pytest.param(
[
{
"hostname": "192.168.1.100",
"vrf": "test",
"notification_type": "trap",
"version": "v2c",
"udp_port": 162,
"community_string": None,
"user": None,
}
],
id="invalid-community-string-version-v2c",
),
pytest.param(
[
{
"hostname": "192.168.1.100",
"vrf": "test",
"notification_type": "trap",
"version": "v3",
"udp_port": 162,
"community_string": None,
"user": None,
}
],
id="invalid-user-version-v3",
),
],
)
def test_invalid(self, notification_hosts: list[SnmpHost]) -> None:
"""Test VerifySnmpNotificationHost.Input invalid inputs."""
with pytest.raises(ValidationError):
VerifySnmpNotificationHost.Input(notification_hosts=notification_hosts)
class TestSnmpGroupInput:
"""Test anta.input_models.snmp.SnmpGroup."""
@pytest.mark.parametrize(
("group_name", "version", "read_view", "write_view", "notify_view", "authentication"),
[
pytest.param("group1", "v3", "", "write_1", None, "auth", id="snmp-auth"),
],
)
def test_valid(self, group_name: str, read_view: str, version: SnmpVersion, write_view: str, notify_view: str, authentication: SnmpVersionV3AuthType) -> None:
"""Test SnmpGroup valid inputs."""
SnmpGroup(group_name=group_name, version=version, read_view=read_view, write_view=write_view, notify_view=notify_view, authentication=authentication)
@pytest.mark.parametrize(
("group_name", "version", "read_view", "write_view", "notify_view", "authentication"),
[
pytest.param("group1", "v3", "", "write_1", None, None, id="snmp-invalid-auth"),
],
)
def test_invalid(self, group_name: str, read_view: str, version: SnmpVersion, write_view: str, notify_view: str, authentication: SnmpVersionV3AuthType) -> None:
"""Test SnmpGroup invalid inputs."""
with pytest.raises(ValidationError):
SnmpGroup(group_name=group_name, version=version, read_view=read_view, write_view=write_view, notify_view=notify_view, authentication=authentication)