diff options
Diffstat (limited to 'resources/libraries/python')
| -rw-r--r-- | resources/libraries/python/honeycomb/Netconf.py | 170 | ||||
| -rw-r--r-- | resources/libraries/python/honeycomb/Notifications.py | 119 |
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." ) |