atlan-application-sdk 0.1.1rc27__tar.gz → 0.1.1rc28__tar.gz

atlan_application_sdk-0.1.1rc28/.github/release_notes.md

@@ -0,0 +1,7 @@
+## v0.1.1rc28 (August 12, 2025)
+
+Full Changelog: https://github.com/atlanhq/application-sdk/compare/v0.1.1rc27...v0.1.1rc28
+
+### Features
+
+- add pyatlan Client to Application SDK (#642) (by @Ernest Hill in [c00d682](https://github.com/atlanhq/application-sdk/commit/c00d682))

{atlan_application_sdk-0.1.1rc27 → atlan_application_sdk-0.1.1rc28}/.github/workflows/codeql.yaml

@@ -25,7 +25,7 @@ jobs:
         language: ["python"]
     steps:
       - name: Checkout repository
-        uses: actions/checkout@v4
+        uses: actions/checkout@v5
       - name: Initialize CodeQL
         uses: github/codeql-action/init@v3
         with:
@@ -43,6 +43,6 @@ jobs:
     name: Pre-commit Checks
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
       - uses: "./.github/actions/setup-deps"
       - uses: pre-commit/action@v3.0.1

{atlan_application_sdk-0.1.1rc27 → atlan_application_sdk-0.1.1rc28}/.github/workflows/pre-release.yaml

@@ -18,7 +18,7 @@ jobs:
 
     steps:
       - name: Checkout code
-        uses: actions/checkout@v3
+        uses: actions/checkout@v5
         with:
           fetch-depth: 0  # Need full history for changelog
 

{atlan_application_sdk-0.1.1rc27 → atlan_application_sdk-0.1.1rc28}/.github/workflows/pull_request.yaml

@@ -19,7 +19,7 @@ jobs:
     name: Conventional Commits
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
         with:
           fetch-depth: 0
       - uses: webiny/action-conventional-commits@v1.3.0
@@ -30,7 +30,7 @@ jobs:
     name: Docstring Coverage
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
       - uses: "./.github/actions/docstring-coverage"
         with:
           module-name: application_sdk
@@ -41,7 +41,7 @@ jobs:
     name: Trivy Code Scan
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
       - uses: "./.github/actions/trivy"
         with:
           add-report-comment-to-pr: "true"
@@ -57,7 +57,7 @@ jobs:
         python-version: ["3.11", "3.12", "3.13"]
         os: [ubuntu-22.04, macOS-latest, windows-latest]
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
       - uses: "./.github/actions/unit-tests"
         with:
           github-token: ${{ secrets.ORG_PAT_GITHUB }}
@@ -71,7 +71,7 @@ jobs:
       matrix: ${{ steps.set-matrix.outputs.matrix }}
     steps:
       - name: Checkout repository
-        uses: actions/checkout@v4
+        uses: actions/checkout@v5
 
       - name: Get PR Labels
         id: get-labels
@@ -150,7 +150,7 @@ jobs:
       matrix:
         os: [ubuntu-22.04, macOS-latest, windows-latest]
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
       - uses: "./.github/actions/e2e-examples"
         with:
           operating-system: ${{ matrix.os }}
@@ -170,7 +170,7 @@ jobs:
       fail-fast: false
       matrix: ${{ fromJson(needs.matrix-builder.outputs.matrix) }}
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
       - uses: "./.github/actions/e2e-apps"
         with:
           github-token: ${{ secrets.ORG_PAT_GITHUB }}
@@ -182,7 +182,7 @@ jobs:
     name: Docs
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
       - uses: "./.github/actions/setup-deps"
       - name: Install graphviz
         run: |

{atlan_application_sdk-0.1.1rc27 → atlan_application_sdk-0.1.1rc28}/.github/workflows/push.yaml

@@ -20,7 +20,7 @@ jobs:
     name: Unit Tests
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
       - uses: "./.github/actions/unit-tests"
         with:
           github-token: ${{ secrets.ORG_PAT_GITHUB }}
@@ -30,7 +30,7 @@ jobs:
     name: E2E (Examples)
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
       - uses: "./.github/actions/e2e-examples"
         with:
           operating-system: ubuntu-latest
@@ -48,7 +48,7 @@ jobs:
       matrix:
         repo_name: ["atlan-postgres-app"]
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
       - uses: "./.github/actions/e2e-apps"
         with:
           github-token: ${{ secrets.ORG_PAT_GITHUB }}
@@ -60,7 +60,7 @@ jobs:
     name: Docs
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
       - uses: "./.github/actions/setup-deps"
       - name: Install graphviz
         run: |

{atlan_application_sdk-0.1.1rc27 → atlan_application_sdk-0.1.1rc28}/.github/workflows/scale-tests.yaml

@@ -23,7 +23,7 @@ jobs:
     timeout-minutes: 15
 
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
         with:
           fetch-depth: 0
 

{atlan_application_sdk-0.1.1rc27 → atlan_application_sdk-0.1.1rc28}/.github/workflows/schedule.yaml

@@ -27,5 +27,5 @@ jobs:
     name: Trivy Code Scan
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
       - uses: "./.github/actions/trivy"

{atlan_application_sdk-0.1.1rc27 → atlan_application_sdk-0.1.1rc28}/.github/workflows/tag-and-publish.yaml

@@ -18,7 +18,7 @@ jobs:
 
     steps:
       - name: Checkout code
-        uses: actions/checkout@v3
+        uses: actions/checkout@v5
         with:
           fetch-depth: 0  # Need full history for release notes
 

{atlan_application_sdk-0.1.1rc27 → atlan_application_sdk-0.1.1rc28}/.github/workflows/trivy-container.yaml

@@ -25,7 +25,7 @@ jobs:
     runs-on: "ubuntu-24.04"
     steps:
       - name: Checkout code
-        uses: actions/checkout@v4
+        uses: actions/checkout@v5
 
       - uses: "atlanhq/application-sdk/.github/actions/trivy-container@main"
         with:

{atlan_application_sdk-0.1.1rc27 → atlan_application_sdk-0.1.1rc28}/CHANGELOG.md

@@ -1,5 +1,14 @@
 # Changelog
 
+
+## v0.1.1rc28 (August 12, 2025)
+
+Full Changelog: https://github.com/atlanhq/application-sdk/compare/v0.1.1rc27...v0.1.1rc28
+
+### Features
+
+- add pyatlan Client to Application SDK (#642) (by @Ernest Hill in [c00d682](https://github.com/atlanhq/application-sdk/commit/c00d682))
+
 ## v0.1.1rc27 (August 12, 2025)
 
 Full Changelog: https://github.com/atlanhq/application-sdk/compare/v0.1.1rc26...v0.1.1rc27

{atlan_application_sdk-0.1.1rc27 → atlan_application_sdk-0.1.1rc28}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: atlan-application-sdk
-Version: 0.1.1rc27
+Version: 0.1.1rc28
 Summary: Atlan Application SDK is a Python library for developing applications on the Atlan Platform
 Project-URL: Repository, https://github.com/atlanhq/application-sdk
 Project-URL: Documentation, https://github.com/atlanhq/application-sdk/README.md
@@ -26,7 +26,7 @@ Requires-Dist: fastapi[standard]>=0.115.0
 Requires-Dist: loguru>=0.7.3
 Requires-Dist: opentelemetry-exporter-otlp>=1.27.0
 Requires-Dist: psutil>=7.0.0
-Requires-Dist: pyatlan>=7.1.1
+Requires-Dist: pyatlan>=7.1.6
 Requires-Dist: pydantic>=2.10.6
 Requires-Dist: python-dotenv>=1.1.0
 Requires-Dist: uvloop>=0.21.0; sys_platform != 'win32'

atlan_application_sdk-0.1.1rc28/application_sdk/clients/atlan_client.py

@@ -0,0 +1,70 @@
+from typing import Optional
+
+from pyatlan.client.atlan import AtlanClient
+
+from application_sdk.common.error_codes import ClientError
+from application_sdk.constants import (
+    ATLAN_API_KEY,
+    ATLAN_API_TOKEN_GUID,
+    ATLAN_BASE_URL,
+    ATLAN_CLIENT_ID,
+    ATLAN_CLIENT_SECRET,
+)
+from application_sdk.observability.logger_adaptor import get_logger
+
+logger = get_logger(__name__)
+
+
+def get_client(
+    base_url: Optional[str] = None,
+    api_key: Optional[str] = None,
+    api_token_guid: Optional[str] = None,
+) -> AtlanClient:
+    """
+    Returns an authenticated AtlanClient instance using provided parameters or environment variables.
+
+    Selects authentication method based on the presence of parameters or environment variables and validates the required configuration.
+    In general, the use of environment variables is recommended. Any parameters specified will override the environment variables.
+
+    Args:
+    base_url: Atlan base URL (overrides ATLAN_BASE_URL)
+    api_key: Atlan API key (overrides ATLAN_API_KEY)
+    api_token_guid: API token GUID (overrides API_TOKEN_GUID)
+    """
+    # Resolve final values (parameters override env vars)
+    final_token_guid = api_token_guid or ATLAN_API_TOKEN_GUID
+    final_base_url = base_url or ATLAN_BASE_URL
+    final_api_key = api_key or ATLAN_API_KEY
+
+    # Priority 1: Token-based auth (recommended for production)
+    if final_token_guid:
+        if final_base_url or final_api_key:  # Check original params, not env vars
+            logger.warning(
+                "Token auth takes precedence - ignoring base_url/api_key parameters as well as ATLAN_BASE_URL and ATLAN_API_KEY environment variables."
+            )
+        return _get_client_from_token(final_token_guid)
+
+    # Priority 2: API key + base URL auth
+    if not final_base_url:
+        raise ClientError(
+            "ATLAN_BASE_URL is required (via parameter or environment variable)"
+        )
+    if not final_api_key:
+        raise ClientError(
+            "ATLAN_API_KEY is required (via parameter or environment variable)"
+        )
+
+    logger.info("Using API key-based authentication")
+    return AtlanClient(base_url=final_base_url, api_key=final_api_key)
+
+
+def _get_client_from_token(api_token_guid: str):
+    if not ATLAN_CLIENT_ID:
+        raise ClientError(
+            f"{ClientError.AUTH_CONFIG_ERROR}: Environment variable CLIENT_ID is required when API_TOKEN_GUID is set."
+        )
+    if not ATLAN_CLIENT_SECRET:
+        raise ClientError(
+            f"{ClientError.AUTH_CONFIG_ERROR}: Environment variable CLIENT_SECRET is required when API_TOKEN_GUID is set."
+        )
+    return AtlanClient.from_token_guid(guid=api_token_guid)

{atlan_application_sdk-0.1.1rc27 → atlan_application_sdk-0.1.1rc28}/application_sdk/constants.py

@@ -217,3 +217,10 @@ TRACES_FILE_NAME = "traces.parquet"
 ENABLE_OBSERVABILITY_DAPR_SINK = (
     os.getenv("ATLAN_ENABLE_OBSERVABILITY_DAPR_SINK", "false").lower() == "true"
 )
+
+# atlan_client configuration (non ATLAN_ prefix are rooted in pyatlan SDK, to be revisited)
+ATLAN_API_TOKEN_GUID = os.getenv("API_TOKEN_GUID")
+ATLAN_BASE_URL = os.getenv("ATLAN_BASE_URL")
+ATLAN_API_KEY = os.getenv("ATLAN_API_KEY")
+ATLAN_CLIENT_ID = os.getenv("CLIENT_ID")
+ATLAN_CLIENT_SECRET = os.getenv("CLIENT_SECRET")

{atlan_application_sdk-0.1.1rc27 → atlan_application_sdk-0.1.1rc28}/application_sdk/version.py

@@ -2,4 +2,4 @@
 Version information for the application_sdk package.
 """
 
-__version__ = "0.1.1rc27"
+__version__ = "0.1.1rc28"

{atlan_application_sdk-0.1.1rc27 → atlan_application_sdk-0.1.1rc28}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "atlan-application-sdk"
-version = "0.1.1rc27"
+version = "0.1.1rc28"
 description = "Atlan Application SDK is a Python library for developing applications on the Atlan Platform"
 license = "Apache-2.0"
 authors = [
@@ -24,7 +24,7 @@ dependencies = [
     "opentelemetry-exporter-otlp>=1.27.0",
     "psutil>=7.0.0",
     "fastapi[standard]>=0.115.0",
-    "pyatlan>=7.1.1",
+    "pyatlan>=7.1.6",
     "pydantic>=2.10.6",
     "loguru>=0.7.3",
     "uvloop>=0.21.0; sys_platform != 'win32'",

atlan_application_sdk-0.1.1rc28/tests/unit/clients/test_atlan_client.py

@@ -0,0 +1,303 @@
+from unittest.mock import MagicMock, patch
+
+import pytest
+
+from application_sdk.clients.atlan_client import get_client
+from application_sdk.common.error_codes import ClientError
+
+
+@pytest.mark.parametrize(
+    "params,constants,msg",
+    [
+        # Error: missing base_url
+        (
+            {"base_url": None, "api_key": "api_key_789", "api_token_guid": None},
+            {
+                "ATLAN_API_TOKEN_GUID": None,
+                "ATLAN_BASE_URL": None,
+                "ATLAN_API_KEY": None,
+                "ATLAN_CLIENT_ID": None,
+                "ATLAN_CLIENT_SECRET": None,
+            },
+            "ATLAN_BASE_URL is required",
+        ),
+        # Error: missing api_key
+        (
+            {"base_url": "https://atlan.com", "api_key": None, "api_token_guid": None},
+            {
+                "ATLAN_API_TOKEN_GUID": None,
+                "ATLAN_BASE_URL": None,
+                "ATLAN_API_KEY": None,
+                "ATLAN_CLIENT_ID": None,
+                "ATLAN_CLIENT_SECRET": None,
+            },
+            "ATLAN_API_KEY is required",
+        ),
+        # Error: missing both base_url and api_key
+        (
+            {"base_url": None, "api_key": None, "api_token_guid": None},
+            {
+                "ATLAN_API_TOKEN_GUID": None,
+                "ATLAN_BASE_URL": None,
+                "ATLAN_API_KEY": None,
+                "ATLAN_CLIENT_ID": None,
+                "ATLAN_CLIENT_SECRET": None,
+            },
+            "ATLAN_BASE_URL is required",
+        ),
+        # Error: missing client_id
+        (
+            {"base_url": None, "api_key": None, "api_token_guid": "guid_param"},
+            {
+                "ATLAN_API_TOKEN_GUID": None,
+                "ATLAN_BASE_URL": None,
+                "ATLAN_API_KEY": None,
+                "ATLAN_CLIENT_ID": None,
+                "ATLAN_CLIENT_SECRET": None,
+            },
+            "Environment variable CLIENT_ID is required when API_TOKEN_GUID is set",
+        ),
+        # Error: missing client_secret
+        (
+            {"base_url": None, "api_key": None, "api_token_guid": "guid_param"},
+            {
+                "ATLAN_API_TOKEN_GUID": None,
+                "ATLAN_BASE_URL": None,
+                "ATLAN_API_KEY": None,
+                "ATLAN_CLIENT_ID": "CLIENT_ID_789",
+                "ATLAN_CLIENT_SECRET": None,
+            },
+            "Environment variable CLIENT_SECRET is required when API_TOKEN_GUID is set",
+        ),
+    ],
+    ids=[
+        "missing_base_url",
+        "missing_api_key",
+        "missing_both_base_url_and_api_key",
+        "missing_client_id",
+        "missing_client_secret",
+    ],
+)
+def test_get_client_bad_params(params, constants, msg):
+    # Arrange
+    with patch(
+        "application_sdk.clients.atlan_client.ATLAN_API_TOKEN_GUID",
+        constants["ATLAN_API_TOKEN_GUID"],
+    ), patch(
+        "application_sdk.clients.atlan_client.ATLAN_BASE_URL",
+        constants["ATLAN_BASE_URL"],
+    ), patch(
+        "application_sdk.clients.atlan_client.ATLAN_API_KEY", constants["ATLAN_API_KEY"]
+    ), patch(
+        "application_sdk.clients.atlan_client.ATLAN_CLIENT_ID",
+        constants["ATLAN_CLIENT_ID"],
+    ), patch(
+        "application_sdk.clients.atlan_client.ATLAN_CLIENT_SECRET",
+        constants["ATLAN_CLIENT_SECRET"],
+    ):
+        # Act / Assert
+        with pytest.raises(ClientError) as excinfo:
+            get_client(**params)
+        # Assert
+        assert msg in str(excinfo.value)
+
+
+@pytest.mark.parametrize(
+    "params,constants,expected_call,expected_args,expected_log,raises",
+    [
+        # Happy path: token auth via param
+        (
+            {"base_url": None, "api_key": None, "api_token_guid": "guid_param"},
+            {
+                "ATLAN_API_TOKEN_GUID": None,
+                "ATLAN_BASE_URL": None,
+                "ATLAN_API_KEY": None,
+                "ATLAN_CLIENT_ID": "cid",
+                "ATLAN_CLIENT_SECRET": "csecret",
+            },
+            "_get_client_from_token",
+            ("guid_param",),
+            None,
+            None,
+        ),
+        # Happy path: token auth via env
+        (
+            {"base_url": None, "api_key": None, "api_token_guid": None},
+            {
+                "ATLAN_API_TOKEN_GUID": "guid_const",
+                "ATLAN_BASE_URL": None,
+                "ATLAN_API_KEY": None,
+                "ATLAN_CLIENT_ID": "cid",
+                "ATLAN_CLIENT_SECRET": "csecret",
+            },
+            "_get_client_from_token",
+            ("guid_const",),
+            None,
+            None,
+        ),
+        # Happy path: token auth, base_url/api_key params present (should log warning)
+        (
+            {
+                "base_url": "https://atlan.com",
+                "api_key": "api_key_123",
+                "api_token_guid": "guid_param",
+            },
+            {
+                "ATLAN_API_TOKEN_GUID": None,
+                "ATLAN_BASE_URL": None,
+                "ATLAN_API_KEY": None,
+                "ATLAN_CLIENT_ID": "cid",
+                "ATLAN_CLIENT_SECRET": "csecret",
+            },
+            "_get_client_from_token",
+            ("guid_param",),
+            "warning",
+            None,
+        ),
+    ],
+    ids=[
+        "param_token_guid",
+        "env_token_guid",
+        "token_guid_with_base_url_api_key",
+    ],
+)
+def test_get_client_with_token_guid(
+    params, constants, expected_call, expected_args, expected_log, raises
+):
+    # Arrange
+
+    with patch(
+        "application_sdk.clients.atlan_client.ATLAN_API_TOKEN_GUID",
+        constants["ATLAN_API_TOKEN_GUID"],
+    ), patch(
+        "application_sdk.clients.atlan_client.ATLAN_BASE_URL",
+        constants["ATLAN_BASE_URL"],
+    ), patch(
+        "application_sdk.clients.atlan_client.ATLAN_API_KEY", constants["ATLAN_API_KEY"]
+    ), patch(
+        "application_sdk.clients.atlan_client.ATLAN_CLIENT_ID",
+        constants["ATLAN_CLIENT_ID"],
+    ), patch(
+        "application_sdk.clients.atlan_client.ATLAN_CLIENT_SECRET",
+        constants["ATLAN_CLIENT_SECRET"],
+    ), patch(
+        "application_sdk.clients.atlan_client._get_client_from_token"
+    ) as mock_get_client_from_token, patch(
+        "application_sdk.clients.atlan_client.AtlanClient"
+    ) as mock_atlan_client, patch(
+        "application_sdk.clients.atlan_client.logger"
+    ) as mock_logger:
+        mock_client_instance = MagicMock()
+        mock_get_client_from_token.return_value = mock_client_instance
+        mock_atlan_client.return_value = mock_client_instance
+
+        # Act & Assert
+        # Act
+        result = get_client(**params)
+        # Assert
+        mock_get_client_from_token.assert_called_once_with(*expected_args)
+        assert result == mock_client_instance
+        if expected_log == "warning":
+            mock_logger.warning.assert_called_once()
+
+
+@pytest.mark.parametrize(
+    "params,constants,expected_call,expected_args,expected_log,raises",
+    [
+        # Happy path: API key auth via params
+        (
+            {
+                "base_url": "https://atlan.com",
+                "api_key": "api_key_123",
+                "api_token_guid": None,
+            },
+            {
+                "ATLAN_API_TOKEN_GUID": None,
+                "ATLAN_BASE_URL": None,
+                "ATLAN_API_KEY": None,
+                "ATLAN_CLIENT_ID": "cid",
+                "ATLAN_CLIENT_SECRET": "csecret",
+            },
+            "AtlanClient",
+            {"base_url": "https://atlan.com", "api_key": "api_key_123"},
+            "info",
+            None,
+        ),
+        # Happy path: API key auth via constants
+        (
+            {"base_url": None, "api_key": None, "api_token_guid": None},
+            {
+                "ATLAN_API_TOKEN_GUID": None,
+                "ATLAN_BASE_URL": "https://const.atlan.com",
+                "ATLAN_API_KEY": "const_api_key",
+                "ATLAN_CLIENT_ID": None,
+                "ATLAN_CLIENT_SECRET": None,
+            },
+            "AtlanClient",
+            {"base_url": "https://const.atlan.com", "api_key": "const_api_key"},
+            "info",
+            None,
+        ),
+        # Edge case: param overrides env
+        (
+            {
+                "base_url": "https://param.atlan.com",
+                "api_key": "param_api_key",
+                "api_token_guid": None,
+            },
+            {
+                "ATLAN_API_TOKEN_GUID": None,
+                "ATLAN_BASE_URL": "https://const.atlan.com",
+                "ATLAN_API_KEY": "const_api_key",
+                "ATLAN_CLIENT_ID": None,
+                "ATLAN_CLIENT_SECRET": None,
+            },
+            "AtlanClient",
+            {"base_url": "https://param.atlan.com", "api_key": "param_api_key"},
+            "info",
+            None,
+        ),
+    ],
+    ids=[
+        "param_base_url_and_api_key",
+        "env_base_url_and_api_key",
+        "param_precedence_over_env",
+    ],
+)
+def test_get_client_with_api_key(
+    params, constants, expected_call, expected_args, expected_log, raises
+):
+    # Arrange
+    with patch(
+        "application_sdk.clients.atlan_client.ATLAN_API_TOKEN_GUID",
+        constants["ATLAN_API_TOKEN_GUID"],
+    ), patch(
+        "application_sdk.clients.atlan_client.ATLAN_BASE_URL",
+        constants["ATLAN_BASE_URL"],
+    ), patch(
+        "application_sdk.clients.atlan_client.ATLAN_API_KEY", constants["ATLAN_API_KEY"]
+    ), patch(
+        "application_sdk.clients.atlan_client.ATLAN_CLIENT_ID",
+        constants["ATLAN_CLIENT_ID"],
+    ), patch(
+        "application_sdk.clients.atlan_client.ATLAN_CLIENT_SECRET",
+        constants["ATLAN_CLIENT_SECRET"],
+    ), patch(
+        "application_sdk.clients.atlan_client._get_client_from_token"
+    ) as mock_get_client_from_token, patch(
+        "application_sdk.clients.atlan_client.AtlanClient"
+    ) as mock_atlan_client, patch(
+        "application_sdk.clients.atlan_client.logger"
+    ) as mock_logger:
+        mock_client_instance = MagicMock()
+        mock_get_client_from_token.return_value = mock_client_instance
+        mock_atlan_client.return_value = mock_client_instance
+
+        # Act & Assert
+        # Act
+        result = get_client(**params)
+        # Assert
+        mock_atlan_client.assert_called_once_with(**expected_args)
+        assert result == mock_client_instance
+        if expected_log == "info":
+            mock_logger.info.assert_called_once()

{atlan_application_sdk-0.1.1rc27 → atlan_application_sdk-0.1.1rc28}/uv.lock

@@ -1,5 +1,5 @@
 version = 1
-revision = 3
+revision = 2
 requires-python = ">=3.11"
 resolution-markers = [
     "python_full_version >= '3.13'",
@@ -131,7 +131,7 @@ wheels = [
 
 [[package]]
 name = "atlan-application-sdk"
-version = "0.1.1rc27"
+version = "0.1.1rc28"
 source = { editable = "." }
 dependencies = [
     { name = "aiohttp" },
@@ -224,7 +224,7 @@ requires-dist = [
     { name = "pandera", extras = ["io"], marker = "extra == 'tests'", specifier = ">=0.23.1" },
     { name = "psutil", specifier = ">=7.0.0" },
     { name = "pyarrow", marker = "extra == 'scale-data-generator'", specifier = ">=20.0.0" },
-    { name = "pyatlan", specifier = ">=7.1.1" },
+    { name = "pyatlan", specifier = ">=7.1.6" },
     { name = "pydantic", specifier = ">=2.10.6" },
     { name = "pyiceberg", marker = "extra == 'iceberg'", specifier = ">=0.8.1" },
     { name = "pytest-order", marker = "extra == 'tests'", specifier = ">=1.3.0" },
@@ -2360,7 +2360,7 @@ wheels = [
 
 [[package]]
 name = "pyatlan"
-version = "7.1.3"
+version = "7.1.6"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "jinja2" },
@@ -2374,9 +2374,9 @@ dependencies = [
     { name = "tenacity" },
     { name = "urllib3" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/5e/b2/e8dbbf06ac69f9da5810ee9cd59400dc25e31da0ee3c0a336509cb5f9d48/pyatlan-7.1.3.tar.gz", hash = "sha256:49f97473d47cd539b4fbfbc853111951af3ae1ca4e6b147c41eb1ed487e30a1d", size = 633105, upload-time = "2025-07-21T14:49:32.729Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/f6/a9/c34bff8f3f0b7b745ac399e1f5ff4470d4af23327531832b4e85b3a96bb2/pyatlan-7.1.6.tar.gz", hash = "sha256:1f0cce31f37206191fb909f63877a9db6f86d7c76085b7c1549a62fe69f86fef", size = 655078, upload-time = "2025-08-07T11:34:01.307Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/c6/65/777025b822ef4e5b5b1203bec4ccaa23a28f052c9f69fa23dd18462bcfe7/pyatlan-7.1.3-py3-none-any.whl", hash = "sha256:3e61986b7c12212c9893185cc0a235871b3f59b655fd94353f13e0aba92bed0a", size = 1152642, upload-time = "2025-07-21T14:49:31.007Z" },
+    { url = "https://files.pythonhosted.org/packages/3d/ca/72624efdfe9dcd9930304ea6827915ea4137005a84489c7f916d936f649b/pyatlan-7.1.6-py3-none-any.whl", hash = "sha256:0a82e283c36024b56788f195d3fa6138754626778eca14af6c22e9bbabd6f988", size = 1178356, upload-time = "2025-08-07T11:33:59.621Z" },
 ]
 
 [[package]]