diff --git a/napalm_linux/linux.py b/napalm_linux/linux.py index b4d78cb..d750e9b 100644 --- a/napalm_linux/linux.py +++ b/napalm_linux/linux.py @@ -93,6 +93,7 @@ class LinuxDriver(OSDriver): package manager unless overridden by ``optional_args["pkg_manager"]``. """ + TYPE_LABEL = "Linux" VENDOR = "Linux" NETMIKO_DEVICE_TYPE = "linux" diff --git a/tests/test_linux.py b/tests/test_linux.py index cdb6049..e465458 100644 --- a/tests/test_linux.py +++ b/tests/test_linux.py @@ -341,6 +341,21 @@ def test_apply_updates_unsupported_pm_raises(driver): driver.apply_updates(["curl"]) +# --------------------------------------------------------------------------- +# TYPE_LABEL +# --------------------------------------------------------------------------- + + +def test_type_label_is_linux(): + assert LinuxDriver.TYPE_LABEL == "Linux" + + +def test_type_label_overrides_os_driver(): + from napalm_device_types import OSDriver + assert OSDriver.TYPE_LABEL == "OS" + assert LinuxDriver.TYPE_LABEL != OSDriver.TYPE_LABEL + + # --------------------------------------------------------------------------- # _collect_platform_info # ---------------------------------------------------------------------------