#!/usr/bin/env python3
"""
Tests for the inventory manager module using hosts.json.
Provides additional coverage focusing on the real-world hosts.json file.
"""
import os
from pathlib import Path
from src.inventory.manager import InventoryManager
from src.schemas.models import DeviceErrorResult, DeviceListResult, NetworkOS
class TestHostsInventory:
"""Test cases for the inventory manager using the test_devices.json file."""
def test_get_device_from_hosts(self):
"""Test retrieving devices from the test_devices.json file."""
# Reset and initialize
InventoryManager._instance = None
InventoryManager._devices = {}
InventoryManager._initialized = False
hosts_file = os.path.join(Path(__file__).parent, "test_devices.json")
InventoryManager.initialize(hosts_file)
# Test getting a device that should exist in test_devices.json
device = InventoryManager.get_device("test-device-1")
# Assertions
assert not isinstance(
device, DeviceErrorResult
), "Should successfully retrieve test-device-1 device"
assert str(device.ip_address) == "10.0.0.1"
assert device.port == 57777
assert device.nos == NetworkOS.IOSXR
assert device.username == "test_user"
assert device.password == "test_pass"
def test_get_invalid_device_from_hosts(self):
"""Test retrieving non-existent device from test_devices.json."""
# Reset and initialize
InventoryManager._instance = None
InventoryManager._devices = {}
InventoryManager._initialized = False
hosts_file = os.path.join(Path(__file__).parent, "test_devices.json")
InventoryManager.initialize(hosts_file)
result = InventoryManager.get_device("non-existent-router")
# Assertions
assert isinstance(result, DeviceErrorResult)
assert "non-existent-router" in result.msg
def test_list_devices_from_hosts(self):
"""Test listing all devices from test_devices.json."""
# Reset and initialize
InventoryManager._instance = None
InventoryManager._devices = {}
InventoryManager._initialized = False
hosts_file = os.path.join(Path(__file__).parent, "test_devices.json")
InventoryManager.initialize(hosts_file)
# Get list of devices
result = InventoryManager.list_devices()
# Assertions
assert isinstance(result, DeviceListResult)
# test_devices.json should have 2 devices
assert len(result.devices) == 2
# Check that test-device-1 and test-device-2 exist in the device list
device_names = [d.name for d in result.devices]
assert "test-device-1" in device_names
assert "test-device-2" in device_names
# Verify some specific device details
devices = {d.name: d for d in result.devices}
# Check test-device-2
test_device_2 = devices["test-device-2"]
assert str(test_device_2.ip_address) == "10.0.0.2"
assert test_device_2.port == 57777
assert test_device_2.nos == NetworkOS.IOSXR
