added test for bucket method

Author: chillaq

splitio/models/telemetry.py

@@ -113,11 +113,11 @@ def __init__(self):
     def _reset_all(self):
         """Reset variables"""
         with self._lock:
-            self._treatment = [0] * 23
-            self._treatments = [0] * 23
-            self._treatment_with_config = [0] * 23
-            self._treatments_with_config = [0] * 23
-            self._track = [0] * 23
+            self._treatment = [0] * MAX_LATENCY_BUCKET_COUNT
+            self._treatments = [0] * MAX_LATENCY_BUCKET_COUNT
+            self._treatment_with_config = [0] * MAX_LATENCY_BUCKET_COUNT
+            self._treatments_with_config = [0] * MAX_LATENCY_BUCKET_COUNT
+            self._track = [0] * MAX_LATENCY_BUCKET_COUNT
 
     def add_latency(self, method, latency):
         """
@@ -171,13 +171,13 @@ def __init__(self):
     def _reset_all(self):
         """Reset variables"""
         with self._lock:
-            self._split = [0] * 23
-            self._segment = [0] * 23
-            self._impression = [0] * 23
-            self._impression_count = [0] * 23
-            self._event = [0] * 23
-            self._telemetry = [0] * 23
-            self._token = [0] * 23
+            self._split = [0] * MAX_LATENCY_BUCKET_COUNT
+            self._segment = [0] * MAX_LATENCY_BUCKET_COUNT
+            self._impression = [0] * MAX_LATENCY_BUCKET_COUNT
+            self._impression_count = [0] * MAX_LATENCY_BUCKET_COUNT
+            self._event = [0] * MAX_LATENCY_BUCKET_COUNT
+            self._telemetry = [0] * MAX_LATENCY_BUCKET_COUNT
+            self._token = [0] * MAX_LATENCY_BUCKET_COUNT
 
     def add_latency(self, resource, latency):
         """

tests/models/test_telemetry_model.py

@@ -4,13 +4,37 @@
 
 from splitio.models.telemetry import StorageType, OperationMode, MethodLatencies, MethodExceptions, \
     HTTPLatencies, HTTPErrors, LastSynchronization, TelemetryCounters, TelemetryConfig, \
-    StreamingEvent, StreamingEvents
+    StreamingEvent, StreamingEvents, get_latency_bucket_index
 
 import splitio.models.telemetry as ModelTelemetry
 
 class TelemetryModelTests(object):
     """Telemetry model test cases."""
 
+    def test_latency_bucket_index(self):
+        for i in range(50000):
+            latency = random.randint(10, 9987885)
+            old_bucket = 0
+            result_bucket = 0
+            counter = -1
+            for j in ModelTelemetry.BUCKETS:
+                counter = counter + 1
+                if old_bucket == 0:
+                    if latency < j:
+                        old_bucket = 0
+                        break
+                    old_bucket = j
+                    continue
+                if counter == ModelTelemetry.MAX_LATENCY_BUCKET_COUNT - 1:
+                    result_bucket = 22
+                    break
+                if latency > old_bucket and latency <= j:
+                    result_bucket = counter
+                    break
+                old_bucket = j
+            print(latency, old_bucket, j)
+            assert(result_bucket == ModelTelemetry.get_latency_bucket_index(latency))
+
     def test_storage_type_and_operation_mode(self, mocker):
         assert(StorageType.LOCALHOST == 'localhost')
         assert(StorageType.MEMEORY == 'memory')