Merge pull request #439 from nornir-automation/netconf_plugins

Add netconf_capabilities plugin

nornir/plugins/tasks/networking/__init__.py

@@ -2,6 +2,7 @@
 from .napalm_configure import napalm_configure
 from .napalm_get import napalm_get
 from .napalm_validate import napalm_validate
+from .netconf_capabilities import netconf_capabilities
 from .netconf_edit_config import netconf_edit_config
 from .netconf_get import netconf_get
 from .netconf_get_config import netconf_get_config
@@ -17,6 +18,7 @@
     "napalm_configure",
     "napalm_get",
     "napalm_validate",
+    "netconf_capabilities",
     "netconf_edit_config",
     "netconf_get",
     "netconf_get_config",

tests/plugins/tasks/networking/test_netconf_connection.py → nornir/plugins/tasks/networking/netconf_capabilities.py

@@ -2,13 +2,19 @@
 
 
 def netconf_capabilities(task: Task) -> Result:
-    manager = task.host.get_connection("netconf", task.nornir.config)
-    capabilities = [capability for capability in manager.server_capabilities]
-    return Result(host=task.host, result=capabilities)
+    """
+    Gather Netconf capabilities from device
+
+    Examples:
 
+        Simple example::
 
-def test_netconf(netconf):
-    result = netconf.run(netconf_capabilities)
+            > nr.run(task=netconf_capabilities)
 
-    for _, v in result.items():
-        assert "urn:ietf:params:netconf:capability:writable-running:1.0" in v.result
+    Returns:
+        Result object with the following attributes set:
+          * result (``list``): list with the capabilities of the host
+    """
+    manager = task.host.get_connection("netconf", task.nornir.config)
+    capabilities = [capability for capability in manager.server_capabilities]
+    return Result(host=task.host, result=capabilities)

tests/plugins/tasks/networking/test_netconf_capabilities.py

@@ -0,0 +1,8 @@
+from nornir.plugins.tasks import networking
+
+
+def test_netconf_capabilities(netconf):
+    result = netconf.run(networking.netconf_capabilities)
+
+    for _, v in result.items():
+        assert "urn:ietf:params:netconf:capability:writable-running:1.0" in v.result