X-Git-Url: https://gerrit.opnfv.org/gerrit/gitweb?a=blobdiff_plain;f=yardstick%2Fnetwork_services%2Ftraffic_profile%2Ftraffic_profile.py;h=3e1f8d89f16d6f320d47991aec2b94efebb4fa00;hb=46383867760fa485fa20dbee69ca5ba72f50f467;hp=156cc664419ea79b95a8ca9f95493ad09e25d9f6;hpb=5c8396ba5bbeb12c70dae6ad3e777a28fefbcb4a;p=yardstick.git

diff --git a/yardstick/network_services/traffic_profile/traffic_profile.py b/yardstick/network_services/traffic_profile/traffic_profile.py
index 156cc6644..3e1f8d89f 100644
--- a/yardstick/network_services/traffic_profile/traffic_profile.py
+++ b/yardstick/network_services/traffic_profile/traffic_profile.py
@@ -399,16 +399,19 @@ class TrexProfile(TrafficProfile):
             logging.debug("Imax: %s rate: %s", imix_count, self.rate)
         return imix_count
 
-    def get_streams(self):
-        """ generate trex stream """
+    def get_streams(self, profile_data):
+        """ generate trex stream
+        :param profile_data:
+        :type profile_data:
+        """
         self.streams = []
         self.pps = self.params['traffic_profile'].get('frame_rate', 100)
-        for packet_name in self.profile_data:
-            outer_l2 = self.profile_data[packet_name].get('outer_l2')
+        for packet_name in profile_data:
+            outer_l2 = profile_data[packet_name].get('outer_l2')
             imix_data = self.generate_imix_data(outer_l2)
             if not imix_data:
                 imix_data = {64: self.pps}
-            self.generate_vm(self.profile_data[packet_name])
+            self.generate_vm(profile_data[packet_name])
             for size in imix_data:
                 self._generate_streams(size, imix_data[size])
         self._generate_profile()