frogml 1.2.41__py3-none-any.whl → 1.2.49__py3-none-any.whl

frogml_services_mock/mocks/cluster_v2_service.py

@@ -0,0 +1,146 @@
+from typing import Optional
+
+import grpc
+
+from frogml._proto.qwak.administration.cluster.v2.cluster_pb2 import (
+    ClusterState,
+)
+from frogml._proto.qwak.administration.cluster.v2.cluster_service_pb2 import (
+    CreateClusterCreationRequestResponse,
+    DeleteClusterResponse,
+    GetClusterResponse,
+    GetClusterStateResponse,
+    ListClustersResponse,
+    UpdateClusterConfigurationResponse,
+)
+from frogml._proto.qwak.administration.cluster.v2.cluster_service_pb2_grpc import (
+    ClusterServiceServicer,
+)
+
+
+class ClusterV2ServiceMock(ClusterServiceServicer):
+    def __init__(self):
+        super().__init__()
+        self.__create_cluster_response: Optional[
+            CreateClusterCreationRequestResponse
+        ] = None
+        self.__create_cluster_error: Optional[grpc.StatusCode] = None
+
+        self.__get_cluster_state_response: Optional[GetClusterStateResponse] = None
+        self.__get_cluster_state_error: Optional[grpc.StatusCode] = None
+
+        self.__get_cluster_response: Optional[GetClusterResponse] = None
+        self.__get_cluster_error: Optional[grpc.StatusCode] = None
+
+        self.__list_clusters_response: Optional[ListClustersResponse] = None
+        self.__list_clusters_error: Optional[grpc.StatusCode] = None
+
+        self.__update_cluster_response: Optional[UpdateClusterConfigurationResponse] = (
+            None
+        )
+        self.__update_cluster_error: Optional[grpc.StatusCode] = None
+
+        self.__delete_cluster_response: Optional[DeleteClusterResponse] = None
+        self.__delete_cluster_error: Optional[grpc.StatusCode] = None
+
+    def given_create_cluster(
+        self,
+        response: Optional[CreateClusterCreationRequestResponse] = None,
+        error_code: Optional[grpc.StatusCode] = None,
+    ):
+        self.__create_cluster_response = response
+        self.__create_cluster_error = error_code
+
+    def given_get_cluster_state(
+        self,
+        response: Optional[GetClusterStateResponse] = None,
+        error_code: Optional[grpc.StatusCode] = None,
+    ):
+        self.__get_cluster_state_response = response
+        self.__get_cluster_state_error = error_code
+
+    def given_get_cluster(
+        self,
+        response: Optional[GetClusterResponse] = None,
+        error_code: Optional[grpc.StatusCode] = None,
+    ):
+        self.__get_cluster_response = response
+        self.__get_cluster_error = error_code
+
+    def given_list_clusters(
+        self,
+        response: Optional[ListClustersResponse] = None,
+        error_code: Optional[grpc.StatusCode] = None,
+    ):
+        self.__list_clusters_response = response
+        self.__list_clusters_error = error_code
+
+    def given_update_cluster(
+        self,
+        response: Optional[UpdateClusterConfigurationResponse] = None,
+        error_code: Optional[grpc.StatusCode] = None,
+    ):
+        self.__update_cluster_response = response
+        self.__update_cluster_error = error_code
+
+    def given_delete_cluster(
+        self,
+        response: Optional[DeleteClusterResponse] = None,
+        error_code: Optional[grpc.StatusCode] = None,
+    ):
+        self.__delete_cluster_response = response
+        self.__delete_cluster_error = error_code
+
+    def CreateClusterCreationRequest(self, request, context):
+        if self.__create_cluster_error:
+            context.set_code(self.__create_cluster_error)
+            context.set_details("Failed to create cluster")
+            return CreateClusterCreationRequestResponse()
+        if self.__create_cluster_response:
+            return self.__create_cluster_response
+        return CreateClusterCreationRequestResponse(request_id="mock-request-id")
+
+    def GetClusterState(self, request, context):
+        if self.__get_cluster_state_error:
+            context.set_code(self.__get_cluster_state_error)
+            context.set_details("Failed to get cluster state")
+            return GetClusterStateResponse()
+        if self.__get_cluster_state_response:
+            return self.__get_cluster_state_response
+        return GetClusterStateResponse(cluster_state=ClusterState())
+
+    def GetCluster(self, request, context):
+        if self.__get_cluster_error:
+            context.set_code(self.__get_cluster_error)
+            context.set_details("Failed to get cluster")
+            return GetClusterResponse()
+        if self.__get_cluster_response:
+            return self.__get_cluster_response
+        return GetClusterResponse(cluster_state=ClusterState())
+
+    def ListClusters(self, request, context):
+        if self.__list_clusters_error:
+            context.set_code(self.__list_clusters_error)
+            context.set_details("Failed to list clusters")
+            return ListClustersResponse()
+        if self.__list_clusters_response:
+            return self.__list_clusters_response
+        return ListClustersResponse()
+
+    def UpdateClusterConfiguration(self, request, context):
+        if self.__update_cluster_error:
+            context.set_code(self.__update_cluster_error)
+            context.set_details("Failed to update cluster")
+            return UpdateClusterConfigurationResponse()
+        if self.__update_cluster_response:
+            return self.__update_cluster_response
+        return UpdateClusterConfigurationResponse(request_id="mock-request-id")
+
+    def DeleteCluster(self, request, context):
+        if self.__delete_cluster_error:
+            context.set_code(self.__delete_cluster_error)
+            context.set_details("Failed to delete cluster")
+            return DeleteClusterResponse()
+        if self.__delete_cluster_response:
+            return self.__delete_cluster_response
+        return DeleteClusterResponse(request_id="mock-request-id")

frogml_services_mock/mocks/environment_v0_service.py

@@ -0,0 +1,175 @@
+from typing import Optional
+
+import grpc
+
+from frogml._proto.qwak.administration.v0.environments.environment_pb2 import (
+    QwakEnvironment,
+)
+from frogml._proto.qwak.administration.v0.environments.environment_service_pb2 import (
+    CreateEnvironmentResponse,
+    GetEnvironmentApplicationUserCredentialsResponse,
+    GetEnvironmentResponse,
+    ListEnvironmentsResponse,
+    SetEnvironmentApplicationUserCredentialsResponse,
+    UpdateEnvironmentConfigurationResponse,
+    UpdateEnvironmentPersonalizationResponse,
+)
+from frogml._proto.qwak.administration.v0.environments.environment_service_pb2_grpc import (
+    EnvironmentServiceServicer,
+)
+
+
+class EnvironmentV0ServiceMock(EnvironmentServiceServicer):
+    def __init__(self):
+        super().__init__()
+        self.__create_environment_response: Optional[CreateEnvironmentResponse] = None
+        self.__create_environment_error: Optional[grpc.StatusCode] = None
+
+        self.__get_environment_response: Optional[GetEnvironmentResponse] = None
+        self.__get_environment_error: Optional[grpc.StatusCode] = None
+
+        self.__update_environment_configuration_response: Optional[
+            UpdateEnvironmentConfigurationResponse
+        ] = None
+        self.__update_environment_configuration_error: Optional[grpc.StatusCode] = None
+
+        self.__update_environment_personalization_response: Optional[
+            UpdateEnvironmentPersonalizationResponse
+        ] = None
+        self.__update_environment_personalization_error: Optional[grpc.StatusCode] = (
+            None
+        )
+
+        self.__list_environments_response: Optional[ListEnvironmentsResponse] = None
+        self.__list_environments_error: Optional[grpc.StatusCode] = None
+
+        self.__get_credentials_response: Optional[
+            GetEnvironmentApplicationUserCredentialsResponse
+        ] = None
+        self.__get_credentials_error: Optional[grpc.StatusCode] = None
+
+        self.__set_credentials_response: Optional[
+            SetEnvironmentApplicationUserCredentialsResponse
+        ] = None
+        self.__set_credentials_error: Optional[grpc.StatusCode] = None
+
+    def given_create_environment(
+        self,
+        response: Optional[CreateEnvironmentResponse] = None,
+        error_code: Optional[grpc.StatusCode] = None,
+    ):
+        self.__create_environment_response = response
+        self.__create_environment_error = error_code
+
+    def given_get_environment(
+        self,
+        response: Optional[GetEnvironmentResponse] = None,
+        error_code: Optional[grpc.StatusCode] = None,
+    ):
+        self.__get_environment_response = response
+        self.__get_environment_error = error_code
+
+    def given_update_environment_configuration(
+        self,
+        response: Optional[UpdateEnvironmentConfigurationResponse] = None,
+        error_code: Optional[grpc.StatusCode] = None,
+    ):
+        self.__update_environment_configuration_response = response
+        self.__update_environment_configuration_error = error_code
+
+    def given_update_environment_personalization(
+        self,
+        response: Optional[UpdateEnvironmentPersonalizationResponse] = None,
+        error_code: Optional[grpc.StatusCode] = None,
+    ):
+        self.__update_environment_personalization_response = response
+        self.__update_environment_personalization_error = error_code
+
+    def given_list_environments(
+        self,
+        response: Optional[ListEnvironmentsResponse] = None,
+        error_code: Optional[grpc.StatusCode] = None,
+    ):
+        self.__list_environments_response = response
+        self.__list_environments_error = error_code
+
+    def given_get_credentials(
+        self,
+        response: Optional[GetEnvironmentApplicationUserCredentialsResponse] = None,
+        error_code: Optional[grpc.StatusCode] = None,
+    ):
+        self.__get_credentials_response = response
+        self.__get_credentials_error = error_code
+
+    def given_set_credentials(
+        self,
+        response: Optional[SetEnvironmentApplicationUserCredentialsResponse] = None,
+        error_code: Optional[grpc.StatusCode] = None,
+    ):
+        self.__set_credentials_response = response
+        self.__set_credentials_error = error_code
+
+    def CreateEnvironment(self, request, context):
+        if self.__create_environment_error:
+            context.set_code(self.__create_environment_error)
+            context.set_details("Failed to create environment")
+            return CreateEnvironmentResponse()
+        if self.__create_environment_response:
+            return self.__create_environment_response
+        return CreateEnvironmentResponse(environment=QwakEnvironment(id="mock-env-id"))
+
+    def GetEnvironment(self, request, context):
+        if self.__get_environment_error:
+            context.set_code(self.__get_environment_error)
+            context.set_details("Failed to get environment")
+            return GetEnvironmentResponse()
+        if self.__get_environment_response:
+            return self.__get_environment_response
+        return GetEnvironmentResponse(
+            environment=QwakEnvironment(id=request.environment_id)
+        )
+
+    def UpdateEnvironmentConfiguration(self, request, context):
+        if self.__update_environment_configuration_error:
+            context.set_code(self.__update_environment_configuration_error)
+            context.set_details("Failed to update environment configuration")
+            return UpdateEnvironmentConfigurationResponse()
+        if self.__update_environment_configuration_response:
+            return self.__update_environment_configuration_response
+        return UpdateEnvironmentConfigurationResponse()
+
+    def UpdateEnvironmentPersonalization(self, request, context):
+        if self.__update_environment_personalization_error:
+            context.set_code(self.__update_environment_personalization_error)
+            context.set_details("Failed to update environment personalization")
+            return UpdateEnvironmentPersonalizationResponse()
+        if self.__update_environment_personalization_response:
+            return self.__update_environment_personalization_response
+        return UpdateEnvironmentPersonalizationResponse()
+
+    def ListEnvironments(self, request, context):
+        if self.__list_environments_error:
+            context.set_code(self.__list_environments_error)
+            context.set_details("Failed to list environments")
+            return ListEnvironmentsResponse()
+        if self.__list_environments_response:
+            return self.__list_environments_response
+        return ListEnvironmentsResponse()
+
+    def GetEnvironmentApplicationUserCredentials(self, request, context):
+        if self.__get_credentials_error:
+            context.set_code(self.__get_credentials_error)
+            context.set_details("Failed to get credentials")
+            return GetEnvironmentApplicationUserCredentialsResponse()
+        if self.__get_credentials_response:
+            return self.__get_credentials_response
+        return GetEnvironmentApplicationUserCredentialsResponse()
+
+    def SetEnvironmentApplicationUserCredentials(self, request, context):
+        if self.__set_credentials_error:
+            context.set_code(self.__set_credentials_error)
+            context.set_details("Failed to set credentials")
+            return SetEnvironmentApplicationUserCredentialsResponse()
+        if self.__set_credentials_response:
+            return self.__set_credentials_response
+        return SetEnvironmentApplicationUserCredentialsResponse()

frogml_services_mock/mocks/environment_v1_service.py

@@ -0,0 +1,37 @@
+from typing import Optional
+
+import grpc
+
+from frogml._proto.qwak.administration.v0.environments.environment_pb2 import (
+    QwakEnvironment,
+)
+from frogml._proto.qwak.administration.v1.environments.environment_service_pb2 import (
+    CreateEnvironmentResponse,
+)
+from frogml._proto.qwak.administration.v1.environments.environment_service_pb2_grpc import (
+    EnvironmentServiceServicer,
+)
+
+
+class EnvironmentV1ServiceMock(EnvironmentServiceServicer):
+    def __init__(self):
+        super().__init__()
+        self.__create_environment_response: Optional[CreateEnvironmentResponse] = None
+        self.__create_environment_error: Optional[grpc.StatusCode] = None
+
+    def given_create_environment(
+        self,
+        response: Optional[CreateEnvironmentResponse] = None,
+        error_code: Optional[grpc.StatusCode] = None,
+    ):
+        self.__create_environment_response = response
+        self.__create_environment_error = error_code
+
+    def CreateEnvironment(self, request, context):
+        if self.__create_environment_error:
+            context.set_code(self.__create_environment_error)
+            context.set_details("Failed to create environment")
+            return CreateEnvironmentResponse()
+        if self.__create_environment_response:
+            return self.__create_environment_response
+        return CreateEnvironmentResponse(environment=QwakEnvironment(id="mock-env-id"))

frogml_services_mock/mocks/frogml_mocks.py

@@ -23,11 +23,14 @@ from frogml_services_mock.mocks.build_orchestrator_build_settings_api import (
 from frogml_services_mock.mocks.build_orchestrator_service_api import (
     BuildOrchestratorServiceApiMock,
 )
+from frogml_services_mock.mocks.cluster_v2_service import ClusterV2ServiceMock
 from frogml_services_mock.mocks.data_versioning_service import DataVersioningServiceMock
 from frogml_services_mock.mocks.deployment_management_service import (
     DeploymentManagementServiceMock,
 )
 from frogml_services_mock.mocks.ecosystem_service_api import EcoSystemServiceMock
+from frogml_services_mock.mocks.environment_v0_service import EnvironmentV0ServiceMock
+from frogml_services_mock.mocks.environment_v1_service import EnvironmentV1ServiceMock
 from frogml_services_mock.mocks.execution_management_service import (
     ExecutionManagementServiceMock,
 )
@@ -135,3 +138,6 @@ class FrogmlMocks:
     location_discovery_service: LocationDiscoveryServiceApiMock
     jfrog_tenant_info_service: JFrogTenantInfoServiceMock
     model_deployment_manager_mock: ModelDeploymentManagerMock
+    cluster_v2_service_mock: ClusterV2ServiceMock
+    environment_v0_service_mock: EnvironmentV0ServiceMock
+    environment_v1_service_mock: EnvironmentV1ServiceMock

frogml_services_mock/services_mock.py

@@ -12,6 +12,15 @@ from frogml._proto.jfml.model_version.v1.model_version_manager_service_pb2_grpc
 from frogml._proto.qwak.administration.v0.authentication.authentication_service_pb2_grpc import (
     add_AuthenticationServiceServicer_to_server,
 )
+from frogml._proto.qwak.administration.cluster.v2.cluster_service_pb2_grpc import (
+    add_ClusterServiceServicer_to_server,
+)
+from frogml._proto.qwak.administration.v0.environments.environment_service_pb2_grpc import (
+    add_EnvironmentServiceServicer_to_server as add_EnvironmentV0ServiceServicer_to_server,
+)
+from frogml._proto.qwak.administration.v1.environments.environment_service_pb2_grpc import (
+    add_EnvironmentServiceServicer_to_server as add_EnvironmentV1ServiceServicer_to_server,
+)
 from frogml._proto.qwak.admiral.secret.v0.system_secret_service_pb2_grpc import (
     add_SystemSecretServiceServicer_to_server,
 )
@@ -144,11 +153,14 @@ from frogml_services_mock.mocks.build_orchestrator_build_settings_api import (
 from frogml_services_mock.mocks.build_orchestrator_service_api import (
     BuildOrchestratorServiceApiMock,
 )
+from frogml_services_mock.mocks.cluster_v2_service import ClusterV2ServiceMock
 from frogml_services_mock.mocks.data_versioning_service import DataVersioningServiceMock
 from frogml_services_mock.mocks.deployment_management_service import (
     DeploymentManagementServiceMock,
 )
 from frogml_services_mock.mocks.ecosystem_service_api import EcoSystemServiceMock
+from frogml_services_mock.mocks.environment_v0_service import EnvironmentV0ServiceMock
+from frogml_services_mock.mocks.environment_v1_service import EnvironmentV1ServiceMock
 from frogml_services_mock.mocks.execution_management_service import (
     ExecutionManagementServiceMock,
 )
@@ -262,7 +274,10 @@ def frogml_container():
     )
     from frogml.core.clients.administration import (
         authentication,
+        cluster_v2,
         eco_system,
+        environment,
+        environment_v1,
         self_service,
     )
     from frogml.core.clients.integration_management import integration_manager_client
@@ -278,8 +293,11 @@ def frogml_container():
             analytics,
             batch_job_management,
             build_orchestrator,
+            cluster_v2,
             data_versioning,
             deployment,
+            environment,
+            environment_v1,
             instance_template,
             feature_store,
             file_versioning,
@@ -517,6 +535,21 @@ def attach_servicers(free_port, server):
                 ModelDeploymentManagerMock,
                 add_ModelDeploymentServiceServicer_to_server,
             ),
+            (
+                "cluster_v2_service_mock",
+                ClusterV2ServiceMock,
+                add_ClusterServiceServicer_to_server,
+            ),
+            (
+                "environment_v0_service_mock",
+                EnvironmentV0ServiceMock,
+                add_EnvironmentV0ServiceServicer_to_server,
+            ),
+            (
+                "environment_v1_service_mock",
+                EnvironmentV1ServiceMock,
+                add_EnvironmentV1ServiceServicer_to_server,
+            ),
             ("port", free_port, None),
         ],
     )