feat: Add tests for packet loss

alessandromaggio · Nov 8, 2020 · b46bcef · b46bcef
1 parent dfc6777
commit b46bcef
Showing 1 changed file with 41 additions and 0 deletions.
diff --git a/test/test_executor.py b/test/test_executor.py
@@ -247,6 +247,47 @@ def test_success_half_success(self):
             'Unable to calculate success on all with half responses successful'
         )
 
+    def test_no_packets_lost(self):
+        rs = executor.ResponseList([
+            SuccessfulResponseMock(None, 1),
+            SuccessfulResponseMock(None, 1),
+            SuccessfulResponseMock(None, 1),
+            SuccessfulResponseMock(None, 1)
+        ])
+
+        self.assertEqual(
+            rs.packet_loss, 
+            0.0, 
+            "Unable to calculate packet loss correctly when all resposes successful"
+        )
+
+    def test_all_packets_lost(self):
+        rs = executor.ResponseList([
+            FailingResponseMock(None, 1),
+            FailingResponseMock(None, 1),
+            FailingResponseMock(None, 1),
+            FailingResponseMock(None, 1)
+        ])
+
+        self.assertEqual(
+            rs.packet_loss, 
+            1.0, 
+            "Unable to calculate packet loss correctly when all resposes failed"
+        )
+
+    def test_some_packets_lost(self):
+        rs = executor.ResponseList([
+            SuccessfulResponseMock(None, 1),
+            SuccessfulResponseMock(None, 1),
+            FailingResponseMock(None, 1),
+            FailingResponseMock(None, 1)
+        ])
+
+        self.assertEqual(
+            rs.packet_loss, 
+            0.5, 
+            "Unable to calculate packet loss correctly when some of the responses failed"
+        )
 
 class CommunicatorTestCase(unittest.TestCase):
     """Tests for Communicator"""