tests/unit/benchmark/scenarios/networking/test_vsperf_dpdk.py

   1 #!/usr/bin/env python
   2
   3 # Copyright 2017 Nokia
   4 #
   5 # Licensed under the Apache License, Version 2.0 (the "License");
   6 # you may not use this file except in compliance with the License.
   7 # You may obtain a copy of the License at
   8 #
   9 #   http://www.apache.org/licenses/LICENSE-2.0
  10 #
  11 # Unless required by applicable law or agreed to in writing, software
  12 # distributed under the License is distributed on an "AS IS" BASIS,
  13 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14 # See the License for the specific language governing permissions and
  15 # limitations under the License.
  16
  17 # Unittest for yardstick.benchmark.scenarios.networking.vsperf.VsperfDPDK
  18
  19 from __future__ import absolute_import
  20 try:
  21     from unittest import mock
  22 except ImportError:
  23     import mock
  24 import unittest
  25
  26 from yardstick.benchmark.scenarios.networking import vsperf_dpdk
  27
  28
  29 @mock.patch('yardstick.benchmark.scenarios.networking.vsperf_dpdk.subprocess')
  30 @mock.patch('yardstick.benchmark.scenarios.networking.vsperf_dpdk.ssh')
  31 class VsperfDPDKTestCase(unittest.TestCase):
  32
  33     def setUp(self):
  34         self.ctx = {
  35             "host": {
  36                 "ip": "10.229.47.137",
  37                 "user": "ubuntu",
  38                 "password": "ubuntu",
  39             },
  40         }
  41         self.args = {
  42             'task_id': "1234-5678",
  43             'options': {
  44                 'testname': 'pvp_tput',
  45                 'traffic_type': 'rfc2544_throughput',
  46                 'frame_size': '64',
  47                 'test_params': 'TRAFFICGEN_DURATION=30;',
  48                 'trafficgen_port1': 'ens4',
  49                 'trafficgen_port2': 'ens5',
  50                 'conf_file': 'vsperf-yardstick.conf',
  51                 'setup_script': 'setup_yardstick.sh',
  52                 'moongen_helper_file': '~/moongen.py',
  53                 'moongen_host_ip': '10.5.201.151',
  54                 'moongen_port1_mac': '8c:dc:d4:ae:7c:5c',
  55                 'moongen_port2_mac': '8c:dc:d4:ae:7c:5d',
  56                 'trafficgen_port1_nw': 'test2',
  57                 'trafficgen_port2_nw': 'test3',
  58             },
  59             'sla': {
  60                 'metrics': 'throughput_rx_fps',
  61                 'throughput_rx_fps': 500000,
  62                 'action': 'monitor',
  63             }
  64         }
  65
  66     def test_vsperf_dpdk_setup(self, mock_ssh, mock_subprocess):
  67         p = vsperf_dpdk.VsperfDPDK(self.args, self.ctx)
  68
  69         # setup() specific mocks
  70         mock_subprocess.call().execute.return_value = None
  71
  72         p.setup()
  73         self.assertIsNotNone(p.client)
  74         self.assertEqual(p.setup_done, True)
  75
  76     def test_vsperf_dpdk_teardown(self, mock_ssh, mock_subprocess):
  77         p = vsperf_dpdk.VsperfDPDK(self.args, self.ctx)
  78
  79         # setup() specific mocks
  80         mock_subprocess.call().execute.return_value = None
  81
  82         p.setup()
  83         self.assertIsNotNone(p.client)
  84         self.assertEqual(p.setup_done, True)
  85
  86         p.teardown()
  87         self.assertEqual(p.setup_done, False)
  88
  89     def test_vsperf_dpdk_is_dpdk_setup_no(self, mock_ssh, mock_subprocess):
  90         p = vsperf_dpdk.VsperfDPDK(self.args, self.ctx)
  91
  92         # setup() specific mocks
  93         mock_subprocess.call().execute.return_value = None
  94
  95         p.setup()
  96         self.assertIsNotNone(p.client)
  97         self.assertEqual(p.setup_done, True)
  98
  99         # is_dpdk_setup() specific mocks
 100         mock_ssh.SSH.from_node().execute.return_value = (0, 'dummy', '')
 101
 102         result = p._is_dpdk_setup()
 103         self.assertEqual(result, False)
 104
 105     def test_vsperf_dpdk_is_dpdk_setup_yes(self, mock_ssh, mock_subprocess):
 106         p = vsperf_dpdk.VsperfDPDK(self.args, self.ctx)
 107
 108         # setup() specific mocks
 109         mock_subprocess.call().execute.return_value = None
 110
 111         p.setup()
 112         self.assertIsNotNone(p.client)
 113         self.assertEqual(p.setup_done, True)
 114
 115         # is_dpdk_setup() specific mocks
 116         mock_ssh.SSH.from_node().execute.return_value = (0, '', '')
 117
 118         result = p._is_dpdk_setup()
 119         self.assertEqual(result, True)
 120
 121     @mock.patch('yardstick.benchmark.scenarios.networking.vsperf_dpdk.time')
 122     def test_vsperf_dpdk_dpdk_setup_first(self, mock_time, mock_ssh, mock_subprocess):
 123         p = vsperf_dpdk.VsperfDPDK(self.args, self.ctx)
 124
 125         # setup() specific mocks
 126         mock_subprocess.call().execute.return_value = None
 127
 128         p.setup()
 129         self.assertIsNotNone(p.client)
 130         self.assertEqual(p.setup_done, True)
 131
 132         # is_dpdk_setup() specific mocks
 133         mock_ssh.SSH.from_node().execute.return_value = (0, 'dummy', '')
 134
 135         p.dpdk_setup()
 136         self.assertEqual(p._is_dpdk_setup(), False)
 137         self.assertEqual(p.dpdk_setup_done, True)
 138
 139     @mock.patch('yardstick.benchmark.scenarios.networking.vsperf_dpdk.time')
 140     def test_vsperf_dpdk_dpdk_setup_next(self, mock_time, mock_ssh, mock_subprocess):
 141         p = vsperf_dpdk.VsperfDPDK(self.args, self.ctx)
 142
 143         # setup() specific mocks
 144         mock_ssh.SSH.from_node().execute.return_value = (0, '', '')
 145         mock_subprocess.call().execute.return_value = None
 146
 147         p.setup()
 148         self.assertIsNotNone(p.client)
 149         self.assertEqual(p.setup_done, True)
 150
 151         p.dpdk_setup()
 152         self.assertEqual(p._is_dpdk_setup(), True)
 153         self.assertEqual(p.dpdk_setup_done, True)
 154
 155     @mock.patch('yardstick.benchmark.scenarios.networking.vsperf_dpdk.time')
 156     def test_vsperf_dpdk_dpdk_setup_fail(self, mock_time, mock_ssh, mock_subprocess):
 157         p = vsperf_dpdk.VsperfDPDK(self.args, self.ctx)
 158
 159         # setup() specific mocks
 160         mock_ssh.SSH.from_node().execute.return_value = (0, '', '')
 161         mock_subprocess.call().execute.return_value = None
 162
 163         p.setup()
 164         self.assertIsNotNone(p.client)
 165         mock_ssh.SSH.from_node().execute.return_value = (1, '', '')
 166         self.assertEqual(p.setup_done, True)
 167
 168         self.assertRaises(RuntimeError, p.dpdk_setup)
 169
 170     @mock.patch('yardstick.benchmark.scenarios.networking.vsperf_dpdk.time')
 171     def test_vsperf_dpdk_run_ok(self, mock_time, mock_ssh, mock_subprocess):
 172         p = vsperf_dpdk.VsperfDPDK(self.args, self.ctx)
 173
 174         # setup() specific mocks
 175         mock_ssh.SSH.from_node().execute.return_value = (0, '', '')
 176         mock_subprocess.call().execute.return_value = None
 177
 178         p.setup()
 179         self.assertIsNotNone(p.client)
 180         self.assertEqual(p.setup_done, True)
 181
 182         # run() specific mocks
 183         mock_subprocess.call().execute.return_value = None
 184         mock_ssh.SSH.from_node().execute.return_value = (
 185             0, 'throughput_rx_fps\r\n14797660.000\r\n', '')
 186
 187         result = {}
 188         p.run(result)
 189
 190         self.assertEqual(result['throughput_rx_fps'], '14797660.000')
 191
 192     def test_vsperf_dpdk_run_falied_vsperf_execution(self, mock_ssh,
 193                                                 mock_subprocess):
 194         p = vsperf_dpdk.VsperfDPDK(self.args, self.ctx)
 195
 196         # setup() specific mocks
 197         mock_ssh.SSH.from_node().execute.return_value = (0, '', '')
 198         mock_subprocess.call().execute.return_value = None
 199
 200         p.setup()
 201         self.assertIsNotNone(p.client)
 202         self.assertEqual(p.setup_done, True)
 203
 204         # run() specific mocks
 205         mock_subprocess.call().execute.return_value = None
 206         mock_subprocess.call().execute.return_value = None
 207         mock_ssh.SSH.from_node().execute.return_value = (1, '', '')
 208
 209         result = {}
 210         self.assertRaises(RuntimeError, p.run, result)
 211
 212     def test_vsperf_dpdk_run_falied_csv_report(self, mock_ssh, mock_subprocess):
 213         p = vsperf_dpdk.VsperfDPDK(self.args, self.ctx)
 214
 215         # setup() specific mocks
 216         mock_ssh.SSH.from_node().execute.return_value = (0, '', '')
 217         mock_subprocess.call().execute.return_value = None
 218
 219         p.setup()
 220         self.assertIsNotNone(p.client)
 221         self.assertEqual(p.setup_done, True)
 222
 223         # run() specific mocks
 224         mock_subprocess.call().execute.return_value = None
 225         mock_subprocess.call().execute.return_value = None
 226         mock_ssh.SSH.from_node().execute.return_value = (0, '', '')
 227         mock_ssh.SSH.from_node().execute.return_value = (1, '', '')
 228
 229         result = {}
 230         self.assertRaises(RuntimeError, p.run, result)
 231
 232 def main():
 233     unittest.main()
 234
 235
 236 if __name__ == '__main__':
 237     main()