1 files changed, 88 insertions, 0 deletions
diff --git a/yardstick/network_services/vnf_generic/vnf/tg_pktgen.py b/yardstick/network_services/vnf_generic/vnf/tg_pktgen.py
new file mode 100644
index 000000000..5da2178af
--- /dev/null
+++ b/yardstick/network_services/vnf_generic/vnf/tg_pktgen.py
@@ -0,0 +1,88 @@
+# Copyright (c) 2018-2019 Intel Corporation
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import logging
+import time
+
+from yardstick.common import constants
+from yardstick.common import exceptions
+from yardstick.common import utils
+from yardstick.network_services.vnf_generic.vnf import base as vnf_base
+
+
+LOG = logging.getLogger(__name__)
+
+
+class PktgenTrafficGen(vnf_base.GenericTrafficGen):
+    """DPDK Pktgen traffic generator
+
+    Website: http://pktgen-dpdk.readthedocs.io/en/latest/index.html
+    """
+
+    TIMEOUT = 30
+
+    def __init__(self, name, vnfd):
+        vnf_base.GenericTrafficGen.__init__(self, name, vnfd)
+        self._traffic_profile = None
+        self._node_ip = vnfd['mgmt-interface'].get('ip')
+        self._lua_node_port = self._get_lua_node_port(
+            vnfd['mgmt-interface'].get('service_ports', []))
+        self._rate = 1
+
+    def instantiate(self, scenario_cfg, context_cfg):  # pragma: no cover
+        pass
+
+    def run_traffic(self, traffic_profile):
+        self._traffic_profile = traffic_profile
+        self._traffic_profile.init(self._node_ip, self._lua_node_port)
+        utils.wait_until_true(self._is_running, timeout=self.TIMEOUT,
+                              sleep=2)
+
+    def terminate(self):  # pragma: no cover
+        pass
+
+    def collect_kpi(self):  # pragma: no cover
+        pass
+
+    def scale(self, flavor=''):  # pragma: no cover
+        pass
+
+    def wait_for_instantiate(self):  # pragma: no cover
+        pass
+
+    def runner_method_start_iteration(self):
+        # pragma: no cover
+        LOG.debug('Start method')
+        # NOTE(ralonsoh): 'rate' should be modified between iterations. The
+        # current implementation is just for testing.
+        self._rate += 1
+        self._traffic_profile.start()
+        self._traffic_profile.rate(self._rate)
+        time.sleep(4)
+        self._traffic_profile.stop()
+
+    @staticmethod
+    def _get_lua_node_port(service_ports):
+        for port in (port for port in service_ports if
+                     int(port['port']) == constants.LUA_PORT):
+            return int(port['node_port'])
+        # NOTE(ralonsoh): in case LUA port is not present, an exception should
+        # be raised.
+
+    def _is_running(self):
+        try:
+            self._traffic_profile.help()
+            return True
+        except exceptions.PktgenActionError:
+            return False