Adding upstream version 0.13.0.
Signed-off-by: Daniel Baumann <daniel@debian.org>
This commit is contained in:
parent
c0ae77e0f6
commit
ecf5ca3300
GPG key ID:
FBB4F0E80A80222F
272 changed files with 33172 additions and 0 deletions
393
tests/units/inventory/test_models.py
Normal file
393
tests/units/inventory/test_models.py
Normal file
|
|
@ -0,0 +1,393 @@
|
|||
# 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 Test_InventoryUnitModels:
|
||||
"""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))
|
||||
assert False
|
||||
else:
|
||||
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))
|
||||
assert False
|
||||
else:
|
||||
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:
|
||||
assert False
|
||||
|
||||
@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))
|
||||
assert False
|
||||
else:
|
||||
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:
|
||||
assert False
|
||||
|
||||
@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 Test_AntaInventoryInputModel:
|
||||
"""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().keys() 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))
|
||||
assert False
|
||||
else:
|
||||
logging.info("Checking if all root keys are correctly lodaded")
|
||||
assert all(elem in inventory.model_dump().keys() for elem in inventory_def["input"].keys())
|
||||
|
||||
@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"].keys():
|
||||
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"].keys():
|
||||
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"].keys():
|
||||
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:
|
||||
assert False
|
||||
|
||||
|
||||
class Test_InventoryDeviceModel:
|
||||
"""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")
|
||||
|
||||
for entity in test_definition["input"]:
|
||||
try:
|
||||
AsyncEOSDevice(**entity)
|
||||
except TypeError as exc:
|
||||
logging.warning("Error: %s", str(exc))
|
||||
assert False
|
||||
|
||||
@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")
|
||||
|
||||
for entity in test_definition["input"]:
|
||||
try:
|
||||
AsyncEOSDevice(**entity)
|
||||
except TypeError as exc:
|
||||
logging.info("Error: %s", str(exc))
|
||||
else:
|
||||
assert False
|
||||
Loading…
Add table
Add a link
Reference in a new issue