1 files changed, 59 insertions, 0 deletions
diff --git a/testsuites/vstf/vstf_scripts/vstf/agent/perf/ethtool.py b/testsuites/vstf/vstf_scripts/vstf/agent/perf/ethtool.py
new file mode 100644
index 00000000..3f4a3728
--- /dev/null
+++ b/testsuites/vstf/vstf_scripts/vstf/agent/perf/ethtool.py
@@ -0,0 +1,59 @@
+##############################################################################
+# Copyright (c) 2015 Huawei Technologies Co.,Ltd and others.
+#
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Apache License, Version 2.0
+# which accompanies this distribution, and is available at
+# http://www.apache.org/licenses/LICENSE-2.0
+##############################################################################
+
+import vstf.common.utils as utils
+
+__all__ = ["autoneg_on", "autoneg_off", "autoneg_query"]
+
+_para_map = {
+    "Autonegotiate": ("-A", "-a", "autoneg"),
+    "RX": ("-A", "-a", "rx"),
+    "TX": ("-A", "-a", "tx"),
+}
+
+
+def autoneg_on(iface, nspace=None):
+    return _set(nspace, iface, Autonegotiate="on", RX="on", TX="on")
+
+
+def autoneg_off(iface, nspace=None):
+    return _set(nspace, iface, Autonegotiate="off", RX="off", TX="off")
+
+
+def autoneg_query(iface, nspace=None):
+    return _query(nspace, iface, "-a")
+
+
+def _set(nspace, iface, **kwargs):
+    cmds = {}
+    for item, value in kwargs.items():
+        opt, _, key = _para_map[item]
+        cmds.setdefault(opt, [])
+        cmds[opt].append(key)
+        cmds[opt].append(value)
+
+    for key, value in cmds.items():
+        cmd = _namespace(nspace)
+        cmd += ["ethtool", key, iface] + value
+        utils.call(cmd)
+
+    return True
+
+
+def _query(nspace, iface, item):
+    cmd = _namespace(nspace)
+    cmd += ["ethtool", item, iface]
+    return utils.check_output(cmd)
+
+
+def _namespace(nspace):
+    result = ""
+    if nspace:
+        result = "ip netns exec %(namespace)s " % {"namespace": nspace}
+    return result.split()