aboutsummaryrefslogtreecommitdiffstats
path: root/yardstick/network_services/traffic_profile/prox_profile.py
blob: de4b3f9a0d729245366fbff8d09883081679b162 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
# Copyright (c) 2016-2017 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.
""" Fixed traffic profile definitions """

from __future__ import absolute_import

import logging
import multiprocessing

from yardstick.network_services.traffic_profile.base import TrafficProfile
from yardstick.network_services.vnf_generic.vnf.prox_helpers import ProxProfileHelper

LOG = logging.getLogger(__name__)


class ProxProfile(TrafficProfile):
    """
    This profile adds a single stream at the beginning of the traffic session
    """

    @staticmethod
    def sort_vpci(traffic_gen):
        """Return the list of external interfaces ordered by vpci and name

        :param traffic_gen: (ProxTrafficGen) traffic generator
        :return: list of ordered interfaces
        """
        def key_func(interface):
            return interface['virtual-interface']['vpci'], interface['name']

        return sorted(traffic_gen.vnfd_helper['vdu'][0]['external-interface'],
                      key=key_func)

    @staticmethod
    def fill_samples(samples, traffic_gen):
        vpci_if_name_ascending = ProxProfile.sort_vpci(traffic_gen)
        for vpci_idx, intf in enumerate(vpci_if_name_ascending):
            name = intf[1]
            # TODO: VNFDs KPIs values needs to be mapped to TRex structure
            xe_port = traffic_gen.resource_helper.sut.port_stats([vpci_idx])
            samples[name] = {
                "in_packets": xe_port[6],
                "out_packets": xe_port[7],
            }

    def __init__(self, tp_config):
        super(ProxProfile, self).__init__(tp_config)
        self.queue = None
        self.done = multiprocessing.Event()
        self.results = []

        # TODO: get init values from tp_config
        self.prox_config = tp_config["traffic_profile"]
        self.pkt_sizes = [int(x) for x in self.prox_config.get("packet_sizes", [])]
        self.pkt_size_iterator = iter(self.pkt_sizes)
        self.duration = int(self.prox_config.get("duration", 5))
        self.precision = float(self.prox_config.get('test_precision', 1.0))
        self.tolerated_loss = float(self.prox_config.get('tolerated_loss', 0.0))

        # TODO: is this ever a function of packet size?
        self.lower_bound = float(self.prox_config.get('lower_bound', 10.0))
        self.upper_bound = float(self.prox_config.get('upper_bound', 100.0))
        self.step_value = float(self.prox_config.get('step_value', 10.0))
        self._profile_helper = None

    def make_profile_helper(self, traffic_gen):
        if self._profile_helper is None:
            self._profile_helper = ProxProfileHelper.make_profile_helper(traffic_gen)
        return self._profile_helper

    def init(self, queue):
        self.pkt_size_iterator = iter(self.pkt_sizes)
        self.queue = queue
        self.queue.cancel_join_thread()

    def bounds_iterator(self, logger=None):
        if logger:
            logger.debug("Interval [%s, %s), step: %d", self.lower_bound,
                         self.upper_bound, self.step_value)

        test_value = self.lower_bound
        while test_value <= self.upper_bound:
            if logger:
                logger.info("Testing with value %s", test_value)
            yield test_value
            test_value += self.step_value

    @property
    def min_pkt_size(self):
        """Return the minimum required packet size for the test.

        Defaults to 64. Individual test must override this method if they have
        other requirements.

        Returns:
            int. The minimum required packet size for the test.
        """
        return 64

    def run_test_with_pkt_size(self, traffic_generator, pkt_size, duration):
        raise NotImplementedError

    def execute_traffic(self, traffic_generator):
        self.make_profile_helper(traffic_generator)

        try:
            pkt_size = next(self.pkt_size_iterator)
        except StopIteration:
            self.done.set()
            return

        # Adjust packet size upwards if it's less than the minimum
        # required packet size for the test.
        if pkt_size < self.min_pkt_size:
            pkt_size += self.min_pkt_size - 64

        duration = self.duration
        self.run_test_with_pkt_size(traffic_generator, pkt_size, duration)