Files
infix/test/infamy/transport.py
T
Mattias Walström 82ff83c50d infamy: Set hostname from logical topology
If no hostname is specified, set it to the dut-name in
logical topology
2024-11-07 09:47:36 +01:00

80 lines
2.1 KiB
Python

from abc import ABC, abstractmethod
from infamy.neigh import ll6ping
def infer_put_dict(name, models):
if not models.get("ietf-system"):
models["ietf-system"] = { "system": { "hostname": name} }
else:
if not models["ietf-system"].get("system"):
models["ieft-system"]["system"] = {}
if not models["ietf-system"]["system"].get("hostname"):
models["ietf-system"]["system"]["hostname"] = name
class Transport(ABC):
"""Common functions for NETCONF/RESTCONF"""
@abstractmethod
def get_data(self, xpath=None, parse=True):
pass
@abstractmethod
def get_config_dict(self, modname):
pass
@abstractmethod
def put_config_dict(self, modname, edit):
pass
@abstractmethod
def get_dict(self, xpath=None):
pass
@abstractmethod
def delete_xpath(self, xpath):
pass
@abstractmethod
def copy(self, source, target):
pass
@abstractmethod
def reboot(self):
pass
@abstractmethod
def get_current_time_with_offset(self):
"""Needed since libyang is too nice and removes the original offset"""
pass
@abstractmethod
def call_action(self, xpath):
pass
@abstractmethod
def get_iface(self, iface):
"""Should be common, but is not due to bug in rousette"""
pass
def get_mgmt_ip(self):
"""Return managment IP address used for RESTCONF/NETCONF"""
return self.location.host
def get_mgmt_iface(self):
"""Return managment interface used for RESTCONF/NETCONF"""
return self.location.interface
def has_model(self, model_name):
"""Check if the device has the given YANG model loaded."""
return model_name in self.modules
def reachable(self):
"""Check if the device reachable on ll6"""
neigh = ll6ping(self.location.interface, flags=["-w1", "-c1", "-L", "-n"])
return bool(neigh)
def test_reset(self):
self.call_action("/infix-test:test/reset")
def startup_override(self):
self.call_action("/infix-test:test/override-startup")