aboutsummaryrefslogtreecommitdiffstats
path: root/core/traffic_controller_rfc2544.py
diff options
context:
space:
mode:
Diffstat (limited to 'core/traffic_controller_rfc2544.py')
-rw-r--r--core/traffic_controller_rfc2544.py9
1 files changed, 6 insertions, 3 deletions
diff --git a/core/traffic_controller_rfc2544.py b/core/traffic_controller_rfc2544.py
index 980205f6..e230c832 100644
--- a/core/traffic_controller_rfc2544.py
+++ b/core/traffic_controller_rfc2544.py
@@ -50,15 +50,18 @@ class TrafficControllerRFC2544(TrafficController, IResults):
else:
traffic['l2'] = {'framesize': packet_size}
- if traffic['traffic_type'] == 'back2back':
+ if traffic['traffic_type'] == 'rfc2544_back2back':
result = self._traffic_gen_class.send_rfc2544_back2back(
traffic, tests=self._tests, duration=self._duration, lossrate=self._lossrate)
- elif traffic['traffic_type'] == 'continuous':
+ elif traffic['traffic_type'] == 'rfc2544_continuous':
result = self._traffic_gen_class.send_cont_traffic(
traffic, duration=self._duration)
- else:
+ elif traffic['traffic_type'] == 'rfc2544_throughput':
result = self._traffic_gen_class.send_rfc2544_throughput(
traffic, tests=self._tests, duration=self._duration, lossrate=self._lossrate)
+ else:
+ raise RuntimeError("Unsupported traffic type {} was \
+ detected".format(traffic['traffic_type']))
result = self._append_results(result, packet_size)
self._results.append(result)