391 lines
13 KiB
Python
391 lines
13 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 models unit tests."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
from typing import Any
|
|
|
|
import pytest
|
|
from pydantic import ValidationError
|
|
|
|
from anta.device import AsyncEOSDevice
|
|
from anta.inventory.models import AntaInventoryHost, AntaInventoryInput, AntaInventoryNetwork, AntaInventoryRange
|
|
from tests.data.json_data import (
|
|
INVENTORY_DEVICE_MODEL_INVALID,
|
|
INVENTORY_DEVICE_MODEL_VALID,
|
|
INVENTORY_MODEL_HOST_CACHE,
|
|
INVENTORY_MODEL_HOST_INVALID,
|
|
INVENTORY_MODEL_HOST_VALID,
|
|
INVENTORY_MODEL_INVALID,
|
|
INVENTORY_MODEL_NETWORK_CACHE,
|
|
INVENTORY_MODEL_NETWORK_INVALID,
|
|
INVENTORY_MODEL_NETWORK_VALID,
|
|
INVENTORY_MODEL_RANGE_CACHE,
|
|
INVENTORY_MODEL_RANGE_INVALID,
|
|
INVENTORY_MODEL_RANGE_VALID,
|
|
INVENTORY_MODEL_VALID,
|
|
)
|
|
from tests.lib.utils import generate_test_ids_dict
|
|
|
|
|
|
class TestInventoryUnitModels:
|
|
"""Test components of AntaInventoryInput model."""
|
|
|
|
@pytest.mark.parametrize("test_definition", INVENTORY_MODEL_HOST_VALID, ids=generate_test_ids_dict)
|
|
def test_anta_inventory_host_valid(self, test_definition: dict[str, Any]) -> None:
|
|
"""Test host input model.
|
|
|
|
Test structure:
|
|
---------------
|
|
|
|
{
|
|
'name': 'ValidIPv4_Host',
|
|
'input': '1.1.1.1',
|
|
'expected_result': 'valid'
|
|
}
|
|
|
|
"""
|
|
try:
|
|
host_inventory = AntaInventoryHost(host=test_definition["input"])
|
|
except ValidationError as exc:
|
|
logging.warning("Error: %s", str(exc))
|
|
raise AssertionError from exc
|
|
assert test_definition["input"] == str(host_inventory.host)
|
|
|
|
@pytest.mark.parametrize("test_definition", INVENTORY_MODEL_HOST_INVALID, ids=generate_test_ids_dict)
|
|
def test_anta_inventory_host_invalid(self, test_definition: dict[str, Any]) -> None:
|
|
"""Test host input model.
|
|
|
|
Test structure:
|
|
---------------
|
|
|
|
{
|
|
'name': 'ValidIPv4_Host',
|
|
'input': '1.1.1.1/32',
|
|
'expected_result': 'invalid'
|
|
}
|
|
|
|
"""
|
|
with pytest.raises(ValidationError):
|
|
AntaInventoryHost(host=test_definition["input"])
|
|
|
|
@pytest.mark.parametrize("test_definition", INVENTORY_MODEL_HOST_CACHE, ids=generate_test_ids_dict)
|
|
def test_anta_inventory_host_cache(self, test_definition: dict[str, Any]) -> None:
|
|
"""Test host disable_cache.
|
|
|
|
Test structure:
|
|
---------------
|
|
|
|
{
|
|
'name': 'Cache',
|
|
'input': {"host": '1.1.1.1', "disable_cache": True},
|
|
'expected_result': True
|
|
}
|
|
|
|
"""
|
|
if "disable_cache" in test_definition["input"]:
|
|
host_inventory = AntaInventoryHost(host=test_definition["input"]["host"], disable_cache=test_definition["input"]["disable_cache"])
|
|
else:
|
|
host_inventory = AntaInventoryHost(host=test_definition["input"]["host"])
|
|
assert test_definition["expected_result"] == host_inventory.disable_cache
|
|
|
|
@pytest.mark.parametrize("test_definition", INVENTORY_MODEL_NETWORK_VALID, ids=generate_test_ids_dict)
|
|
def test_anta_inventory_network_valid(self, test_definition: dict[str, Any]) -> None:
|
|
"""Test Network input model with valid data.
|
|
|
|
Test structure:
|
|
---------------
|
|
|
|
{
|
|
'name': 'ValidIPv4_Subnet',
|
|
'input': '1.1.1.0/24',
|
|
'expected_result': 'valid'
|
|
}
|
|
|
|
"""
|
|
try:
|
|
network_inventory = AntaInventoryNetwork(network=test_definition["input"])
|
|
except ValidationError as exc:
|
|
logging.warning("Error: %s", str(exc))
|
|
raise AssertionError from exc
|
|
assert test_definition["input"] == str(network_inventory.network)
|
|
|
|
@pytest.mark.parametrize("test_definition", INVENTORY_MODEL_NETWORK_INVALID, ids=generate_test_ids_dict)
|
|
def test_anta_inventory_network_invalid(self, test_definition: dict[str, Any]) -> None:
|
|
"""Test Network input model with invalid data.
|
|
|
|
Test structure:
|
|
---------------
|
|
|
|
{
|
|
'name': 'ValidIPv4_Subnet',
|
|
'input': '1.1.1.0/16',
|
|
'expected_result': 'invalid'
|
|
}
|
|
|
|
"""
|
|
try:
|
|
AntaInventoryNetwork(network=test_definition["input"])
|
|
except ValidationError as exc:
|
|
logging.warning("Error: %s", str(exc))
|
|
else:
|
|
raise AssertionError
|
|
|
|
@pytest.mark.parametrize("test_definition", INVENTORY_MODEL_NETWORK_CACHE, ids=generate_test_ids_dict)
|
|
def test_anta_inventory_network_cache(self, test_definition: dict[str, Any]) -> None:
|
|
"""Test network disable_cache.
|
|
|
|
Test structure:
|
|
---------------
|
|
|
|
{
|
|
'name': 'Cache',
|
|
'input': {"network": '1.1.1.1/24', "disable_cache": True},
|
|
'expected_result': True
|
|
}
|
|
|
|
"""
|
|
if "disable_cache" in test_definition["input"]:
|
|
network_inventory = AntaInventoryNetwork(network=test_definition["input"]["network"], disable_cache=test_definition["input"]["disable_cache"])
|
|
else:
|
|
network_inventory = AntaInventoryNetwork(network=test_definition["input"]["network"])
|
|
assert test_definition["expected_result"] == network_inventory.disable_cache
|
|
|
|
@pytest.mark.parametrize("test_definition", INVENTORY_MODEL_RANGE_VALID, ids=generate_test_ids_dict)
|
|
def test_anta_inventory_range_valid(self, test_definition: dict[str, Any]) -> None:
|
|
"""Test range input model.
|
|
|
|
Test structure:
|
|
---------------
|
|
|
|
{
|
|
'name': 'ValidIPv4_Range',
|
|
'input': {'start':'10.1.0.1', 'end':'10.1.0.10'},
|
|
'expected_result': 'valid'
|
|
}
|
|
|
|
"""
|
|
try:
|
|
range_inventory = AntaInventoryRange(
|
|
start=test_definition["input"]["start"],
|
|
end=test_definition["input"]["end"],
|
|
)
|
|
except ValidationError as exc:
|
|
logging.warning("Error: %s", str(exc))
|
|
raise AssertionError from exc
|
|
assert test_definition["input"]["start"] == str(range_inventory.start)
|
|
assert test_definition["input"]["end"] == str(range_inventory.end)
|
|
|
|
@pytest.mark.parametrize("test_definition", INVENTORY_MODEL_RANGE_INVALID, ids=generate_test_ids_dict)
|
|
def test_anta_inventory_range_invalid(self, test_definition: dict[str, Any]) -> None:
|
|
"""Test range input model.
|
|
|
|
Test structure:
|
|
---------------
|
|
|
|
{
|
|
'name': 'ValidIPv4_Range',
|
|
'input': {'start':'10.1.0.1', 'end':'10.1.0.10/32'},
|
|
'expected_result': 'invalid'
|
|
}
|
|
|
|
"""
|
|
try:
|
|
AntaInventoryRange(
|
|
start=test_definition["input"]["start"],
|
|
end=test_definition["input"]["end"],
|
|
)
|
|
except ValidationError as exc:
|
|
logging.warning("Error: %s", str(exc))
|
|
else:
|
|
raise AssertionError
|
|
|
|
@pytest.mark.parametrize("test_definition", INVENTORY_MODEL_RANGE_CACHE, ids=generate_test_ids_dict)
|
|
def test_anta_inventory_range_cache(self, test_definition: dict[str, Any]) -> None:
|
|
"""Test range disable_cache.
|
|
|
|
Test structure:
|
|
---------------
|
|
|
|
{
|
|
'name': 'Cache',
|
|
'input': {"start": '1.1.1.1', "end": "1.1.1.10", "disable_cache": True},
|
|
'expected_result': True
|
|
}
|
|
|
|
"""
|
|
if "disable_cache" in test_definition["input"]:
|
|
range_inventory = AntaInventoryRange(
|
|
start=test_definition["input"]["start"],
|
|
end=test_definition["input"]["end"],
|
|
disable_cache=test_definition["input"]["disable_cache"],
|
|
)
|
|
else:
|
|
range_inventory = AntaInventoryRange(start=test_definition["input"]["start"], end=test_definition["input"]["end"])
|
|
assert test_definition["expected_result"] == range_inventory.disable_cache
|
|
|
|
|
|
class TestAntaInventoryInputModel:
|
|
"""Unit test of AntaInventoryInput model."""
|
|
|
|
def test_inventory_input_structure(self) -> None:
|
|
"""Test inventory keys are those expected."""
|
|
inventory = AntaInventoryInput()
|
|
logging.info("Inventory keys are: %s", str(inventory.model_dump().keys()))
|
|
assert all(elem in inventory.model_dump() for elem in ["hosts", "networks", "ranges"])
|
|
|
|
@pytest.mark.parametrize("inventory_def", INVENTORY_MODEL_VALID, ids=generate_test_ids_dict)
|
|
def test_anta_inventory_intput_valid(self, inventory_def: dict[str, Any]) -> None:
|
|
"""Test loading valid data to inventory class.
|
|
|
|
Test structure:
|
|
---------------
|
|
|
|
{
|
|
"name": "Valid_Host_Only",
|
|
"input": {
|
|
"hosts": [
|
|
{
|
|
"host": "192.168.0.17"
|
|
},
|
|
{
|
|
"host": "192.168.0.2"
|
|
}
|
|
]
|
|
},
|
|
"expected_result": "valid"
|
|
}
|
|
|
|
"""
|
|
try:
|
|
inventory = AntaInventoryInput(**inventory_def["input"])
|
|
except ValidationError as exc:
|
|
logging.warning("Error: %s", str(exc))
|
|
raise AssertionError from exc
|
|
logging.info("Checking if all root keys are correctly lodaded")
|
|
assert all(elem in inventory.model_dump() for elem in inventory_def["input"])
|
|
|
|
@pytest.mark.parametrize("inventory_def", INVENTORY_MODEL_INVALID, ids=generate_test_ids_dict)
|
|
def test_anta_inventory_intput_invalid(self, inventory_def: dict[str, Any]) -> None:
|
|
"""Test loading invalid data to inventory class.
|
|
|
|
Test structure:
|
|
---------------
|
|
|
|
{
|
|
"name": "Valid_Host_Only",
|
|
"input": {
|
|
"hosts": [
|
|
{
|
|
"host": "192.168.0.17"
|
|
},
|
|
{
|
|
"host": "192.168.0.2/32"
|
|
}
|
|
]
|
|
},
|
|
"expected_result": "invalid"
|
|
}
|
|
|
|
"""
|
|
try:
|
|
if "hosts" in inventory_def["input"]:
|
|
logging.info(
|
|
"Loading %s into AntaInventoryInput hosts section",
|
|
str(inventory_def["input"]["hosts"]),
|
|
)
|
|
AntaInventoryInput(hosts=inventory_def["input"]["hosts"])
|
|
if "networks" in inventory_def["input"]:
|
|
logging.info(
|
|
"Loading %s into AntaInventoryInput networks section",
|
|
str(inventory_def["input"]["networks"]),
|
|
)
|
|
AntaInventoryInput(networks=inventory_def["input"]["networks"])
|
|
if "ranges" in inventory_def["input"]:
|
|
logging.info(
|
|
"Loading %s into AntaInventoryInput ranges section",
|
|
str(inventory_def["input"]["ranges"]),
|
|
)
|
|
AntaInventoryInput(ranges=inventory_def["input"]["ranges"])
|
|
except ValidationError as exc:
|
|
logging.warning("Error: %s", str(exc))
|
|
else:
|
|
raise AssertionError
|
|
|
|
|
|
class TestInventoryDeviceModel:
|
|
"""Unit test of InventoryDevice model."""
|
|
|
|
@pytest.mark.parametrize("test_definition", INVENTORY_DEVICE_MODEL_VALID, ids=generate_test_ids_dict)
|
|
def test_inventory_device_valid(self, test_definition: dict[str, Any]) -> None:
|
|
"""Test loading valid data to InventoryDevice class.
|
|
|
|
Test structure:
|
|
---------------
|
|
|
|
{
|
|
"name": "Valid_Inventory",
|
|
"input": [
|
|
{
|
|
'host': '1.1.1.1',
|
|
'username': 'arista',
|
|
'password': 'arista123!'
|
|
},
|
|
{
|
|
'host': '1.1.1.1',
|
|
'username': 'arista',
|
|
'password': 'arista123!'
|
|
}
|
|
],
|
|
"expected_result": "valid"
|
|
}
|
|
|
|
"""
|
|
if test_definition["expected_result"] == "invalid":
|
|
pytest.skip("Not concerned by the test")
|
|
|
|
try:
|
|
for entity in test_definition["input"]:
|
|
AsyncEOSDevice(**entity)
|
|
except TypeError as exc:
|
|
logging.warning("Error: %s", str(exc))
|
|
raise AssertionError from exc
|
|
|
|
@pytest.mark.parametrize("test_definition", INVENTORY_DEVICE_MODEL_INVALID, ids=generate_test_ids_dict)
|
|
def test_inventory_device_invalid(self, test_definition: dict[str, Any]) -> None:
|
|
"""Test loading invalid data to InventoryDevice class.
|
|
|
|
Test structure:
|
|
---------------
|
|
|
|
{
|
|
"name": "Valid_Inventory",
|
|
"input": [
|
|
{
|
|
'host': '1.1.1.1',
|
|
'username': 'arista',
|
|
'password': 'arista123!'
|
|
},
|
|
{
|
|
'host': '1.1.1.1',
|
|
'username': 'arista',
|
|
'password': 'arista123!'
|
|
}
|
|
],
|
|
"expected_result": "valid"
|
|
}
|
|
|
|
"""
|
|
if test_definition["expected_result"] == "valid":
|
|
pytest.skip("Not concerned by the test")
|
|
|
|
try:
|
|
for entity in test_definition["input"]:
|
|
AsyncEOSDevice(**entity)
|
|
except TypeError as exc:
|
|
logging.info("Error: %s", str(exc))
|
|
else:
|
|
raise AssertionError
|