aboutsummaryrefslogtreecommitdiffstats
path: root/tests/unit/network_services/traffic_profile/test_prox_profile.py
diff options
context:
space:
mode:
authorRoss Brattain <ross.b.brattain@intel.com>2017-09-28 17:13:34 +0000
committerGerrit Code Review <gerrit@opnfv.org>2017-09-28 17:13:34 +0000
commit0db868278e13d21998bd677bfc782557d44ab381 (patch)
treeda68d24711262194f1d2a4f4753c1ea1a290b453 /tests/unit/network_services/traffic_profile/test_prox_profile.py
parentab4ee807fae82dc0b426dfdc562a72ed71ca18e0 (diff)
parentce9b1175dd9e382c623ee91755b1943eaf0e6cdb (diff)
Merge "Addition of Prox NSB BNG and BNG-QoS test"
Diffstat (limited to 'tests/unit/network_services/traffic_profile/test_prox_profile.py')
-rw-r--r--tests/unit/network_services/traffic_profile/test_prox_profile.py9
1 files changed, 9 insertions, 0 deletions
diff --git a/tests/unit/network_services/traffic_profile/test_prox_profile.py b/tests/unit/network_services/traffic_profile/test_prox_profile.py
index 9899d9909..c32cc2878 100644
--- a/tests/unit/network_services/traffic_profile/test_prox_profile.py
+++ b/tests/unit/network_services/traffic_profile/test_prox_profile.py
@@ -26,6 +26,7 @@ stl_patch.start()
if stl_patch:
from yardstick.network_services.traffic_profile.prox_profile import ProxProfile
+ from yardstick.network_services.vnf_generic.vnf.prox_helpers import ProxResourceHelper
class TestProxProfile(unittest.TestCase):
@@ -78,6 +79,13 @@ class TestProxProfile(unittest.TestCase):
}
traffic_generator = mock.MagicMock()
+
+ setup_helper = traffic_generator.setup_helper
+ setup_helper.find_in_section.return_value = None
+
+ prox_resource_helper = ProxResourceHelper(setup_helper)
+ traffic_generator.resource_helper = prox_resource_helper
+
profile = ProxProfile(tp_config)
self.assertFalse(profile.done)
@@ -86,6 +94,7 @@ class TestProxProfile(unittest.TestCase):
profile.execute_traffic(traffic_generator)
self.assertIsNone(profile.execute_traffic(traffic_generator))
+ self.assertTrue(profile.done)
def test_bounds_iterator(self):
tp_config = {