X-Git-Url: https://gerrit.opnfv.org/gerrit/gitweb?a=blobdiff_plain;f=yardstick%2Ftests%2Funit%2F__init__.py;h=c05f91c81e67fa74f517dcd83b519bb35618b139;hb=0aa73e85efc60a9f5f0dc1f063aad1d342a123ca;hp=59318aba72dbabeaace2f0b12df62d1a74863630;hpb=ecfb54a60eb8e4395b582f114c92f69f547c3e45;p=yardstick.git

diff --git a/yardstick/tests/unit/__init__.py b/yardstick/tests/unit/__init__.py
index 59318aba7..c05f91c81 100644
--- a/yardstick/tests/unit/__init__.py
+++ b/yardstick/tests/unit/__init__.py
@@ -16,65 +16,8 @@ import sys
 
 import mock
 
+from yardstick import tests
 
-STL_MOCKS = {
-    'trex_stl_lib': mock.MagicMock(),
-    'trex_stl_lib.base64': mock.MagicMock(),
-    'trex_stl_lib.binascii': mock.MagicMock(),
-    'trex_stl_lib.collections': mock.MagicMock(),
-    'trex_stl_lib.copy': mock.MagicMock(),
-    'trex_stl_lib.datetime': mock.MagicMock(),
-    'trex_stl_lib.functools': mock.MagicMock(),
-    'trex_stl_lib.imp': mock.MagicMock(),
-    'trex_stl_lib.inspect': mock.MagicMock(),
-    'trex_stl_lib.json': mock.MagicMock(),
-    'trex_stl_lib.linecache': mock.MagicMock(),
-    'trex_stl_lib.math': mock.MagicMock(),
-    'trex_stl_lib.os': mock.MagicMock(),
-    'trex_stl_lib.platform': mock.MagicMock(),
-    'trex_stl_lib.pprint': mock.MagicMock(),
-    'trex_stl_lib.random': mock.MagicMock(),
-    'trex_stl_lib.re': mock.MagicMock(),
-    'trex_stl_lib.scapy': mock.MagicMock(),
-    'trex_stl_lib.socket': mock.MagicMock(),
-    'trex_stl_lib.string': mock.MagicMock(),
-    'trex_stl_lib.struct': mock.MagicMock(),
-    'trex_stl_lib.sys': mock.MagicMock(),
-    'trex_stl_lib.threading': mock.MagicMock(),
-    'trex_stl_lib.time': mock.MagicMock(),
-    'trex_stl_lib.traceback': mock.MagicMock(),
-    'trex_stl_lib.trex_stl_async_client': mock.MagicMock(),
-    'trex_stl_lib.trex_stl_client': mock.MagicMock(),
-    'trex_stl_lib.trex_stl_exceptions': mock.MagicMock(),
-    'trex_stl_lib.trex_stl_ext': mock.MagicMock(),
-    'trex_stl_lib.trex_stl_jsonrpc_client': mock.MagicMock(),
-    'trex_stl_lib.trex_stl_packet_builder_interface': mock.MagicMock(),
-    'trex_stl_lib.trex_stl_packet_builder_scapy': mock.MagicMock(),
-    'trex_stl_lib.trex_stl_port': mock.MagicMock(),
-    'trex_stl_lib.trex_stl_stats': mock.MagicMock(),
-    'trex_stl_lib.trex_stl_streams': mock.MagicMock(),
-    'trex_stl_lib.trex_stl_types': mock.MagicMock(),
-    'trex_stl_lib.types': mock.MagicMock(),
-    'trex_stl_lib.utils': mock.MagicMock(),
-    'trex_stl_lib.utils.argparse': mock.MagicMock(),
-    'trex_stl_lib.utils.collections': mock.MagicMock(),
-    'trex_stl_lib.utils.common': mock.MagicMock(),
-    'trex_stl_lib.utils.json': mock.MagicMock(),
-    'trex_stl_lib.utils.os': mock.MagicMock(),
-    'trex_stl_lib.utils.parsing_opts': mock.MagicMock(),
-    'trex_stl_lib.utils.pwd': mock.MagicMock(),
-    'trex_stl_lib.utils.random': mock.MagicMock(),
-    'trex_stl_lib.utils.re': mock.MagicMock(),
-    'trex_stl_lib.utils.string': mock.MagicMock(),
-    'trex_stl_lib.utils.sys': mock.MagicMock(),
-    'trex_stl_lib.utils.text_opts': mock.MagicMock(),
-    'trex_stl_lib.utils.text_tables': mock.MagicMock(),
-    'trex_stl_lib.utils.texttable': mock.MagicMock(),
-    'trex_stl_lib.warnings': mock.MagicMock(),
-    'trex_stl_lib.yaml': mock.MagicMock(),
-    'trex_stl_lib.zlib': mock.MagicMock(),
-    'trex_stl_lib.zmq': mock.MagicMock(),
-}
 
-mock_stl = mock.patch.dict(sys.modules, STL_MOCKS)
+mock_stl = mock.patch.dict(sys.modules, tests.STL_MOCKS)
 mock_stl.start()