From 8f1101df131a4d3e03b377738507d88b745831c0 Mon Sep 17 00:00:00 2001
From: "Yiting.Li" <liyiting@huawei.com>
Date: Tue, 22 Dec 2015 17:11:12 -0800
Subject:     Upload the contribution of vstf as bottleneck network framework.

    End to End Performance test

    JIRA:BOTTLENECK-29

Change-Id: Ib2c553c8b60d6cda9e7a7b52b737c9139f706ebd
Signed-off-by: Yiting.Li <liyiting@huawei.com>
---
 vstf/vstf/common/decorator.py | 88 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 88 insertions(+)
 create mode 100755 vstf/vstf/common/decorator.py

(limited to 'vstf/vstf/common/decorator.py')

diff --git a/vstf/vstf/common/decorator.py b/vstf/vstf/common/decorator.py
new file mode 100755
index 00000000..cafbb3ff
--- /dev/null
+++ b/vstf/vstf/common/decorator.py
@@ -0,0 +1,88 @@
+#!/usr/bin/python
+# -*- coding: utf8 -*-
+# author: wly
+# date: 2015-09-09
+# see license for license details
+_DEFAULTS = "vstf check key defaults".encode()
+
+
+def check(key, choices=[], defaults=_DEFAULTS):
+    def _deco(func):
+        def __deco(*args, **kwargs):
+            if key not in kwargs:
+                if defaults != _DEFAULTS:
+                    kwargs[key] = defaults
+                else:
+                    raise Exception("Error: '%s' is needed in %s" % (key, func))
+
+            if choices and kwargs[key] not in choices:
+                raise Exception("Error: %s :%s" % (key, kwargs[key]))
+            ret = func(*args, **kwargs)
+            return ret
+
+        return __deco
+
+    return _deco
+
+
+def dcheck(key, choices=[]):
+    def _deco(func):
+        def __deco(*args):
+            if len(args) == 2:
+                values = args[1]
+            elif len(args) == 1:
+                values = args[0]
+            else:
+                values = None
+            if isinstance(values, dict):
+                if key not in values:
+                    raise Exception("Error: '%s' is needed in %s" % (key, func))
+                if choices and values[key] not in choices:
+                    raise Exception("Error: %s :%s" % (key, values[key]))
+            ret = func(*args)
+            return ret
+
+        return __deco
+
+    return _deco
+
+
+def vstf_input(key, types=str, choices=[], default=None):
+    def _deco(func):
+        def __deco(*args):
+            ret = func(*args)
+            if not isinstance(ret, dict):
+                ret = {}
+            in_str = "----> %s : " % key
+            if choices:
+                in_str = "---- %s\n" % (str(choices)) + in_str
+            while True:
+                if types == list or types == dict:
+                    value = input(in_str)
+                else:
+                    value = raw_input(in_str)
+                    value = types(value)
+                if not value:
+                    value = default
+                if not choices or value in choices:
+                    break
+            ret.update({key: value})
+            return ret
+
+        return __deco
+
+    return _deco
+
+
+def namespace():
+    def _deco(func):
+        def __deco(*args, **kwargs):
+            ret = func(*args, **kwargs)
+            nspace = kwargs.get("namespace", None)
+            if nspace:
+                ret = "ip netns exec %(namespace)s " % {"namespace": nspace} + ret
+            return ret
+
+        return __deco
+
+    return _deco
-- 
cgit 1.2.3-korg