From 2659bfa861fd129b22b4769a9ed402517ef3db44 Mon Sep 17 00:00:00 2001
From: "jan.hradil" <jan.hradil@pantheon.tech>
Date: Tue, 11 Apr 2017 09:28:07 +0200
Subject: CSIT-534 HC Test: policer tests

Added empty lines where needed
Repaired Phrasese "Given" --> "And" where necessary in mgmt-cfg-policer-apihc-func.robot
Corrected English in mgmt-cfg-policer-apihc-func.robot - in TC07 Documentation

Change-Id: Id99d7fd04d8f2dcae1820baa208ce338fad3a290
Signed-off-by: jan.hradil <jan.hradil@pantheon.tech>
---
 resources/libraries/python/honeycomb/Routing.py | 57 +++++++++++++++++++++++++
 1 file changed, 57 insertions(+)

(limited to 'resources/libraries/python/honeycomb/Routing.py')

diff --git a/resources/libraries/python/honeycomb/Routing.py b/resources/libraries/python/honeycomb/Routing.py
index c39031d920..a4d606bd5d 100644
--- a/resources/libraries/python/honeycomb/Routing.py
+++ b/resources/libraries/python/honeycomb/Routing.py
@@ -273,3 +273,60 @@ class RoutingKeywords(object):
             return {k: str(v) for k, v in dict_of_str.items()}
         except (KeyError, TypeError):
             return {}
+
+    @staticmethod
+    def configure_policer(node, policy_name, policer_data=None):
+        """Configure Policer on the specified node.
+
+        :param node: Honeycomb node.
+        :param policer_data: Dictionary of configurations to apply. \
+        If it is None then the existing configuration is removed.
+        :type node: dict
+        :type policer_data: dict
+        :returns: Content of response.
+        :rtype: bytearray
+        :raises HoneycombError: If policer could not be configured.
+        """
+
+        path = '/' + policy_name
+
+        if not policer_data:
+            status_code, _ = HcUtil.delete_honeycomb_data(
+                node, 'config_policer', path)
+        else:
+            data = {
+                'policer': policer_data
+            }
+
+            status_code, _ = HcUtil.put_honeycomb_data(
+                node, 'config_policer', data, path)
+
+        if status_code not in (HTTPCodes.OK, HTTPCodes.ACCEPTED):
+            raise HoneycombError(
+                'Configuring policer failed. Status code:{0}'\
+                    .format(status_code))
+
+    @staticmethod
+    def get_policer_oper_data(node, policy_name):
+        """Get operational data about Policer on the node.
+
+        :param node: Honeycomb node.
+        :type node: dict
+        :returns: dict of Policer operational data.
+        :rtype: dict
+        :raises HoneycombError: If status code differs from successful.
+        """
+
+        path = '/' + policy_name
+
+        status_code, resp = HcUtil.\
+            get_honeycomb_data(node, "oper_policer", path)
+
+        if status_code != HTTPCodes.OK:
+            raise HoneycombError(
+                "Not possible to get operational information about Policer. "
+                "Status code: {0}.".format(status_code))
+        try:
+            return resp['policer']
+        except (KeyError, TypeError):
+            return {}
-- 
cgit 1.2.3-korg