aboutsummaryrefslogtreecommitdiffstats
path: root/resources/libraries/python/honeycomb
diff options
context:
space:
mode:
Diffstat (limited to 'resources/libraries/python/honeycomb')
-rw-r--r--resources/libraries/python/honeycomb/Netconf.py170
-rw-r--r--resources/libraries/python/honeycomb/Notifications.py119
2 files changed, 182 insertions, 107 deletions
diff --git a/resources/libraries/python/honeycomb/Netconf.py b/resources/libraries/python/honeycomb/Netconf.py
new file mode 100644
index 0000000000..3c98387334
--- /dev/null
+++ b/resources/libraries/python/honeycomb/Netconf.py
@@ -0,0 +1,170 @@
+# Copyright (c) 2016 Cisco and/or its affiliates.
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at:
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Keywords used to connect to Honeycomb through Netconf, send messages
+ and receive replies."""
+
+from time import time
+
+import paramiko
+import socket
+from robot.api import logger
+from interruptingcow import timeout
+
+from resources.libraries.python.honeycomb.HoneycombUtil import HoneycombError
+
+
+class Netconf(object):
+ """Implements methods for creating and managing Netconf sessions."""
+
+ def __init__(self, delimiter=']]>]]>'):
+ """Initializer.
+
+ Note: Passing the channel object as a robotframework argument closes
+ the channel. Class variables are used instead,
+ to persist the connection channel throughout test cases.
+ """
+
+ self.client = None
+ self.channel = None
+ self.delimiter = delimiter
+
+ def create_session(self, node, hello, time_out=10):
+ """Create an SSH session, connect to Honeycomb on the specified node,
+ open a communication channel to the Netconf subsystem and exchange hello
+ messages.
+
+ :param node: Honeycomb node.
+ :param hello: Hello message and capability list to be sent to Honeycomb.
+ :param time_out: Timeout value for the connection in seconds.
+ :type node: dict
+ :type hello: str
+ :type time_out: int
+ """
+
+ start = time()
+ client = paramiko.SSHClient()
+ client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+
+ client.connect(node['host'],
+ username=node['honeycomb']['user'],
+ password=node['honeycomb']['passwd'],
+ pkey=None,
+ port=node['honeycomb']['netconf_port'],
+ timeout=time_out,
+ )
+
+ logger.trace('Connect took {0} seconds'.format(time() - start))
+ logger.debug('New ssh: {0}'.format(client))
+ logger.debug('Connect peer: {0}'.
+ format(client.get_transport().getpeername()))
+ logger.debug(client)
+
+ channel = client.get_transport().open_session()
+ channel.settimeout(time_out)
+ channel.set_combine_stderr(True)
+ channel.get_pty()
+ channel.invoke_subsystem("netconf")
+ logger.debug(channel)
+
+ self.client = client
+ self.channel = channel
+
+ # read OpenDaylight's hello message and capability list
+ self.get_response(
+ size=131072,
+ time_out=time_out,
+ err="Timeout on getting hello message."
+ )
+
+ self.channel.send(hello)
+ if not self.channel.active:
+ raise HoneycombError("Channel closed on capabilities exchange.")
+
+ def get_response(self, size=4096, time_out=10, err="Unspecified Error."):
+ """Iteratively read data from the receive buffer and catenate together
+ until message ends with the message delimiter, or
+ until timeout is reached.
+
+ :param size: Maximum number of bytes to read in one iteration.
+ :param time_out: Timeout value for getting the complete response.
+ :param err: Error message to provide when timeout is reached.
+ :type size:int
+ :type time_out:int
+ :type err:str
+ :return: Content of response.
+ :rtype: str
+ :raises HoneycombError: If the read process times out.
+ """
+
+ reply = ''
+
+ try:
+ with timeout(time_out, exception=RuntimeError):
+ while not reply.endswith(self.delimiter) or \
+ self.channel.recv_ready():
+ try:
+ chunk = self.channel.recv(size)
+ if not chunk:
+ break
+ reply += chunk
+ if self.channel.exit_status_ready():
+ logger.debug('Channel exit status ready.')
+ break
+ except socket.timeout:
+ raise HoneycombError("Socket timeout.",
+ enable_logging=False
+ )
+
+ except RuntimeError:
+ raise HoneycombError(err + " Content of buffer: {0}".format(reply),
+ enable_logging=False
+ )
+
+ logger.trace(reply)
+ return reply.replace(self.delimiter, "")
+
+ def get_all_responses(self, size=4096, time_out=3):
+ """Read responses from the receive buffer and catenate together
+ until a read operation times out.
+
+ :param size: Maximum number of bytes to read in one iteration.
+ :param time_out: Timeout value for getting the complete response.
+ :type size:int
+ :type time_out:int
+ :return: Content of response.
+ :rtype: str
+ """
+
+ response = ""
+ err = "Expected timeout occurred."
+
+ while True:
+ try:
+ response += self.get_response(size, time_out, err)
+ except HoneycombError:
+ break
+
+ return response
+
+ def send(self, message):
+ """Sends provided message through the channel.
+
+ :param message: Message to be sent to Honeycomb.
+ :type message: str
+ """
+
+ if not message.endswith(self.delimiter):
+ message += self.delimiter
+
+ self.channel.send(message)
diff --git a/resources/libraries/python/honeycomb/Notifications.py b/resources/libraries/python/honeycomb/Notifications.py
index 62cda16cf2..512a6cbb08 100644
--- a/resources/libraries/python/honeycomb/Notifications.py
+++ b/resources/libraries/python/honeycomb/Notifications.py
@@ -13,131 +13,35 @@
"""Implementation of keywords for managing Honeycomb notifications."""
-from time import time
-
-import paramiko
from robot.api import logger
-from interruptingcow import timeout
from resources.libraries.python.honeycomb.HoneycombUtil import HoneycombError
+from resources.libraries.python.honeycomb.Netconf import Netconf
-class Notifications(object):
- """Implements keywords for managing Honeycomb notifications.
+class Notifications(Netconf):
+ """Implements keywords for receiving Honeycomb notifications.
The keywords implemented in this class make it possible to:
- - establish SSH session to Honeycomb host
- receive notifications from Honeycomb
- read received notifications
"""
- def __init__(self, hello, subscription):
- """Initializer.
- :param hello: Hello message to be sent to Honeycomb.
- :param subscription: rpc command to subscribe to Honeycomb notifications
- over Netconf.
- :type hello: str
- :type subscription: str
-
- Note: Passing the channel object as a robotframework argument closes
- the channel. Class variables are used instead,
- to persist the connection channel throughout the test case.
- """
-
- self.client = None
- self.channel = None
- self.hello = hello
- self.subscription = subscription
-
- def create_session(self, node, time_out=10):
- """Create an SSH session and connect to Honeycomb on the specified node.
-
- :param node: Honeycomb node.
- :param time_out: Timeout value for the connection in seconds.
- :type node: dict
- :type time_out: int
- """
-
- start = time()
- client = paramiko.SSHClient()
- client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
-
- client.connect(node['host'],
- username=node['honeycomb']['user'],
- password=node['honeycomb']['passwd'],
- pkey=None,
- port=node['honeycomb']['netconf_port'],
- timeout=time_out,
- )
-
- logger.trace('Connect took {0} seconds'.format(time() - start))
- logger.debug('New ssh: {0}'.format(client))
- logger.debug('Connect peer: {0}'.
- format(client.get_transport().getpeername()))
- logger.debug(client)
-
- channel = client.get_transport().open_session()
- channel.settimeout(time_out)
- channel.get_pty()
- channel.invoke_subsystem("netconf")
- logger.debug(channel)
-
- self.client = client
- self.channel = channel
-
- # read OpenDaylight's hello message and capability list
- self._get_response(
- size=131072,
- time_out=time_out,
- err="Timeout on getting hello message."
- )
-
- self.channel.send(self.hello)
- if not self.channel.active:
- raise HoneycombError("Channel closed on capabilities exchange.")
-
- def _get_response(self, size=4096, time_out=10, err="Unspecified Error."):
- """Iteratively read data from the receive buffer and catenate together
- until message ends with the message delimiter, or
- until timeout is reached.
-
- :param size: Maximum number of bytes to read in one iteration.
- :param time_out: Timeout value for getting the complete response.
- :param err: Error message to provide when timeout is reached.
- :type size:int
- :type time_out:int
- :type err:str
- :return: Content of response.
- :rtype: str
- :raises HoneycombError: If the read process times out.
- """
-
- reply = ''
-
- try:
- with timeout(time_out, exception=RuntimeError):
- while not reply.endswith(']]>]]>'):
- if self.channel.recv_ready():
- reply += self.channel.recv(size)
-
- except RuntimeError:
- raise HoneycombError(err+" Content of buffer: {0}".format(reply))
-
- logger.trace(reply)
- return reply
-
- def add_notification_listener(self, time_out=10):
+ def add_notification_listener(self, subscription, time_out=10):
"""Open a new channel on the SSH session, connect to Netconf subsystem
and subscribe to receive Honeycomb notifications.
+ :param subscription: RPC for subscription to notifications.
:param time_out: Timeout value for each read operation in seconds.
+ :type subscription: str
:type time_out: int
:raises HoneycombError: If subscription to notifications fails.
"""
- self.channel.send(self.subscription)
+ logger.debug(subscription)
+ self.send(subscription)
- reply = self._get_response(
+ reply = self.get_response(
time_out=time_out,
err="Timeout on notifications subscription."
)
@@ -157,9 +61,10 @@ class Notifications(object):
:rtype: str
"""
- logger.debug("Getting notification.")
+ logger.debug("Getting notification. Timeout set to {0} seconds."
+ .format(time_out))
- reply = self._get_response(
+ reply = self.get_response(
time_out=time_out,
err="Timeout on getting notification."
)