sven
b1c9c7a40b
modified: netbox_services/engine/allocators.py modified: netbox_services/engine/context.py new file: netbox_services/engine/mappers.py modified: netbox_services/engine/resolver.py new file: netbox_services/engine/templating.py modified: netbox_services/forms.py modified: netbox_services/models.py modified: netbox_services/tests/test_allocators.py new file: netbox_services/tests/test_mappers.py modified: netbox_services/tests/test_resolver.py new file: netbox_services/tests/test_templating.py Simplify layout to template and service instances
83 lines
3.0 KiB
Python
83 lines
3.0 KiB
Python
"""Tests for the central link mapper."""
|
|
|
|
from netbox_services.engine.mappers import map_link, register_link, _LINK_MAP
|
|
|
|
|
|
class TestMapLink:
|
|
def test_interface_to_device(self):
|
|
from dcim.models import Device, Interface
|
|
assert map_link(Interface, Device) == "device"
|
|
|
|
def test_ipaddress_to_interface(self):
|
|
from dcim.models import Interface
|
|
from ipam.models import IPAddress
|
|
assert map_link(IPAddress, Interface) == "assigned_object"
|
|
|
|
def test_ipaddress_to_vminterface(self):
|
|
from ipam.models import IPAddress
|
|
from virtualization.models import VMInterface
|
|
assert map_link(IPAddress, VMInterface) == "assigned_object"
|
|
|
|
def test_interface_to_module(self):
|
|
from dcim.models import Interface, Module
|
|
assert map_link(Interface, Module) == "module"
|
|
|
|
def test_device_to_site(self):
|
|
from dcim.models import Device, Site
|
|
assert map_link(Device, Site) == "site"
|
|
|
|
def test_device_to_rack(self):
|
|
from dcim.models import Device, Rack
|
|
assert map_link(Device, Rack) == "rack"
|
|
|
|
def test_prefix_to_site_scope(self):
|
|
from dcim.models import Site
|
|
from ipam.models import Prefix
|
|
assert map_link(Prefix, Site) == "scope"
|
|
|
|
def test_vlan_to_vlangroup(self):
|
|
from ipam.models import VLAN, VLANGroup
|
|
assert map_link(VLAN, VLANGroup) == "group"
|
|
|
|
def test_interface_self_referential(self):
|
|
from dcim.models import Interface
|
|
assert map_link(Interface, Interface) == "parent"
|
|
|
|
def test_vminterface_to_virtualmachine(self):
|
|
from virtualization.models import VMInterface, VirtualMachine
|
|
assert map_link(VMInterface, VirtualMachine) == "virtual_machine"
|
|
|
|
def test_poweroutlet_to_powerport(self):
|
|
from dcim.models import PowerOutlet, PowerPort
|
|
assert map_link(PowerOutlet, PowerPort) == "power_port"
|
|
|
|
def test_frontport_to_rearport(self):
|
|
from dcim.models import FrontPort, RearPort
|
|
assert map_link(FrontPort, RearPort) == "rear_port"
|
|
|
|
def test_unmapped_returns_none(self):
|
|
from dcim.models import Device, Rack
|
|
from ipam.models import IPAddress
|
|
assert map_link(IPAddress, Device) is None
|
|
assert map_link(Device, Device) is None
|
|
assert map_link(Rack, Device) is None
|
|
|
|
|
|
class TestRegisterLink:
|
|
def test_register_new_link(self):
|
|
from dcim.models import Device, Rack
|
|
assert map_link(Rack, Device) is None
|
|
register_link("dcim.rack", "dcim.device", "device")
|
|
assert map_link(Rack, Device) == "device"
|
|
# cleanup
|
|
del _LINK_MAP[("dcim.rack", "dcim.device")]
|
|
assert map_link(Rack, Device) is None
|
|
|
|
def test_override_existing_link(self):
|
|
from dcim.models import Interface
|
|
original = map_link(Interface, Interface)
|
|
register_link("dcim.interface", "dcim.interface", "lag")
|
|
assert map_link(Interface, Interface) == "lag"
|
|
# restore
|
|
register_link("dcim.interface", "dcim.interface", original)
|