Support Network Capacity Testing

two metrics:Number of Connection, Number of Frame

JIRA:YARDSTICK-283

Change-Id: I71f1de25667437bbeac7c06749ff8fb38f41a791
Signed-off-by: kubi <jean.gaoliang@huawei.com>
Signed-off-by: rexlee8776 <limingjiang@huawei.com>

4 files changed, 187 insertions, 0 deletions

diff --git a/samples/networkcapacity.yaml b/samples/networkcapacity.yaml
new file mode 100644
index 000000000..7c62cc2b5
--- /dev/null
+++ b/samples/networkcapacity.yaml
@@ -0,0 +1,21 @@
+---
+# Sample benchmark task config file
+# Measure compute capacity and scale.
+# Including number of cores, number of threads, available memory size and
+# cache size.
+
+schema: "yardstick:task:0.1"
+
+scenarios:
+-
+  type: NetworkCapacity
+  host: node1.LF
+
+  runner:
+    type: Iteration
+    iterations: 1
+
+context:
+  type: Node
+  name: LF
+  file: etc/yardstick/nodes/compass_sclab_virtual/pod.yaml
diff --git a/tests/unit/benchmark/scenarios/networking/test_networkcapacity.py b/tests/unit/benchmark/scenarios/networking/test_networkcapacity.py
new file mode 100644
index 000000000..e3a096446
--- /dev/null
+++ b/tests/unit/benchmark/scenarios/networking/test_networkcapacity.py
@@ -0,0 +1,56 @@
+#!/usr/bin/env python

+

+##############################################################################

+# Copyright (c) 2016 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

+##############################################################################

+

+# Unittest for yardstick.benchmark.scenarios.networking.networkcapacity.NetworkCapacity

+

+import mock

+import unittest

+import os

+import json

+

+from yardstick.benchmark.scenarios.networking import networkcapacity

+

+SAMPLE_OUTPUT = '{"Number of connections":"308","Number of frames received": "166503"}'

+

+@mock.patch('yardstick.benchmark.scenarios.networking.networkcapacity.ssh')

+class NetworkCapacityTestCase(unittest.TestCase):

+

+    def setUp(self):

+        self.ctx = {

+                'host': {

+                    'ip': '172.16.0.137',

+                    'user': 'cirros',

+                    'password': "root"

+                },

+        }

+

+        self.result = {}

+

+    def test_capacity_successful_setup(self, mock_ssh):

+        c = networkcapacity.NetworkCapacity({}, self.ctx)

+        mock_ssh.SSH().execute.return_value = (0, '', '')

+        c.setup()

+        self.assertIsNotNone(c.client)

+        self.assertTrue(c.setup_done)

+

+    def test_capacity_successful(self, mock_ssh):

+        c = networkcapacity.NetworkCapacity({}, self.ctx)

+

+        mock_ssh.SSH().execute.return_value = (0, SAMPLE_OUTPUT, '')

+        c.run(self.result)

+        expected_result = json.loads(SAMPLE_OUTPUT)

+        self.assertEqual(self.result, expected_result)

+

+    def test_capacity_unsuccessful_script_error(self, mock_ssh):

+        c = networkcapacity.NetworkCapacity({}, self.ctx)

+

+        mock_ssh.SSH().execute.return_value = (1, '', 'FOOBAR')

+        self.assertRaises(RuntimeError, c.run, self.result)

diff --git a/yardstick/benchmark/scenarios/networking/networkcapacity.bash b/yardstick/benchmark/scenarios/networking/networkcapacity.bash
new file mode 100644
index 000000000..a18f97e0b
--- /dev/null
+++ b/yardstick/benchmark/scenarios/networking/networkcapacity.bash
@@ -0,0 +1,41 @@
+#!/bin/bash
+
+##############################################################################
+# Copyright (c) 2016 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
+##############################################################################
+
+# Measure compute capacity and scale of a host
+
+set -e
+OUTPUT_FILE=/tmp/netperf-out.log
+
+# run capacity test
+run_capacity()
+{
+    netstat -s > $OUTPUT_FILE
+}
+
+# write the result to stdout in json format
+output_json()
+{
+    CONNECTIONS=$(awk '/active/{print $1}' $OUTPUT_FILE)
+    FRAMES=$(awk '/total\ packets\ received/{print $1}' $OUTPUT_FILE)
+    echo -e "{ \
+        \"Number of connections\":\"$CONNECTIONS\", \
+        \"Number of frames received\": \"$FRAMES\" \
+    }"
+}
+
+main()
+{
+    run_capacity
+
+    output_json
+}
+
+main
diff --git a/yardstick/benchmark/scenarios/networking/networkcapacity.py b/yardstick/benchmark/scenarios/networking/networkcapacity.py
new file mode 100644
index 000000000..57d3b5072
--- /dev/null
+++ b/yardstick/benchmark/scenarios/networking/networkcapacity.py
@@ -0,0 +1,69 @@
+##############################################################################

+# Copyright (c) 2016 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 pkg_resources

+import logging

+import json

+

+import yardstick.ssh as ssh

+from yardstick.benchmark.scenarios import base

+

+LOG = logging.getLogger(__name__)

+

+

+class NetworkCapacity(base.Scenario):

+    """Measure Network capacity and scale.

+

+    This scenario reads network status including number of connections,

+    number of frames sent/received.

+    """

+    __scenario_type__ = "NetworkCapacity"

+    TARGET_SCRIPT = "networkcapacity.bash"

+

+    def __init__(self, scenario_cfg, context_cfg):

+        self.scenario_cfg = scenario_cfg

+        self.context_cfg = context_cfg

+        self.setup_done = False

+

+    def setup(self):

+        """scenario setup"""

+        self.target_script = pkg_resources.resource_filename(

+            "yardstick.benchmark.scenarios.networking",

+            NetworkCapacity.TARGET_SCRIPT)

+

+        host = self.context_cfg['host']

+        if host is None:

+            raise RuntimeError('No right node.please check the configuration')

+        host_user = host.get('user', 'ubuntu')

+        host_ip = host.get('ip', None)

+        host_pwd = host.get('password', None)

+

+        LOG.debug("user:%s, host:%s", host_user, host_ip)

+        self.client = ssh.SSH(host_user, host_ip, password=host_pwd)

+        self.client.wait(timeout=600)

+

+        # copy script to host

+        self.client.run("cat > ~/networkcapacity.sh",

+                        stdin=open(self.target_script, 'rb'))

+

+        self.setup_done = True

+

+    def run(self, result):

+        """execute the benchmark"""

+

+        if not self.setup_done:

+            self.setup()

+

+        cmd = "sudo bash networkcapacity.sh"

+

+        LOG.debug("Executing command: %s", cmd)

+        status, stdout, stderr = self.client.execute(cmd)

+        if status:

+            raise RuntimeError(stderr)

+

+        result.update(json.loads(stdout))