[deps] Fixed deps and tests

Author: Aryamanz29

pyatlan_v9/client/aio/asset.py

@@ -57,15 +57,15 @@
 )
 from pyatlan.client.constants import BULK_UPDATE, DELETE_ENTITIES_BY_GUIDS
 from pyatlan.errors import ErrorCode, NotFoundError, PermissionError
+from pyatlan.model.aio import AsyncIndexSearchResults, AsyncLineageListResults
+from pyatlan.model.fields.atlan_fields import AtlanField
+from pyatlan.utils import unflatten_custom_metadata_for_entity
 from pyatlan_v9.client.asset import (
     _handle_v9_glossary_anchor,
     _is_glossary_category,
     _matches_asset_type,
     _process_find_response_v9,
 )
-from pyatlan.model.aio import AsyncIndexSearchResults, AsyncLineageListResults
-from pyatlan.model.fields.atlan_fields import AtlanField
-from pyatlan.utils import unflatten_custom_metadata_for_entity
 from pyatlan_v9.model.aggregation import Aggregations
 from pyatlan_v9.model.aio.core import AsyncAtlanRequest
 from pyatlan_v9.model.assets import (

pyatlan_v9/client/asset.py

@@ -37,14 +37,14 @@
 # ---------------------------------------------------------------------------
 from pyatlan.client.asset import (  # noqa: F401
     AssetIdentity,
-    Batch as _LegacyBatch,
     CategoryHierarchy,
     CustomMetadataHandling,
     FailedBatch,
     IndexSearchResults,
     LineageListResults,
     SearchResults,
 )
+from pyatlan.client.asset import Batch as _LegacyBatch
 from pyatlan.client.common import (
     ApiCaller,
     DeleteByGuid,

pyatlan_v9/model/assets/schema.py

@@ -486,7 +486,10 @@ def _schema__from_nested_bytes(data: bytes, serde: Serde) -> Schema:
 # ---------------------------------------------------------------------------
 # Schema field descriptors (inherited from SQL in legacy, set explicitly here)
 # ---------------------------------------------------------------------------
-from pyatlan.model.fields.atlan_fields import KeywordField, KeywordTextField  # noqa: E402
+from pyatlan.model.fields.atlan_fields import (  # noqa: E402
+    KeywordField,
+    KeywordTextField,
+)
 
 Schema.DATABASE_NAME = KeywordTextField(
     "databaseName", "databaseName.keyword", "databaseName"

pyproject.toml

@@ -37,7 +37,7 @@ dependencies = [
     "PyYAML~=6.0.3",
     "httpx~=0.28.1",
     "httpx-retries~=0.4.5",
-    "authlib~=1.6.6",
+    "authlib~=1.6.9",
     "msgspec~=0.20.0",
 ]
 

requirements.txt

@@ -5,7 +5,7 @@ annotated-types==0.7.0 ; python_full_version < '3.14' and platform_python_implem
     # via pydantic
 anyio==4.10.0 ; python_full_version < '3.14' and platform_python_implementation == 'CPython'
     # via httpx
-authlib==1.6.6 ; python_full_version < '3.14' and platform_python_implementation == 'CPython'
+authlib==1.6.9 ; python_full_version < '3.14' and platform_python_implementation == 'CPython'
     # via pyatlan
 backports-asyncio-runner==1.2.0 ; python_full_version < '3.11' and platform_python_implementation == 'CPython'
     # via pytest-asyncio
@@ -100,6 +100,8 @@ more-itertools==10.7.0 ; python_full_version < '3.14' and platform_machine != 'p
     # via
     #   jaraco-classes
     #   jaraco-functools
+msgspec==0.20.0 ; python_full_version < '3.14' and platform_python_implementation == 'CPython'
+    # via pyatlan
 multidict==6.6.4 ; python_full_version < '3.14' and platform_python_implementation == 'CPython'
     # via yarl
 mypy==1.18.2 ; python_full_version < '3.14' and platform_python_implementation == 'CPython'

tests_v9/integration/aio/test_asset_batch.py

@@ -5,8 +5,8 @@
 import pytest
 import pytest_asyncio
 
-from pyatlan_v9.client.aio.batch import AsyncBatch
 from pyatlan_v9.client.aio.atlan import AsyncAtlanClient
+from pyatlan_v9.client.aio.batch import AsyncBatch
 from pyatlan_v9.model.assets import (
     Asset,
     Connection,

tests_v9/unit/aio/test_client.py

@@ -10,13 +10,13 @@
 import pytest
 from httpx import Headers
 
-from pyatlan_v9.client.aio.batch import AsyncBatch
 from pyatlan.client.common import Search
 from pyatlan.client.common.asset import LOGGER as SHARED_LOGGER
 from pyatlan.model.aio.asset import AsyncIndexSearchResults
 from pyatlan.utils import get_python_version
 from pyatlan_v9.client.aio.asset import V9AsyncAssetClient as AsyncAssetClient
 from pyatlan_v9.client.aio.atlan import AsyncAtlanClient
+from pyatlan_v9.client.aio.batch import AsyncBatch
 from pyatlan_v9.client.aio.group import V9AsyncGroupClient as AsyncGroupClient
 from pyatlan_v9.client.aio.search_log import (
     V9AsyncSearchLogClient as AsyncSearchLogClient,

tests_v9/unit/constants.py

@@ -2,6 +2,14 @@
 # Copyright 2024 Atlan Pte. Ltd.
 
 # Non-validation constants from the legacy test suite.
+from pyatlan_v9.model.assets import AtlasGlossary
+from pyatlan_v9.model.enums import AtlanWorkflowPhase
+from pyatlan_v9.model.workflow import (
+    ScheduleQueriesSearchRequest,
+    WorkflowMetadata,
+    WorkflowResponse,
+    WorkflowSpec,
+)
 from tests.unit.constants import (  # noqa: F401, F403
     APPLICABLE_AI_ASSET_TYPES,
     APPLICABLE_ASSET_TYPES,
@@ -18,15 +26,6 @@
 )
 from tests.unit.model.constants import *  # noqa: F401, F403
 
-from pyatlan_v9.model.assets import AtlasGlossary
-from pyatlan_v9.model.enums import AtlanWorkflowPhase
-from pyatlan_v9.model.workflow import (
-    ScheduleQueriesSearchRequest,
-    WorkflowMetadata,
-    WorkflowResponse,
-    WorkflowSpec,
-)
-
 TEST_ASSET_CLIENT_METHODS = {
     "find_personas_by_name": [
         ([[123], ["attributes"]], "name\n  str type expected"),

tests_v9/unit/test_client.py

@@ -80,21 +80,11 @@
     TEST_ROLE_CLIENT_METHODS,
     TEST_SL_CLIENT_METHODS,
 )
-from tests_v9.unit.constants import (
-    TEST_ASSET_CLIENT_METHODS as _V9_ASSET_METHODS,
-)
-from tests_v9.unit.constants import (
-    TEST_GROUP_CLIENT_METHODS as _V9_GROUP_METHODS,
-)
-from tests_v9.unit.constants import (
-    TEST_TOKEN_CLIENT_METHODS as _V9_TOKEN_METHODS,
-)
-from tests_v9.unit.constants import (
-    TEST_TYPEDEF_CLIENT_METHODS as _V9_TYPEDEF_METHODS,
-)
-from tests_v9.unit.constants import (
-    TEST_USER_CLIENT_METHODS as _V9_USER_METHODS,
-)
+from tests_v9.unit.constants import TEST_ASSET_CLIENT_METHODS as _V9_ASSET_METHODS
+from tests_v9.unit.constants import TEST_GROUP_CLIENT_METHODS as _V9_GROUP_METHODS
+from tests_v9.unit.constants import TEST_TOKEN_CLIENT_METHODS as _V9_TOKEN_METHODS
+from tests_v9.unit.constants import TEST_TYPEDEF_CLIENT_METHODS as _V9_TYPEDEF_METHODS
+from tests_v9.unit.constants import TEST_USER_CLIENT_METHODS as _V9_USER_METHODS
 
 # v9 uses pyatlan search DSL internally; " " name validation raises WithName not FindXByName
 _V9_WHITESPACE_NAME_MSG = "1 validation error for WithName\nvalue\n  ensure this value has at least 1 characters"