Merge pull request #822 from superannotateai/FRIDAY_4586

Add SA_ENV var handling

Author: VaghinakDev

src/superannotate/lib/infrastructure/services/http_client.py

@@ -50,6 +50,7 @@ def __init__(self, api_url: str, token: str, verify_ssl: bool = True):
         super().__init__(api_url, token)
         self._verify_ssl = verify_ssl
         self._version = os.environ.get("sa_version")
+        self._env = os.environ.get("SA_ENV")
 
     @property
     def verify_ssl(self):
@@ -86,7 +87,8 @@ def default_headers(self):
                 ).encode("utf-8")
             ).decode("utf-8"),
             "User-Agent": f"Python-SDK-Version: {self._version}; Python: {platform.python_version()};"
-            f"OS: {platform.system()}; Team: {self.team_id}",
+            f"OS: {platform.system()}; Team: {self.team_id}"
+            f"{'; Env: ' + self._env if self._env else ''}",
         }
 
     @property

tests/integration/work_management/test_list_users.py

@@ -1,5 +1,4 @@
 import pytest
-
 from superannotate import SAClient
 from tests.integration.base import BaseTestCase
 

tests/integration/work_management/test_pause_resume_user_activity.py

@@ -1,5 +1,3 @@
-from unittest import skip
-
 from lib.core.exceptions import AppException
 from src.superannotate import SAClient
 from tests.integration.base import BaseTestCase

tests/unit/test_http_client.py

@@ -0,0 +1,52 @@
+import os
+import platform
+from unittest import TestCase
+from unittest.mock import patch
+
+from src.superannotate.lib.infrastructure.services.http_client import HttpClient
+
+
+class TestHttpClient(TestCase):
+    def setUp(self):
+        self.api_url = "https://api.example.com"
+        self.team_id = 123
+        self.token = f"test_token={self.team_id}"
+
+    @patch.dict(os.environ, {"sa_version": "1.0.0", "SA_ENV": "test"})
+    def test_default_headers_with_env(self):
+        client = HttpClient(self.api_url, self.token)
+        headers = client.default_headers
+
+        expected_user_agent = (
+            f"Python-SDK-Version: 1.0.0; Python: {platform.python_version()};"
+            f"OS: {platform.system()}; Team: {self.team_id}; Env: test"
+        )
+
+        assert headers["Authorization"] == self.token
+        assert headers["authtype"] == "sdk"
+        assert headers["Content-Type"] == "application/json"
+        assert headers["User-Agent"] == expected_user_agent
+
+    @patch.dict(os.environ, {"sa_version": "2.0.0"}, clear=True)
+    def test_default_headers_without_env(self):
+        client = HttpClient(self.api_url, self.token)
+        headers = client.default_headers
+
+        expected_user_agent = (
+            f"Python-SDK-Version: 2.0.0; Python: {platform.python_version()};"
+            f"OS: {platform.system()}; Team: {self.team_id}"
+        )
+
+        assert headers["User-Agent"] == expected_user_agent
+        assert "Env:" not in headers["User-Agent"]
+
+    def test_default_headers_no_version(self):
+        with patch.dict(os.environ, {}, clear=True):
+            client = HttpClient(self.api_url, self.token)
+            headers = client.default_headers
+
+            expected_user_agent = (
+                f"Python-SDK-Version: None; Python: {platform.python_version()};"
+                f"OS: {platform.system()}; Team: {self.team_id}"
+            )
+            assert headers["User-Agent"] == expected_user_agent