X-Git-Url: https://gerrit.opnfv.org/gerrit/gitweb?a=blobdiff_plain;f=tests%2Funit%2Fnetwork_services%2Ftraffic_profile%2Ftest_prox_mpls.py;h=00a690d2aaed48aa065c04ba9d708f74ae558f09;hb=0db868278e13d21998bd677bfc782557d44ab381;hp=642fecc35896b1851940226a1191a2167adbc4ec;hpb=b2552579f300ec9f4c2b516eb44cac0efb3d4dd6;p=yardstick.git

diff --git a/tests/unit/network_services/traffic_profile/test_prox_mpls.py b/tests/unit/network_services/traffic_profile/test_prox_mpls.py
index 642fecc35..00a690d2a 100644
--- a/tests/unit/network_services/traffic_profile/test_prox_mpls.py
+++ b/tests/unit/network_services/traffic_profile/test_prox_mpls.py
@@ -51,10 +51,11 @@ class TestProxMplsTagUntagProfile(unittest.TestCase):
         fail_tuple = ProxTestDataTuple(10.0, 1, 2, 3, 4, [5.6, 5.7, 5.8], 850, 1000, 123.4)
 
         traffic_generator = mock.MagicMock()
-        traffic_generator.run_test = target
 
         profile = ProxMplsTagUntagProfile(tp_config)
         profile.init(mock.MagicMock())
+        profile._profile_helper = profile_helper = mock.MagicMock()
+        profile_helper.run_test = target
 
         profile.execute_traffic(traffic_generator)
         self.assertEqual(round(profile.current_lower, 2), 74.69)
@@ -82,10 +83,11 @@ class TestProxMplsTagUntagProfile(unittest.TestCase):
         fail_tuple = ProxTestDataTuple(10.0, 1, 2, 3, 4, [5.6, 5.7, 5.8], 850, 1000, 123.4)
 
         traffic_generator = mock.MagicMock()
-        traffic_generator.run_test = target
 
         profile = ProxMplsTagUntagProfile(tp_config)
         profile.init(mock.MagicMock())
+        profile._profile_helper = profile_helper = mock.MagicMock()
+        profile_helper.run_test = target
 
         profile.execute_traffic(traffic_generator)
         self.assertEqual(round(profile.current_lower, 2), 24.06)