| Index: client/tests/kvm/tests/netperf.py
|
| diff --git a/client/tests/kvm/tests/netperf.py b/client/tests/kvm/tests/netperf.py
|
| index 7c341fad5e145bc248ca1edc75a946783e534b80..dc21e0fbb5ff098f91799581510e5097fef145bf 100644
|
| --- a/client/tests/kvm/tests/netperf.py
|
| +++ b/client/tests/kvm/tests/netperf.py
|
| @@ -1,7 +1,7 @@
|
| import logging, commands, os
|
| from autotest_lib.client.common_lib import error
|
| from autotest_lib.client.bin import utils
|
| -import kvm_test_utils, kvm_subprocess
|
| +import kvm_test_utils
|
|
|
| def run_netperf(test, params, env):
|
| """
|
| @@ -26,19 +26,20 @@ def run_netperf(test, params, env):
|
| result_file = os.path.join(test.resultsdir, "output_%s" % test.iteration)
|
|
|
| firewall_flush = "iptables -F"
|
| - session.cmd_output(firewall_flush)
|
| + session.get_command_output(firewall_flush)
|
|
|
| for i in params.get("netperf_files").split():
|
| if not vm.copy_files_to(os.path.join(netperf_dir, i), "/tmp"):
|
| raise error.TestError("Could not copy file %s to guest" % i)
|
|
|
| - try:
|
| - session.cmd(firewall_flush)
|
| - except kvm_subprocess.ShellError:
|
| + if session.get_command_status(firewall_flush):
|
| logging.warning("Could not flush firewall rules on guest")
|
|
|
| - session.cmd(setup_cmd % "/tmp", timeout=200)
|
| - session.cmd(params.get("netserver_cmd") % "/tmp")
|
| + if session.get_command_status(setup_cmd % "/tmp", timeout=200):
|
| + raise error.TestFail("Fail to setup netperf on guest")
|
| +
|
| + if session.get_command_status(params.get("netserver_cmd") % "/tmp"):
|
| + raise error.TestFail("Fail to start netperf server on guest")
|
|
|
| try:
|
| logging.info("Setup and run netperf client on host")
|
| @@ -48,18 +49,15 @@ def run_netperf(test, params, env):
|
| result.write("Netperf test results\n")
|
|
|
| for i in params.get("protocols").split():
|
| - packet_size = params.get("packet_size", "1500")
|
| - for size in packet_size.split():
|
| - cmd = params.get("netperf_cmd") % (netperf_dir, i,
|
| - guest_ip, size)
|
| - logging.info("Netperf: protocol %s", i)
|
| - try:
|
| - netperf_output = utils.system_output(cmd,
|
| - retain_output=True)
|
| - result.write("%s\n" % netperf_output)
|
| - except:
|
| - logging.error("Test of protocol %s failed", i)
|
| - list_fail.append(i)
|
| + cmd = params.get("netperf_cmd") % (netperf_dir, i, guest_ip)
|
| + logging.info("Netperf: protocol %s", i)
|
| + try:
|
| + netperf_output = utils.system_output(cmd,
|
| + retain_output=True)
|
| + result.write("%s\n" % netperf_output)
|
| + except:
|
| + logging.error("Test of protocol %s failed", i)
|
| + list_fail.append(i)
|
|
|
| result.close()
|
|
|
| @@ -68,5 +66,5 @@ def run_netperf(test, params, env):
|
| ", ".join(list_fail))
|
|
|
| finally:
|
| - session.cmd_output("killall netserver")
|
| + session.get_command_output("killall netserver")
|
| session.close()
|
|
|