Merge pull request #428 from atlanhq/FT-773

FT-773: Add X-Atlan-Client-Origin header to session requests

pyatlan/client/atlan.py

@@ -132,6 +132,7 @@ def get_session():
         {
             "x-atlan-agent": "sdk",
             "x-atlan-agent-id": "python",
+            "x-atlan-client-origin": "product_sdk",
             "User-Agent": f"Atlan-PythonSDK/{VERSION}",
         }
     )

tests/unit/test_client.py

@@ -1,5 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0
 # Copyright 2022 Atlan Pte. Ltd.
+from importlib.resources import read_text
 from json import load, loads
 from pathlib import Path
 from unittest.mock import DEFAULT, Mock, call, patch
@@ -2204,3 +2205,17 @@ def test_asset_attribute_none_assignment(self):
         assert request_json
         assert request_json["attributes"]["certificateStatus"] is None
         assert request_json["attributes"]["certificateStatusMessage"] is None
+
+
+def test_atlan_client_headers(client: AtlanClient):
+    VERSION = read_text("pyatlan", "version.txt").strip()
+    expected = {
+        "User-Agent": f"Atlan-PythonSDK/{VERSION}",
+        "Accept-Encoding": "gzip, deflate",
+        "Accept": "*/*",
+        "Connection": "keep-alive",
+        "x-atlan-agent": "sdk",
+        "x-atlan-agent-id": "python",
+        "x-atlan-client-origin": "product_sdk",
+    }
+    assert expected == client._session.headers