NFVBENCH-153 Add support for python3

[nfvbench.git] / test / test_nfvbench.py

diff --git a/test/test_nfvbench.py b/test/test_nfvbench.py
index e41ab33..4a8a574 100644 (file)
--- a/test/test_nfvbench.py
+++ b/test/test_nfvbench.py
@@ -13,8 +13,6 @@
 #    License for the specific language governing permissions and limitations
 #    under the License.
 #
-from mock_trex import no_op
-
 import json
 import logging
 import sys
@@ -34,6 +32,7 @@ from nfvbench.traffic_client import IpBlock
 from nfvbench.traffic_client import TrafficClient
 import nfvbench.traffic_gen.traffic_utils as traffic_utils
 
+from .mock_trex import no_op
 
 # just to get rid of the unused function warning
 no_op()
@@ -331,7 +330,11 @@ def _get_dummy_tg_config(chain_type, rate, scc=1, fc=10, step_ip='0.0.0.1',
         'cores': None,
         'mbuf_factor': None,
         'disable_hdrh': None,
-        'mbuf_64': None
+        'mbuf_64': None,
+        'service_mode': False,
+        'no_flow_stats': False,
+        'no_latency_stats': False,
+        'no_latency_streams': False
 
     })