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

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),
         ],
     )