feat: new headers and payload ( connection id and interval)

UnleashClient/__init__.py

@@ -110,6 +110,7 @@ def __init__(
         self.unleash_app_name = app_name
         self.unleash_environment = environment
         self.unleash_instance_id = instance_id
+        self.connection_id = str(uuid.uuid4())
         self.unleash_refresh_interval = refresh_interval
         self.unleash_request_timeout = request_timeout
         self.unleash_request_retries = request_retries
@@ -183,6 +184,10 @@ def __init__(
                 engine=self.engine,
             )
 
+    @property
+    def unleash_refresh_interval_str_millis(self) -> str:
+        return str(self.unleash_refresh_interval * 1000)
+
     def initialize_client(self, fetch_toggles: bool = True) -> None:
         """
         Initializes client and starts communication with central unleash server(s).
@@ -215,7 +220,8 @@ def initialize_client(self, fetch_toggles: bool = True) -> None:
             try:
                 headers = {
                     **self.unleash_custom_headers,
-                    "unleash-connection-id": str(uuid.uuid4()),
+                    "unleash-connection-id": self.connection_id,
+                    "unleash-interval-id": self.unleash_refresh_interval_str_millis,
                     "unleash-appname": self.unleash_app_name,
                     "unleash-sdk": f"{SDK_NAME}:{SDK_VERSION}",
                 }
@@ -224,6 +230,7 @@ def initialize_client(self, fetch_toggles: bool = True) -> None:
                 metrics_args = {
                     "url": self.unleash_url,
                     "app_name": self.unleash_app_name,
+                    "connection_id": self.connection_id,
                     "instance_id": self.unleash_instance_id,
                     "headers": headers,
                     "custom_options": self.unleash_custom_options,
@@ -237,6 +244,7 @@ def initialize_client(self, fetch_toggles: bool = True) -> None:
                         self.unleash_url,
                         self.unleash_app_name,
                         self.unleash_instance_id,
+                        self.connection_id,
                         self.unleash_metrics_interval,
                         headers,
                         self.unleash_custom_options,

UnleashClient/api/register.py

@@ -21,6 +21,7 @@ def register_client(
     url: str,
     app_name: str,
     instance_id: str,
+    connection_id: str,
     metrics_interval: int,
     headers: dict,
     custom_options: dict,
@@ -47,6 +48,7 @@ def register_client(
     registration_request = {
         "appName": app_name,
         "instanceId": instance_id,
+        "connectionId": connection_id,
         "sdkVersion": f"{SDK_NAME}:{SDK_VERSION}",
         "strategies": [*supported_strategies],
         "started": datetime.now(timezone.utc).isoformat(),

UnleashClient/periodic_tasks/send_metrics.py

@@ -12,6 +12,7 @@ def aggregate_and_send_metrics(
     url: str,
     app_name: str,
     instance_id: str,
+    connection_id: str,
     headers: dict,
     custom_options: dict,
     request_timeout: int,
@@ -22,6 +23,7 @@ def aggregate_and_send_metrics(
     metrics_request = {
         "appName": app_name,
         "instanceId": instance_id,
+        "connectionId": connection_id,
         "bucket": metrics_bucket,
         "platformName": python_implementation(),
         "platformVersion": python_version(),

tests/unit_tests/api/test_metrics.py

@@ -1,3 +1,5 @@
+import json
+
 import responses
 from pytest import mark, param
 from requests import ConnectionError
@@ -36,5 +38,9 @@ def test_send_metrics(payload, status, expected):
         URL, MOCK_METRICS_REQUEST, CUSTOM_HEADERS, CUSTOM_OPTIONS, REQUEST_TIMEOUT
     )
 
+    request = json.loads(responses.calls[0].request.body)
+
     assert len(responses.calls) == 1
     assert expected(result)
+
+    assert request["connectionId"] == MOCK_METRICS_REQUEST.get("connectionId")

tests/unit_tests/api/test_register.py

@@ -6,6 +6,7 @@
 
 from tests.utilities.testing_constants import (
     APP_NAME,
+    CONNECTION_ID,
     CUSTOM_HEADERS,
     CUSTOM_OPTIONS,
     INSTANCE_ID,
@@ -40,6 +41,7 @@ def test_register_client(payload, status, expected):
         URL,
         APP_NAME,
         INSTANCE_ID,
+        CONNECTION_ID,
         METRICS_INTERVAL,
         CUSTOM_HEADERS,
         CUSTOM_OPTIONS,
@@ -59,6 +61,7 @@ def test_register_includes_metadata():
         URL,
         APP_NAME,
         INSTANCE_ID,
+        CONNECTION_ID,
         METRICS_INTERVAL,
         CUSTOM_HEADERS,
         CUSTOM_OPTIONS,
@@ -71,5 +74,6 @@ def test_register_includes_metadata():
 
     assert request["yggdrasilVersion"] is not None
     assert request["specVersion"] == CLIENT_SPEC_VERSION
+    assert request["connectionId"] == CONNECTION_ID
     assert request["platformName"] is not None
     assert request["platformVersion"] is not None

tests/unit_tests/test_client.py

@@ -1,3 +1,4 @@
+import json
 import time
 import warnings
 from datetime import datetime, timezone
@@ -1078,3 +1079,37 @@ def test_identification_headers_unique_connection_id():
         "UNLEASH-CONNECTION-ID"
     ]
     assert connection_id_first_client != connection_id_second_client
+
+
+@responses.activate
+def test_identification_values_are_passed_in():
+    responses.add(responses.POST, URL + REGISTER_URL, json={}, status=202)
+    responses.add(
+        responses.GET, URL + FEATURES_URL, json=MOCK_FEATURE_RESPONSE, status=200
+    )
+    responses.add(responses.POST, URL + METRICS_URL, json={}, status=202)
+
+    input_interval = 1
+    unleash_client = UnleashClient(
+        URL, APP_NAME, refresh_interval=input_interval, metrics_interval=1
+    )
+    expected_connection_id = unleash_client.connection_id
+    expected_interval = str(input_interval * 1000)
+
+    unleash_client.initialize_client()
+    register_request = responses.calls[0].request
+    reqister_body = json.loads(register_request.body)
+
+    assert register_request.headers["UNLEASH-CONNECTION-ID"] == expected_connection_id
+    assert reqister_body["connectionId"] == expected_connection_id
+
+    features_request = responses.calls[1].request
+
+    assert features_request.headers["UNLEASH-CONNECTION-ID"] == expected_connection_id
+    assert features_request.headers["UNLEASH-INTERVAL-ID"] == expected_interval
+
+    time.sleep(2)
+    metrics_request = responses.calls[2].request
+
+    assert metrics_request.headers["UNLEASH-CONNECTION-ID"] == expected_connection_id
+    assert metrics_request.headers["UNLEASH-INTERVAL-ID"] == expected_interval

tests/utilities/mocks/mock_metrics.py

@@ -1,6 +1,7 @@
 MOCK_METRICS_REQUEST = {
     "appName": "appName",
     "instanceId": "instanceId",
+    "connectionId": "connectionId",
     "bucket": {
         "start": "2016-11-03T07:16:43.572Z",
         "stop": "2016-11-03T07:16:53.572Z",

tests/utilities/testing_constants.py

@@ -4,6 +4,7 @@
 APP_NAME = "pytest"
 ENVIRONMENT = "unit"
 INSTANCE_ID = "123"
+CONNECTION_ID = "test-connection-id"
 REFRESH_INTERVAL = 3
 REFRESH_JITTER = None
 METRICS_INTERVAL = 2