feat: connect service account

cognitedata · Dec 12, 2024 · 6a9d95c · 6a9d95c
1 parent 6f76aa4
commit 6a9d95c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/cognite/client/_api/iam.py b/cognite/client/_api/iam.py
@@ -114,6 +114,7 @@ def __init__(self, config: ClientConfig, api_version: str | None, cognite_client
         self.user_profiles = UserProfilesAPI(config, api_version, cognite_client)
         # TokenAPI only uses base_url, so we pass `api_version=None`:
         self.token = TokenAPI(config, api_version=None, cognite_client=cognite_client)
+        self.service_accounts = ServiceAccountsAPI(config, api_version, cognite_client)
 
     @staticmethod
     def compare_capabilities(

diff --git a/cognite/client/testing.py b/cognite/client/testing.py
@@ -35,7 +35,15 @@
 from cognite.client._api.hosted_extractors.jobs import JobsAPI
 from cognite.client._api.hosted_extractors.mappings import MappingsAPI
 from cognite.client._api.hosted_extractors.sources import SourcesAPI
-from cognite.client._api.iam import IAMAPI, GroupsAPI, SecurityCategoriesAPI, SessionsAPI, TokenAPI
+from cognite.client._api.iam import (
+    IAMAPI,
+    GroupsAPI,
+    SecurityCategoriesAPI,
+    ServiceAccountsAPI,
+    ServiceAccountSecretsAPI,
+    SessionsAPI,
+    TokenAPI,
+)
 from cognite.client._api.labels import LabelsAPI
 from cognite.client._api.postgres_gateway import PostgresGatewaysAPI
 from cognite.client._api.postgres_gateway.tables import TablesAPI as PostgresTablesAPI
@@ -129,6 +137,8 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
         self.iam = MagicMock(spec=IAMAPI)
         self.iam.groups = MagicMock(spec_set=GroupsAPI)
         self.iam.security_categories = MagicMock(spec_set=SecurityCategoriesAPI)
+        self.iam.service_accounts = MagicMock(spec=ServiceAccountsAPI)
+        self.iam.service_accounts.secrets = MagicMock(spec_set=ServiceAccountSecretsAPI)
         self.iam.sessions = MagicMock(spec_set=SessionsAPI)
         self.iam.user_profiles = MagicMock(spec_set=UserProfilesAPI)
         self.iam.token = MagicMock(spec_set=TokenAPI)