OpenContrail test suitestable/colorado

Change-Id: I61168093a2a05d47377ef47c8638ae1554b1a999
Signed-off-by: Stuart Mackie <wsmackie@juniper.net>

1 files changed, 110 insertions, 0 deletions

diff --git a/Testcases/cfgm_common/utils.py b/Testcases/cfgm_common/utils.py
new file mode 100644
index 0000000..a023912
--- /dev/null
+++ b/Testcases/cfgm_common/utils.py
@@ -0,0 +1,110 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+
+# vim: tabstop=4 shiftwidth=4 softtabstop=4
+
+# Copyright (c) 2015 Juniper Networks
+# All Rights Reserved.
+#
+#    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.
+#
+# @author: Numan Siddique, eNovance.
+
+
+import urllib
+from collections import OrderedDict
+import sys
+import cgitb
+import cStringIO
+
+def detailed_traceback():
+    buf = cStringIO.StringIO()
+    cgitb.Hook(format="text", file=buf).handle(sys.exc_info())
+    tb_txt = buf.getvalue()
+    buf.close()
+    return tb_txt
+# end detailed_traceback
+
+def encode_string(enc_str, encoding='utf-8'):
+    """Encode the string using urllib.quote_plus
+
+    Eg. @input:
+            enc_str = 'neté
+            type - 'unicode' or 'str'
+        @retval
+            enc_str = 'net%C3%A9%C3%B9'
+            type - str
+    """
+    try:
+        enc_str.encode()
+    except (UnicodeDecodeError, UnicodeEncodeError):
+        if type(enc_str) is unicode:
+            enc_str = enc_str.encode(encoding)
+        enc_str = urllib.quote_plus(enc_str)
+    except Exception:
+        pass
+    return enc_str
+
+
+def decode_string(dec_str, encoding='utf-8'):
+    """Decode the string previously encoded using urllib.quote_plus.
+
+    Eg. If dec_str = 'net%C3%A9%C3%B9'
+           type - 'unicode' or 'str'
+        @retval
+            ret_dec_str = 'neté
+            type - unicode
+    """
+    ret_dec_str = dec_str
+    try:
+        if type(ret_dec_str) is unicode:
+            ret_dec_str = str(ret_dec_str)
+        ret_dec_str = urllib.unquote_plus(ret_dec_str)
+        return ret_dec_str.decode(encoding)
+    except Exception:
+        return dec_str
+
+
+class CacheContainer(object):
+    def __init__(self, size):
+        self.container_size = size
+        self.dictionary = OrderedDict()
+
+    def __getitem__(self, key, default=None):
+        value = self.dictionary[key]
+        # item accessed - put it in the front
+        del self.dictionary[key]
+        self.dictionary[key] = value
+
+        return value
+
+    def __setitem__(self, key, value):
+        self.dictionary[key] = value
+        if len(self.dictionary.keys()) > self.container_size:
+            # container is full, loose the least used item
+            self.dictionary.popitem(last=False)
+
+    def __contains__(self, key):
+        return key in self.dictionary
+
+    def __repr__(self):
+        return str(self.dictionary)
+
+
+def CamelCase(input):
+    words = input.replace('_', '-').split('-')
+    name = ''
+    for w in words:
+        name += w.capitalize()
+    return name
+#end CamelCase