78 lines
3.5 KiB
Python
78 lines
3.5 KiB
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.
|
|
"""ANTA Inventory unit tests."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
from typing import TYPE_CHECKING
|
|
|
|
import pytest
|
|
from pydantic import ValidationError
|
|
|
|
from anta.inventory import AntaInventory
|
|
from anta.inventory.exceptions import InventoryIncorrectSchemaError, InventoryRootKeyError
|
|
|
|
if TYPE_CHECKING:
|
|
from _pytest.mark.structures import ParameterSet
|
|
|
|
FILE_DIR: Path = Path(__file__).parent.parent.resolve() / "data" / "inventory"
|
|
|
|
|
|
INIT_VALID_PARAMS: list[ParameterSet] = [
|
|
pytest.param(
|
|
{"anta_inventory": {"hosts": [{"host": "192.168.0.17"}, {"host": "192.168.0.2"}, {"host": "my.awesome.host.com"}]}},
|
|
id="Inventory_with_host_only",
|
|
),
|
|
pytest.param({"anta_inventory": {"networks": [{"network": "192.168.0.0/24"}]}}, id="ValidInventory_with_networks_only"),
|
|
pytest.param(
|
|
{"anta_inventory": {"ranges": [{"start": "10.0.0.1", "end": "10.0.0.11"}, {"start": "10.0.0.101", "end": "10.0.0.111"}]}},
|
|
id="Inventory_with_ranges_only",
|
|
),
|
|
pytest.param(
|
|
{"anta_inventory": {"hosts": [{"host": "192.168.0.17", "port": 443}, {"host": "192.168.0.2", "port": 80}]}},
|
|
id="Inventory_with_host_port",
|
|
),
|
|
pytest.param(
|
|
{"anta_inventory": {"hosts": [{"host": "192.168.0.17", "tags": ["leaf"]}, {"host": "192.168.0.2", "tags": ["spine"]}]}},
|
|
id="Inventory_with_host_tags",
|
|
),
|
|
pytest.param({"anta_inventory": {"networks": [{"network": "192.168.0.0/24", "tags": ["leaf"]}]}}, id="ValidInventory_with_networks_tags"),
|
|
pytest.param(
|
|
{
|
|
"anta_inventory": {
|
|
"ranges": [{"start": "10.0.0.1", "end": "10.0.0.11", "tags": ["leaf"]}, {"start": "10.0.0.101", "end": "10.0.0.111", "tags": ["spine"]}]
|
|
}
|
|
},
|
|
id="Inventory_with_ranges_tags",
|
|
),
|
|
]
|
|
|
|
|
|
INIT_INVALID_PARAMS = [
|
|
pytest.param({"anta_inventory": {"hosts": [{"host": "192.168.0.17/32"}, {"host": "192.168.0.2"}]}}, id="Inventory_with_host_only"),
|
|
pytest.param({"anta_inventory": {"networks": [{"network": "192.168.42.0/8"}]}}, id="Inventory_wrong_network_bits"),
|
|
pytest.param({"anta_inventory": {"networks": [{"network": "toto"}]}}, id="Inventory_wrong_network"),
|
|
pytest.param({"anta_inventory": {"ranges": [{"start": "toto", "end": "192.168.42.42"}]}}, id="Inventory_wrong_range"),
|
|
pytest.param({"anta_inventory": {"ranges": [{"start": "fe80::cafe", "end": "192.168.42.42"}]}}, id="Inventory_wrong_range_type_mismatch"),
|
|
pytest.param(
|
|
{"inventory": {"ranges": [{"start": "10.0.0.1", "end": "10.0.0.11"}, {"start": "10.0.0.100", "end": "10.0.0.111"}]}},
|
|
id="Invalid_Root_Key",
|
|
),
|
|
]
|
|
|
|
|
|
class TestAntaInventory:
|
|
"""Tests for anta.inventory.AntaInventory."""
|
|
|
|
@pytest.mark.parametrize("yaml_file", INIT_VALID_PARAMS, indirect=["yaml_file"])
|
|
def test_parse_valid(self, yaml_file: Path) -> None:
|
|
"""Parse valid YAML file to create ANTA inventory."""
|
|
AntaInventory.parse(filename=yaml_file, username="arista", password="arista123")
|
|
|
|
@pytest.mark.parametrize("yaml_file", INIT_INVALID_PARAMS, indirect=["yaml_file"])
|
|
def test_parse_invalid(self, yaml_file: Path) -> None:
|
|
"""Parse invalid YAML file to create ANTA inventory."""
|
|
with pytest.raises((InventoryIncorrectSchemaError, InventoryRootKeyError, ValidationError)):
|
|
AntaInventory.parse(filename=yaml_file, username="arista", password="arista123")
|