qwak-core 0.6.5__py3-none-any.whl → 0.6.7__py3-none-any.whl

{qwak_core-0.6.5.dist-info → qwak_core-0.6.7.dist-info}/RECORD

@@ -34,6 +34,9 @@ _qwak_proto/qwak/administration/authenticated_user/v1/details_pb2_grpc.py,sha256
 _qwak_proto/qwak/administration/cluster/v2/cluster_pb2.py,sha256=UejDfmq-HJY76cJr6Dqp8rr2sioIXOBI4-MrINFM-iU,7635
 _qwak_proto/qwak/administration/cluster/v2/cluster_pb2.pyi,sha256=uzek313Ky25mMDF9_OvMJNKT99duipycHnSXCMyRG9A,17386
 _qwak_proto/qwak/administration/cluster/v2/cluster_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
+_qwak_proto/qwak/administration/cluster/v2/cluster_service_pb2.py,sha256=ZMZ45Ew_deUUyv8xbqHhvMmeWVfyyvmSovg7s5F8dpw,5315
+_qwak_proto/qwak/administration/cluster/v2/cluster_service_pb2.pyi,sha256=_QEpc-9_00aekgCOuO9saqhTCp7SbtAKVHLWWf8Qhf4,7315
+_qwak_proto/qwak/administration/cluster/v2/cluster_service_pb2_grpc.py,sha256=WU6Q9Cfj8PrsPxXg45cnDnf3xQIyQpGu9hvYjqIvrrA,12942
 _qwak_proto/qwak/administration/runtime_configuration/v0/container_registry_config_pb2.py,sha256=7W970swdY_S44ffVrfNvYibBMI4TX4UnPY7lYTyY0aw,2780
 _qwak_proto/qwak/administration/runtime_configuration/v0/container_registry_config_pb2.pyi,sha256=rK7MO6WAjauQz8ShDsqhqYkc69cI_AEqUE42j9vbT4U,2640
 _qwak_proto/qwak/administration/runtime_configuration/v0/container_registry_config_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
@@ -115,6 +118,12 @@ _qwak_proto/qwak/administration/v0/environments/personalization_pb2_grpc.py,sha2
 _qwak_proto/qwak/administration/v0/users/user_pb2.py,sha256=ZXt8_c_uIrDSBQfAmVqrNctab40U0E_XJ_gLN8RLeYo,4676
 _qwak_proto/qwak/administration/v0/users/user_pb2.pyi,sha256=GFbJz424r1aTF1kC50ZDIvYTfgxO02C7OHu2SZyL-nc,10595
 _qwak_proto/qwak/administration/v0/users/user_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
+_qwak_proto/qwak/administration/v1/environments/environment_pb2.py,sha256=6C22w-7Ik12XDwS1Kru8FqTDcu19wcDmImvtfOKjSDc,2403
+_qwak_proto/qwak/administration/v1/environments/environment_pb2.pyi,sha256=7L7jRonQENbdV1daHkyU8AGglaDdGyq5kQqPAPZG38A,2063
+_qwak_proto/qwak/administration/v1/environments/environment_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
+_qwak_proto/qwak/administration/v1/environments/environment_service_pb2.py,sha256=AcnKQXJkug4-wTaI1_bnHvnYt_JaH8kvYiXRmrwCPCM,2792
+_qwak_proto/qwak/administration/v1/environments/environment_service_pb2.pyi,sha256=b-1XJDZf5pW3lF4UoZ2SQ_qGxvPodkRLTDJgXCHdIkE,2352
+_qwak_proto/qwak/administration/v1/environments/environment_service_pb2_grpc.py,sha256=1Vms7ww0HBCoDyhMarxvoYtBpqBy7nNbcrk3BjWZKdw,3161
 _qwak_proto/qwak/admiral/secret/v0/secret_pb2.py,sha256=HuEYHLETaK6KpVGO4gEwMXeyrPniUwKX3Uz6zFDzfIg,5085
 _qwak_proto/qwak/admiral/secret/v0/secret_pb2.pyi,sha256=lQkF7UWO_JmvK5LdSsQ_D33Kzfr56e-i3rm31pT-fsU,10408
 _qwak_proto/qwak/admiral/secret/v0/secret_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
@@ -511,9 +520,9 @@ _qwak_proto/qwak/model_group/model_group_pb2_grpc.py,sha256=OnYtTijqKbxC13VpOUKJ
 _qwak_proto/qwak/model_group/model_group_repository_details_pb2.py,sha256=_MMAM40q5l9Luubv40LEa-Uv4CCkbNmRt3wQ-916Zz0,3735
 _qwak_proto/qwak/model_group/model_group_repository_details_pb2.pyi,sha256=qOjP2x_4qC0gZayL6bkACeZU0GSfkH_6wIuWYHsNOe8,7320
 _qwak_proto/qwak/model_group/model_group_repository_details_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
-_qwak_proto/qwak/models/models_pb2.py,sha256=w2jAVJioQN5Ck1XW9jelfsOqmFRryVjmFukD49I_zUQ,18771
-_qwak_proto/qwak/models/models_pb2.pyi,sha256=T08V38RgCNy2x6ZW4_ZZdK09dZs7TzwXEHaGtNMW0y8,42643
-_qwak_proto/qwak/models/models_pb2_grpc.py,sha256=Z81RKpkngKhsHIgS3DWQ3BzHGYzRa-3iqUvPueMALCA,23625
+_qwak_proto/qwak/models/models_pb2.py,sha256=t3votBStZS8rUnzZU6FHp0jeQoHyZzf1IupRlUFdUyg,19152
+_qwak_proto/qwak/models/models_pb2.pyi,sha256=mUezH4pnv5LN1uxNH2niBuz8ZiU9am-ISUCzwZFjeXk,43564
+_qwak_proto/qwak/models/models_pb2_grpc.py,sha256=7PZKGhhNAlyGOR64kJNNY3aqzcDNvXjQfTullb0gGuU,25404
 _qwak_proto/qwak/models/models_query_pb2.py,sha256=uvIY_69L82Oy4nF0K_a4PwJPrb7vj6Z2a_B22KZeqwE,2809
 _qwak_proto/qwak/models/models_query_pb2.pyi,sha256=MYheg8TQtQuDv4BqHOWinzemOKhj35ZIVMHQy3YrAQA,7631
 _qwak_proto/qwak/models/models_query_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
@@ -619,7 +628,7 @@ _qwak_proto/qwak/workspace/workspace_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXH
 _qwak_proto/qwak/workspace/workspace_service_pb2.py,sha256=93uNm83VVrkLG_XVsuBOBTPs4UJF91YD1xJTOEbRyig,9239
 _qwak_proto/qwak/workspace/workspace_service_pb2.pyi,sha256=nKKCHwnovZhsy8TSVmdz-Vtl0nviOOoX56HD-41Xo08,13726
 _qwak_proto/qwak/workspace/workspace_service_pb2_grpc.py,sha256=yKGuexxTBza99Ihe0DSTniV2ZSd_AG47inHenqfi890,27193
-qwak/__init__.py,sha256=C0MUf6KSvgzS2LTrxIRD_8gvTk_vZPFiytZtvomwphE,585
+qwak/__init__.py,sha256=phXeOFx9au7BigBlw_4AhvnwLKQAGbYB0M88pYmn75o,585
 qwak/automations/__init__.py,sha256=qFZRvCxUUn8gcxkJR0v19ulHW2oJ0x6-Rif7HiheDP4,1522
 qwak/automations/automation_executions.py,sha256=5MeH_epYYWb8NKXgAozwT_jPyyUDednBHG7izloi7RY,3228
 qwak/automations/automations.py,sha256=2Wyrgj2mW5VmJzTKAXj3vQM_svYCuq_Bsu3_Vu9SdR4,12732
@@ -634,11 +643,15 @@ qwak/clients/administration/authenticated_user/__init__.py,sha256=47DEQpj8HBSa-_
 qwak/clients/administration/authenticated_user/client.py,sha256=Rnj2_fr3mAWmF7OgKeK2nvBu6CoxKtJ3GL8o_WaHAeY,1456
 qwak/clients/administration/authentication/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak/clients/administration/authentication/client.py,sha256=ggoM1DFqS3HTP3WIys8sVPluGhKmRE8Lm_9oWPSoZKg,1076
+qwak/clients/administration/cluster_v2/__init__.py,sha256=qsxlbOhYg9p5U-0RStC1CIJ86lEZ5UNIWuFPnL-8wGw,36
+qwak/clients/administration/cluster_v2/client.py,sha256=b9NlvAmXLzYB3K-3Ia-WhISN3ry_WS78Rrs4vC0qgOg,4193
 qwak/clients/administration/eco_system/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak/clients/administration/eco_system/client.py,sha256=fr3LGJQlheF9i8ZjvmcLti-8EfalKJ3fZNSnhV0OjQ8,5340
 qwak/clients/administration/eco_system/eco_system_utils.py,sha256=eSRih37ORNyiC4sdH_PV6rd5GsWswb0Mwzr5TxzhQEg,514
 qwak/clients/administration/environment/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-qwak/clients/administration/environment/client.py,sha256=_6ZaSVrMM-xdgrzz8tN9-3UHbCCdi3HjY8_4WpjQLdI,3273
+qwak/clients/administration/environment/client.py,sha256=DFHOZjRPvJNsWijMztzjcgu3IOsT2vGtFfPneOaY1BI,4332
+qwak/clients/administration/environment_v1/__init__.py,sha256=15_J34QBVU9X0d9_A6stiwYc6oLwkvuCCviROwROPXo,40
+qwak/clients/administration/environment_v1/client.py,sha256=812PaRjl8BdqtvXVqqm70nWB4tqdBR59BAAOrPJH-v8,1900
 qwak/clients/administration/self_service/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak/clients/administration/self_service/client.py,sha256=Pb1ZuSyYlYLdxq34dks1jOjrSNRE-ewFOZOiQjXlFRk,2602
 qwak/clients/alert_management/__init__.py,sha256=zAfCPJTj6oAMoj3vciLjRy8ULTTHCmmrdISuDCHASvk,43
@@ -1059,9 +1072,12 @@ qwak_services_mock/mocks/batch_job_manager_service.py,sha256=zVBBBkYdvWFwEXP_VHB
 qwak_services_mock/mocks/build_orchestrator_build_api.py,sha256=mtb6wEZug9zCiMHZVrfsU0YOP59rgBrpvZDjEInG7B8,5004
 qwak_services_mock/mocks/build_orchestrator_build_settings_api.py,sha256=oj_JTxY_3xj8FkgWkxQER_EoBFDqL2g21NPrU31fD6o,1379
 qwak_services_mock/mocks/build_orchestrator_service_api.py,sha256=IlM6VzovSSV6o15u1dARv7ocOqwRIWgItWMEW_-Fx_g,5226
+qwak_services_mock/mocks/cluster_v2_service.py,sha256=643DqfDmOu0s89YMxMeaJ4I3l2Iw-sObS1_0fChtE0k,5684
 qwak_services_mock/mocks/data_versioning_service.py,sha256=dUUrKwE3xc2bZtHa8YMg6wYn8zFa2BSZ5DaRsb1zT_Q,2453
 qwak_services_mock/mocks/deployment_management_service.py,sha256=T4fmFC1sUgEh5sXH6ilwyPmfRfEBFqyYrGGljl_Abg8,20609
 qwak_services_mock/mocks/ecosystem_service_api.py,sha256=f0-IeZMprFKvi1MvdW9aJZgO7BBdnJHQB8spDiVWy-A,1607
+qwak_services_mock/mocks/environment_v0_service.py,sha256=51Q68gyVCsh4XLsmFhQjB669oLfeSpDpxRXUOgs94HQ,7516
+qwak_services_mock/mocks/environment_v1_service.py,sha256=hV-YMoQ1OC4TxCIr4SYsw3Pa8w3QWE-pp2ZzNaf7d0w,1398
 qwak_services_mock/mocks/execution_management_service.py,sha256=L6uSHniFNaJDZUwRuMnt_-j6NmpJoOsD0RwpoI6w9ag,1190
 qwak_services_mock/mocks/feature_store_data_sources_manager_api.py,sha256=zfiq5x7xxg4IvBS1OndqSnFHDOckMW_-4K_Ajlko6no,5241
 qwak_services_mock/mocks/feature_store_entities_manager_api.py,sha256=I9wsVEsb0qVcD8jgekeC-EG-Ke8SziRLUND8uF4OJbc,3349
@@ -1082,7 +1098,7 @@ qwak_services_mock/mocks/logging_service.py,sha256=rkOud7CFAvb0xfP6WXf3kP541eXdY
 qwak_services_mock/mocks/model_management_service.py,sha256=H3bMTr7VLrGjZjIrjgrjBk3kNDy1UjQXOeUyL5eAL_o,4153
 qwak_services_mock/mocks/project_manager_service.py,sha256=WzcYzN3O04kSpvuTwX0QIF0HCh_ggsV9-gMel_vQVuc,3090
 qwak_services_mock/mocks/prompt_manager_service.py,sha256=tEhVUVwkvRy68r7MNX9V3V8ZBZ1ynqzr6v1zREMm0_s,9895
-qwak_services_mock/mocks/qwak_mocks.py,sha256=XszIZaJANxU6832-hmCDORdOXuuDfglb-w1gQrF2vKs,5997
+qwak_services_mock/mocks/qwak_mocks.py,sha256=Pbl70ILBE4eL_NGk0quDyF2j5PZQJj-Fpin3nlnm6Qg,6410
 qwak_services_mock/mocks/secret_service.py,sha256=BwNyhNSTmxHDACzrr7NHjZ0YchX3wHWuJkl1ioi0Aig,1406
 qwak_services_mock/mocks/self_service_user_service.py,sha256=UK22V-wAc0zHScXCE0QKDO0TdbrHf_uk7tEEIM47MpI,1205
 qwak_services_mock/mocks/system_secret_service.py,sha256=S8MfGGEex7bs0AvnAf8SbPOzE-YDfCvePT1TxXWzHRQ,1896
@@ -1092,9 +1108,9 @@ qwak_services_mock/mocks/utils/exception_handlers.py,sha256=Fhi9cx_qNytDnl4BtZmE
 qwak_services_mock/mocks/vector_serving_api.py,sha256=7ZSmDJVdw7_ne7BxLLrmy8eZkzVCeZ508gfL51wpKeY,5722
 qwak_services_mock/mocks/vectors_management_api.py,sha256=ePh8l3MlDGzs7E4NuPZzF5xK-bC9AdMgDaNZrq3FkHc,3594
 qwak_services_mock/mocks/workspace_manager_service_mock.py,sha256=O9ZSwln4T4kHVkR_usXnDQtarTeNfffSMON0P6wbT4g,7740
-qwak_services_mock/services_mock.py,sha256=0BWwS2re9ryO3JrJJgSNyEQ0lDOjyrpV36oa8t7Pd7A,19163
+qwak_services_mock/services_mock.py,sha256=8XyKtzEWxfvp4bVMSAlJ90GhkXKROZrgWeH25Iq3ETA,20601
 qwak_services_mock/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_services_mock/utils/service_utils.py,sha256=ZlB0CnB1J6oBn6_m7fQO2U8tKoboHdUa6ljjkRMYNXU,265
-qwak_core-0.6.5.dist-info/METADATA,sha256=iNWIdha_DtCoRJii488Nd5YXwD9lrpBZqU1XXqH6URY,1986
-qwak_core-0.6.5.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-qwak_core-0.6.5.dist-info/RECORD,,
+qwak_core-0.6.7.dist-info/METADATA,sha256=oOduH81p6KeOQXYZNJO1FNgyUfq73YmOBk_ycs6zgdY,1986
+qwak_core-0.6.7.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+qwak_core-0.6.7.dist-info/RECORD,,

qwak_services_mock/mocks/cluster_v2_service.py

@@ -0,0 +1,146 @@
+from typing import Optional
+
+import grpc
+
+from _qwak_proto.qwak.administration.cluster.v2.cluster_pb2 import (
+    ClusterState,
+)
+from _qwak_proto.qwak.administration.cluster.v2.cluster_service_pb2 import (
+    CreateClusterCreationRequestResponse,
+    DeleteClusterResponse,
+    GetClusterResponse,
+    GetClusterStateResponse,
+    ListClustersResponse,
+    UpdateClusterConfigurationResponse,
+)
+from _qwak_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")

qwak_services_mock/mocks/environment_v0_service.py

@@ -0,0 +1,175 @@
+from typing import Optional
+
+import grpc
+
+from _qwak_proto.qwak.administration.v0.environments.environment_pb2 import (
+    QwakEnvironment,
+)
+from _qwak_proto.qwak.administration.v0.environments.environment_service_pb2 import (
+    CreateEnvironmentResponse,
+    GetEnvironmentApplicationUserCredentialsResponse,
+    GetEnvironmentResponse,
+    ListEnvironmentsResponse,
+    SetEnvironmentApplicationUserCredentialsResponse,
+    UpdateEnvironmentConfigurationResponse,
+    UpdateEnvironmentPersonalizationResponse,
+)
+from _qwak_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()

qwak_services_mock/mocks/environment_v1_service.py

@@ -0,0 +1,37 @@
+from typing import Optional
+
+import grpc
+
+from _qwak_proto.qwak.administration.v0.environments.environment_pb2 import (
+    QwakEnvironment,
+)
+from _qwak_proto.qwak.administration.v1.environments.environment_service_pb2 import (
+    CreateEnvironmentResponse,
+)
+from _qwak_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"))

qwak_services_mock/mocks/qwak_mocks.py

@@ -24,11 +24,14 @@ from qwak_services_mock.mocks.build_orchestrator_build_settings_api import (
 from qwak_services_mock.mocks.build_orchestrator_service_api import (
     BuildOrchestratorServiceApiMock,
 )
+from qwak_services_mock.mocks.cluster_v2_service import ClusterV2ServiceMock
 from qwak_services_mock.mocks.data_versioning_service import DataVersioningServiceMock
 from qwak_services_mock.mocks.deployment_management_service import (
     DeploymentManagementServiceMock,
 )
 from qwak_services_mock.mocks.ecosystem_service_api import EcoSystemServiceMock
+from qwak_services_mock.mocks.environment_v0_service import EnvironmentV0ServiceMock
+from qwak_services_mock.mocks.environment_v1_service import EnvironmentV1ServiceMock
 from qwak_services_mock.mocks.execution_management_service import (
     ExecutionManagementServiceMock,
 )
@@ -123,3 +126,6 @@ class QwakMocks:
     integration_management_service: IntegrationManagementServiceMock
     prompt_manager_service: PromptManagerServiceMock
     location_discovery_service: LocationDiscoveryServiceApiMock
+    cluster_v2_service_mock: ClusterV2ServiceMock
+    environment_v0_service_mock: EnvironmentV0ServiceMock
+    environment_v1_service_mock: EnvironmentV1ServiceMock

qwak_services_mock/services_mock.py

@@ -4,9 +4,18 @@ from typing import Any, Generator, List, Tuple
 import grpc
 import pytest
 
+from _qwak_proto.qwak.administration.cluster.v2.cluster_service_pb2_grpc import (
+    add_ClusterServiceServicer_to_server,
+)
 from _qwak_proto.qwak.administration.v0.authentication.authentication_service_pb2_grpc import (
     add_AuthenticationServiceServicer_to_server,
 )
+from _qwak_proto.qwak.administration.v0.environments.environment_service_pb2_grpc import (
+    add_EnvironmentServiceServicer_to_server as add_EnvironmentV0ServiceServicer_to_server,
+)
+from _qwak_proto.qwak.administration.v1.environments.environment_service_pb2_grpc import (
+    add_EnvironmentServiceServicer_to_server as add_EnvironmentV1ServiceServicer_to_server,
+)
 from _qwak_proto.qwak.admiral.secret.v0.system_secret_service_pb2_grpc import (
     add_SystemSecretServiceServicer_to_server,
 )
@@ -140,11 +149,14 @@ from qwak_services_mock.mocks.build_orchestrator_build_settings_api import (
 from qwak_services_mock.mocks.build_orchestrator_service_api import (
     BuildOrchestratorServiceApiMock,
 )
+from qwak_services_mock.mocks.cluster_v2_service import ClusterV2ServiceMock
 from qwak_services_mock.mocks.data_versioning_service import DataVersioningServiceMock
 from qwak_services_mock.mocks.deployment_management_service import (
     DeploymentManagementServiceMock,
 )
 from qwak_services_mock.mocks.ecosystem_service_api import EcoSystemServiceMock
+from qwak_services_mock.mocks.environment_v0_service import EnvironmentV0ServiceMock
+from qwak_services_mock.mocks.environment_v1_service import EnvironmentV1ServiceMock
 from qwak_services_mock.mocks.execution_management_service import (
     ExecutionManagementServiceMock,
 )
@@ -246,7 +258,14 @@ def qwak_container():
         vector_store,
         workspace_manager,
     )
-    from qwak.clients.administration import authentication, eco_system, self_service
+    from qwak.clients.administration import (
+        authentication,
+        cluster_v2,
+        eco_system,
+        environment,
+        environment_v1,
+        self_service,
+    )
     from qwak.clients.integration_management import integration_manager_client
     from qwak.clients.prompt_manager import prompt_manager_client
     from qwak.clients.system_secret import system_secret_client
@@ -262,8 +281,11 @@ def qwak_container():
             analytics,
             batch_job_management,
             build_orchestrator,
+            cluster_v2,
             data_versioning,
             deployment,
+            environment,
+            environment_v1,
             instance_template,
             feature_store,
             file_versioning,
@@ -495,6 +517,21 @@ def attach_servicers(free_port, server):
                 LocationDiscoveryServiceApiMock,
                 add_LocationDiscoveryServiceServicer_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),
         ],
     )