Merge "updating the traffic profile to enable static cgnapt for ixnet"

[yardstick.git] / tests / unit / network_services / traffic_profile / test_fixed.py

diff --git a/tests/unit/network_services/traffic_profile/test_fixed.py b/tests/unit/network_services/traffic_profile/test_fixed.py
index f4500dd..8484317 100644 (file)
--- a/tests/unit/network_services/traffic_profile/test_fixed.py
+++ b/tests/unit/network_services/traffic_profile/test_fixed.py
@@ -16,11 +16,19 @@
 #
 
 from __future__ import absolute_import
+
 import unittest
 import mock
 
-from yardstick.network_services.traffic_profile.base import TrafficProfile
-from yardstick.network_services.traffic_profile.fixed import FixedProfile
+from tests.unit import STL_MOCKS
+
+STLClient = mock.MagicMock()
+stl_patch = mock.patch.dict("sys.modules", STL_MOCKS)
+stl_patch.start()
+
+if stl_patch:
+    from yardstick.network_services.traffic_profile.base import TrafficProfile
+    from yardstick.network_services.traffic_profile.fixed import FixedProfile
 
 
 class TestFixedProfile(unittest.TestCase):