[NFVBENCH-165] Use STLVmFixIpv4 to fix inner IP/UDP payload

[nfvbench.git] / test / mock_trex.py

diff --git a/test/mock_trex.py b/test/mock_trex.py
index ed6b20e..49fe5a9 100644 (file)
--- a/test/mock_trex.py
+++ b/test/mock_trex.py
@@ -30,8 +30,6 @@ except ImportError:
     class STLDummy(Exception):
         """Dummy class."""
 
-        pass
-
     trex_lib_mod = ModuleType('trex')
     sys.modules['trex'] = trex_lib_mod
     stl_lib_mod = ModuleType('trex.stl')
@@ -54,6 +52,7 @@ except ImportError:
     api_mod.STLStream = STLDummy
     api_mod.STLTXCont = STLDummy
     api_mod.STLVmFixChecksumHw = STLDummy
+    api_mod.STLVmFixIpv4 = STLDummy
     api_mod.STLVmFlowVar = STLDummy
     api_mod.STLVmFlowVarRepeatableRandom = STLDummy
     api_mod.STLVmWrFlowVar = STLDummy
@@ -77,4 +76,3 @@ except ImportError:
 
 def no_op():
     """Empty function."""
-    pass