CSIT-1071 Add traffic evaluation criteria to MRR

[csit.git] / resources / libraries / python / TrafficGenerator.py

diff --git a/resources/libraries/python/TrafficGenerator.py b/resources/libraries/python/TrafficGenerator.py
index aa839dd..e7c8c48 100644 (file)
--- a/resources/libraries/python/TrafficGenerator.py
+++ b/resources/libraries/python/TrafficGenerator.py
@@ -520,9 +520,20 @@ class TrafficGenerator(AbstractRateProvider):
         :raises Exception: If loss occured.
         """
         if self._loss is None:
-            raise Exception('The traffic generation has not been issued')
+            raise RuntimeError('The traffic generation has not been issued')
         if self._loss != '0':
-            raise Exception('Traffic loss occurred: {0}'.format(self._loss))
+            raise RuntimeError('Traffic loss occurred: {0}'.format(self._loss))
+
+    def fail_if_no_traffic_forwarded(self):
+        """Fail if no traffic forwarded.
+
+        :returns: nothing
+        :raises Exception: If no traffic forwarded.
+        """
+        if self._received is None:
+            raise RuntimeError('The traffic generation has not been issued')
+        if self._received == '0':
+            raise RuntimeError('No traffic forwarded')
 
     def partial_traffic_loss_accepted(self, loss_acceptance,
                                       loss_acceptance_type):