PYCBC-1736: Migrate Analytics Index management away from Wrapper decorators

Changes
-------
* Add analytics mgmt types, request builder
* Add analytics mgmt impl for all 3 APIs
* Update APIs analytics manager classes to use impl and remove wrapper
  decorators
* Remove pytest.skip() from analytics mgmt tests

Change-Id: Id971b77eae63f5fa78d90a5f8b086175f27dc0d9
Reviewed-on: https://review.couchbase.org/c/couchbase-python-client/+/239814
Reviewed-by: Dimitris Christodoulou <dimitris.christodoulou@couchbase.com>
Tested-by: Build Bot <build@couchbase.com>

Author: thejcfactor

acouchbase/management/analytics.py

@@ -0,0 +1,148 @@
+#  Copyright 2016-2023. Couchbase, Inc.
+#  All Rights Reserved.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License")
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.
+
+from __future__ import annotations
+
+from asyncio import AbstractEventLoop
+from typing import (TYPE_CHECKING,
+                    Dict,
+                    Iterable)
+
+from couchbase.management.logic.analytics_mgmt_req_builder import AnalyticsMgmtRequestBuilder
+from couchbase.management.logic.analytics_mgmt_types import (AnalyticsDataset,
+                                                             AnalyticsIndex,
+                                                             AnalyticsLink,
+                                                             AzureBlobExternalAnalyticsLink,
+                                                             CouchbaseRemoteAnalyticsLink,
+                                                             S3ExternalAnalyticsLink)
+
+if TYPE_CHECKING:
+    from acouchbase.logic.client_adapter import AsyncClientAdapter
+    from couchbase.management.logic.analytics_mgmt_types import (ConnectLinkRequest,
+                                                                 CreateDatasetRequest,
+                                                                 CreateDataverseRequest,
+                                                                 CreateIndexRequest,
+                                                                 CreateLinkRequest,
+                                                                 DisconnectLinkRequest,
+                                                                 DropDatasetRequest,
+                                                                 DropDataverseRequest,
+                                                                 DropIndexRequest,
+                                                                 DropLinkRequest,
+                                                                 GetAllDatasetsRequest,
+                                                                 GetAllIndexesRequest,
+                                                                 GetLinksRequest,
+                                                                 GetPendingMutationsRequest,
+                                                                 ReplaceLinkRequest)
+
+
+class AsyncAnalyticsMgmtImpl:
+    def __init__(self, client_adapter: AsyncClientAdapter) -> None:
+        self._client_adapter = client_adapter
+        self._request_builder = AnalyticsMgmtRequestBuilder()
+
+    @property
+    def loop(self) -> AbstractEventLoop:
+        """**INTERNAL**"""
+        return self._client_adapter.loop
+
+    @property
+    def request_builder(self) -> AnalyticsMgmtRequestBuilder:
+        """**INTERNAL**"""
+        return self._request_builder
+
+    async def connect_link(self, req: ConnectLinkRequest) -> None:
+        """**INTERNAL**"""
+        await self._client_adapter.execute_mgmt_request(req)
+
+    async def create_dataset(self, req: CreateDatasetRequest) -> None:
+        """**INTERNAL**"""
+        await self._client_adapter.execute_mgmt_request(req)
+
+    async def create_dataverse(self, req: CreateDataverseRequest) -> None:
+        """**INTERNAL**"""
+        await self._client_adapter.execute_mgmt_request(req)
+
+    async def create_index(self, req: CreateIndexRequest) -> None:
+        """**INTERNAL**"""
+        await self._client_adapter.execute_mgmt_request(req)
+
+    async def create_link(self, req: CreateLinkRequest) -> None:
+        """**INTERNAL**"""
+        await self._client_adapter.execute_mgmt_request(req)
+
+    async def disconnect_link(self, req: DisconnectLinkRequest) -> None:
+        """**INTERNAL**"""
+        await self._client_adapter.execute_mgmt_request(req)
+
+    async def drop_dataset(self, req: DropDatasetRequest) -> None:
+        """**INTERNAL**"""
+        await self._client_adapter.execute_mgmt_request(req)
+
+    async def drop_dataverse(self, req: DropDataverseRequest) -> None:
+        """**INTERNAL**"""
+        await self._client_adapter.execute_mgmt_request(req)
+
+    async def drop_index(self, req: DropIndexRequest) -> None:
+        """**INTERNAL**"""
+        await self._client_adapter.execute_mgmt_request(req)
+
+    async def drop_link(self, req: DropLinkRequest) -> None:
+        """**INTERNAL**"""
+        await self._client_adapter.execute_mgmt_request(req)
+
+    async def get_all_datasets(self, req: GetAllDatasetsRequest) -> Iterable[AnalyticsDataset]:
+        """**INTERNAL**"""
+        datasets = []
+        ret = await self._client_adapter.execute_mgmt_request(req)
+        raw_datasets = ret.raw_result.get('datasets', None)
+        if raw_datasets:
+            datasets = [AnalyticsDataset(**ds) for ds in raw_datasets]
+        return datasets
+
+    async def get_all_indexes(self, req: GetAllIndexesRequest) -> Iterable[AnalyticsIndex]:
+        """**INTERNAL**"""
+        indexes = []
+        ret = await self._client_adapter.execute_mgmt_request(req)
+        raw_indexes = ret.raw_result.get('indexes', None)
+        if raw_indexes:
+            indexes = [AnalyticsIndex(**ds) for ds in raw_indexes]
+
+        return indexes
+
+    async def get_links(self, req: GetLinksRequest) -> Iterable[AnalyticsLink]:
+        """**INTERNAL**"""
+        analytics_links = []
+        ret = await self._client_adapter.execute_mgmt_request(req)
+        cb_links = ret.raw_result.get('couchbase_links', None)
+        if cb_links and len(cb_links) > 0:
+            analytics_links.extend(map(lambda l: CouchbaseRemoteAnalyticsLink.link_from_server_json(l), cb_links))
+        s3_links = ret.raw_result.get('s3_links', None)
+        if s3_links and len(s3_links) > 0:
+            analytics_links.extend(map(lambda l: S3ExternalAnalyticsLink.link_from_server_json(l), s3_links))
+        azure_blob_links = ret.raw_result.get('azure_blob_links', None)
+        if azure_blob_links and len(azure_blob_links) > 0:
+            analytics_links.extend(
+                map(lambda l: AzureBlobExternalAnalyticsLink.link_from_server_json(l), azure_blob_links))
+
+        return analytics_links
+
+    async def get_pending_mutations(self, req: GetPendingMutationsRequest) -> Dict[str, int]:
+        """**INTERNAL**"""
+        ret = await self._client_adapter.execute_mgmt_request(req)
+        return ret.raw_result.get('stats', {})
+
+    async def replace_link(self, req: ReplaceLinkRequest) -> None:
+        """**INTERNAL**"""
+        await self._client_adapter.execute_mgmt_request(req)

acouchbase/management/logic/analytics_mgmt_impl.py

@@ -48,7 +48,6 @@
 from ._test_utils import TestEnvironment
 
 
-@pytest.mark.skip(reason='Skip until PYCBC-1736')
 class AnalyticsManagementTests:
     DATASET_NAME = 'test-dataset'
 
@@ -298,7 +297,6 @@ async def test_v7_dataverse_name_parsing(self, cb_env):
         await cb_env.am.drop_dataverse("test/test_dataverse")
 
 
-@pytest.mark.skip(reason='Skip until PYCBC-1736')
 class AnalyticsManagementLinksTests:
     DATASET_NAME = 'test-dataset'