anyscale 0.24.29__tar.gz → 0.24.30__tar.gz

{anyscale-0.24.29 → anyscale-0.24.30}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: anyscale
-Version: 0.24.29
+Version: 0.24.30
 Summary: Command Line Interface for Anyscale
 Author: Anyscale Inc.
 License: AS License

{anyscale-0.24.29 → anyscale-0.24.30}/anyscale/client/README.md

@@ -85,6 +85,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**archive_compute_template_api_v2_compute_templates_compute_template_id_archive_post**](docs/DefaultApi.md#archive_compute_template_api_v2_compute_templates_compute_template_id_archive_post) | **POST** /api/v2/compute_templates/{compute_template_id}/archive | Archive Compute Template
 *DefaultApi* | [**archive_job_api_v2_decorated_ha_jobs_production_job_id_archive_post**](docs/DefaultApi.md#archive_job_api_v2_decorated_ha_jobs_production_job_id_archive_post) | **POST** /api/v2/decorated_ha_jobs/{production_job_id}/archive | Archive Job
 *DefaultApi* | [**attach_machine_pool_to_cloud_api_v2_machine_pools_attach_post**](docs/DefaultApi.md#attach_machine_pool_to_cloud_api_v2_machine_pools_attach_post) | **POST** /api/v2/machine_pools/attach | Attach Machine Pool To Cloud
+*DefaultApi* | [**ban_organization_api_v2_organizations_organization_id_ban_put**](docs/DefaultApi.md#ban_organization_api_v2_organizations_organization_id_ban_put) | **PUT** /api/v2/organizations/{organization_id}/ban | Ban Organization
 *DefaultApi* | [**batch_create_invitations_api_v2_organization_invitations_batch_create_post**](docs/DefaultApi.md#batch_create_invitations_api_v2_organization_invitations_batch_create_post) | **POST** /api/v2/organization_invitations/batch_create | Batch Create Invitations
 *DefaultApi* | [**batch_get_job_api_v2_decorated_ha_jobs_batch_get_post**](docs/DefaultApi.md#batch_get_job_api_v2_decorated_ha_jobs_batch_get_post) | **POST** /api/v2/decorated_ha_jobs/batch_get | Batch Get Job
 *DefaultApi* | [**begin_cluster_startup_api_v2_cluster_startups_post**](docs/DefaultApi.md#begin_cluster_startup_api_v2_cluster_startups_post) | **POST** /api/v2/cluster_startups/ | Begin Cluster Startup

{anyscale-0.24.29 → anyscale-0.24.30}/anyscale/client/openapi_client/api/default_api.py

@@ -1219,6 +1219,118 @@ class DefaultApi(object):
             _request_timeout=local_var_params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def ban_organization_api_v2_organizations_organization_id_ban_put(self, organization_id, **kwargs):  # noqa: E501
+        """Ban Organization  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.ban_organization_api_v2_organizations_organization_id_ban_put(organization_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str organization_id: (required)
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: object
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        return self.ban_organization_api_v2_organizations_organization_id_ban_put_with_http_info(organization_id, **kwargs)  # noqa: E501
+
+    def ban_organization_api_v2_organizations_organization_id_ban_put_with_http_info(self, organization_id, **kwargs):  # noqa: E501
+        """Ban Organization  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.ban_organization_api_v2_organizations_organization_id_ban_put_with_http_info(organization_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str organization_id: (required)
+        :param _return_http_data_only: response data without head status code
+                                       and headers
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: tuple(object, status_code(int), headers(HTTPHeaderDict))
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        local_var_params = locals()
+
+        all_params = [
+            'organization_id'
+        ]
+        all_params.extend(
+            [
+                'async_req',
+                '_return_http_data_only',
+                '_preload_content',
+                '_request_timeout'
+            ]
+        )
+
+        for key, val in six.iteritems(local_var_params['kwargs']):
+            if key not in all_params:
+                raise ApiTypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method ban_organization_api_v2_organizations_organization_id_ban_put" % key
+                )
+            local_var_params[key] = val
+        del local_var_params['kwargs']
+        # verify the required parameter 'organization_id' is set
+        if self.api_client.client_side_validation and ('organization_id' not in local_var_params or  # noqa: E501
+                                                        local_var_params['organization_id'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `organization_id` when calling `ban_organization_api_v2_organizations_organization_id_ban_put`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'organization_id' in local_var_params:
+            path_params['organization_id'] = local_var_params['organization_id']  # noqa: E501
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/api/v2/organizations/{organization_id}/ban', 'PUT',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='object',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=local_var_params.get('async_req'),
+            _return_http_data_only=local_var_params.get('_return_http_data_only'),  # noqa: E501
+            _preload_content=local_var_params.get('_preload_content', True),
+            _request_timeout=local_var_params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
     def batch_create_invitations_api_v2_organization_invitations_batch_create_post(self, create_organization_invitation, **kwargs):  # noqa: E501
         """Batch Create Invitations  # noqa: E501
 

{anyscale-0.24.29 → anyscale-0.24.30}/anyscale/client/openapi_client/models/company_size.py

@@ -29,11 +29,11 @@ class CompanySize(object):
     allowed enum values
     """
     _1_50 = "1 - 50"
-    _50_200 = "50 - 200"
-    _200_500 = "200 - 500"
+    _51_200 = "51 - 200"
+    _201_500 = "201 - 500"
     _500 = "> 500"
 
-    allowable_values = [_1_50, _50_200, _200_500, _500]  # noqa: E501
+    allowable_values = [_1_50, _51_200, _201_500, _500]  # noqa: E501
 
     """
     Attributes:

{anyscale-0.24.29 → anyscale-0.24.30}/anyscale/client/openapi_client/models/organization.py

@@ -45,7 +45,8 @@ class Organization(object):
         'endpoints_user_id': 'str',
         'endpoints_stripe_customer_id': 'str',
         'endpoints_metronome_customer_id': 'str',
-        'endpoints_is_free_credit_used_up': 'bool'
+        'endpoints_is_free_credit_used_up': 'bool',
+        'metronome_customer_id': 'str'
     }
 
     attribute_map = {
@@ -61,10 +62,11 @@ class Organization(object):
         'endpoints_user_id': 'endpoints_user_id',
         'endpoints_stripe_customer_id': 'endpoints_stripe_customer_id',
         'endpoints_metronome_customer_id': 'endpoints_metronome_customer_id',
-        'endpoints_is_free_credit_used_up': 'endpoints_is_free_credit_used_up'
+        'endpoints_is_free_credit_used_up': 'endpoints_is_free_credit_used_up',
+        'metronome_customer_id': 'metronome_customer_id'
     }
 
-    def __init__(self, id=None, name=None, public_identifier=None, default_cloud_id=None, sso_config_id=None, sso_required=False, is_general_platform_enabled=None, is_private_endpoints_enabled=None, is_usage_blocked=None, endpoints_user_id=None, endpoints_stripe_customer_id=None, endpoints_metronome_customer_id=None, endpoints_is_free_credit_used_up=None, local_vars_configuration=None):  # noqa: E501
+    def __init__(self, id=None, name=None, public_identifier=None, default_cloud_id=None, sso_config_id=None, sso_required=False, is_general_platform_enabled=None, is_private_endpoints_enabled=None, is_usage_blocked=None, endpoints_user_id=None, endpoints_stripe_customer_id=None, endpoints_metronome_customer_id=None, endpoints_is_free_credit_used_up=None, metronome_customer_id=None, local_vars_configuration=None):  # noqa: E501
         """Organization - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
@@ -83,6 +85,7 @@ class Organization(object):
         self._endpoints_stripe_customer_id = None
         self._endpoints_metronome_customer_id = None
         self._endpoints_is_free_credit_used_up = None
+        self._metronome_customer_id = None
         self.discriminator = None
 
         self.id = id
@@ -105,6 +108,8 @@ class Organization(object):
             self.endpoints_metronome_customer_id = endpoints_metronome_customer_id
         if endpoints_is_free_credit_used_up is not None:
             self.endpoints_is_free_credit_used_up = endpoints_is_free_credit_used_up
+        if metronome_customer_id is not None:
+            self.metronome_customer_id = metronome_customer_id
 
     @property
     def id(self):
@@ -391,6 +396,27 @@ class Organization(object):
 
         self._endpoints_is_free_credit_used_up = endpoints_is_free_credit_used_up
 
+    @property
+    def metronome_customer_id(self):
+        """Gets the metronome_customer_id of this Organization.  # noqa: E501
+
+
+        :return: The metronome_customer_id of this Organization.  # noqa: E501
+        :rtype: str
+        """
+        return self._metronome_customer_id
+
+    @metronome_customer_id.setter
+    def metronome_customer_id(self, metronome_customer_id):
+        """Sets the metronome_customer_id of this Organization.
+
+
+        :param metronome_customer_id: The metronome_customer_id of this Organization.  # noqa: E501
+        :type: str
+        """
+
+        self._metronome_customer_id = metronome_customer_id
+
     def to_dict(self):
         """Returns the model properties as a dict"""
         result = {}

{anyscale-0.24.29 → anyscale-0.24.30}/anyscale/commands/workspace_commands.py

@@ -7,9 +7,7 @@ import click
 import requests
 
 from anyscale.authenticate import get_auth_api_client
-from anyscale.client.openapi_client.models.create_experimental_workspace import (
-    CreateExperimentalWorkspace,
-)
+from anyscale.cli_logger import BlockLogger
 from anyscale.client.openapi_client.models.execute_shell_command_options import (
     ExecuteShellCommandOptions,
 )
@@ -21,6 +19,9 @@ from anyscale.util import get_endpoint
 from anyscale.workspace import get_workspace_root_or_throw, write_workspace_id_to_disk
 
 
+log = BlockLogger()  # CLI Logger
+
+
 @click.group("workspace", help="Interact with workspaces on Anyscale.")
 def workspace_cli() -> None:
     pass
@@ -32,8 +33,10 @@ def workspace_cli() -> None:
 @click.option(
     "--name", "-n", required=True, help="Name of the workspace to create.",
 )
-@click.option("--project-id", required=True)
-@click.option("--cloud-id", required=True)
+@click.option("--project-id", required=False)
+@click.option(
+    "--cloud-id", required=True
+)  # Note: we could further improve UX by making cloud_id optional since we can always infer it from compute config
 @click.option("--cluster-env-build-id", required=False)
 @click.option(
     "--docker", required=False, default=None, help=("Custom docker image URI."),
@@ -83,18 +86,17 @@ def create(  # noqa: PLR0913
         # Create docker build_id
         cluster_env_build_id = BYODInfo(docker, python_version, ray_version).encode()
 
-    auth_api_client = get_auth_api_client()
-    api_client = auth_api_client.api_client
-    api_client.create_workspace_api_v2_experimental_workspaces_post(
-        CreateExperimentalWorkspace(
-            name=name,
-            project_id=project_id,
-            cloud_id=cloud_id,
-            compute_config_id=compute_config_id,
-            cluster_environment_build_id=cluster_env_build_id,
-        )
+    workspace_controller = WorkspaceController()
+    workspace_controller.create(
+        name=name,
+        cloud_id=cloud_id,
+        compute_config_id=compute_config_id,
+        project_id=project_id,
+        cluster_environment_build_id=cluster_env_build_id,
     )
 
+    log.info(f"Workspace {name} created successfully.")
+
 
 @workspace_cli.command(
     name="start", help="Start an existing workspace on Anyscale.",
@@ -331,6 +333,7 @@ def copy_command(remote_path, local_path) -> None:
             anyscale workspace cp "/tmp/" ~/Downlaods
 
     """
+
     _do_copy(remote_path, local_path)
 
 
@@ -352,6 +355,7 @@ def _do_pull(pull_git_state):
         rsync_filters=[".gitignore"],
         rsync_excludes=[".anyscale.yaml", ".git/objects/info/alternates"]
         + ([] if pull_git_state else [".git"]),
+        delete_existing_files_in_destination=True,
     )
 
 
@@ -368,6 +372,7 @@ def _do_push(push_git_state):
         # different from the remote one (has large .pack files).
         rsync_excludes=[".anyscale.yaml", ".git/objects/info/alternates"]
         + ([] if push_git_state else [".git"]),
+        delete_existing_files_in_destination=True,
     )
 
 

{anyscale-0.24.29 → anyscale-0.24.30}/anyscale/controllers/cloud_controller.py

@@ -1398,6 +1398,10 @@ class CloudController(BaseController):
             self._update_customer_aggregated_logs_config(
                 cloud_id=cloud_id, is_enabled=enable_log_ingestion,
             )
+            self.log.info(
+                f"Successfully updated log ingestion configuration for cloud, "
+                f"{cloud_id} to {enable_log_ingestion}"
+            )
 
     def set_default_cloud(
         self, cloud_name: Optional[str], cloud_id: Optional[str],

{anyscale-0.24.29 → anyscale-0.24.30}/anyscale/controllers/workspace_controller.py

@@ -3,14 +3,19 @@ import shlex
 import subprocess
 from typing import Any, List, Optional
 
+import click
 import tabulate
 
 from anyscale.cli_logger import BlockLogger
+from anyscale.client.openapi_client.models.create_experimental_workspace import (
+    CreateExperimentalWorkspace,
+)
 from anyscale.client.openapi_client.models.experimental_workspace import (
     ExperimentalWorkspace,
 )
 from anyscale.controllers.base_controller import BaseController
 from anyscale.feature_flags import FLAG_DEFAULT_WORKING_DIR_FOR_PROJ
+from anyscale.project import get_default_project
 from anyscale.util import get_endpoint
 from anyscale.utils.workspace_utils import (
     extract_workspace_parameters,
@@ -39,6 +44,41 @@ class WorkspaceController(BaseController):
         super().__init__(initialize_auth_api_client=initialize_auth_api_client)
         self.log = log
 
+    def create(
+        self,
+        name: str,
+        cloud_id: str,
+        compute_config_id: str,
+        cluster_environment_build_id: str,
+        project_id: Optional[str],
+    ) -> None:
+        compute_config = self.api_client.get_compute_template_api_v2_compute_templates_template_id_get(
+            compute_config_id
+        ).result
+
+        if compute_config.config.cloud_id != cloud_id:
+            raise click.ClickException(
+                "The compute config and cloud must be in the same cloud."
+            )
+
+        if not project_id:
+            default_project = get_default_project(
+                parent_cloud_id=cloud_id,
+                api_client=self.api_client,
+                anyscale_api_client=self.anyscale_api_client,  # type: ignore
+            )
+            project_id = default_project.id
+
+        self.api_client.create_workspace_api_v2_experimental_workspaces_post(
+            CreateExperimentalWorkspace(
+                name=name,
+                project_id=project_id,
+                cloud_id=cloud_id,
+                compute_config_id=compute_config_id,
+                cluster_environment_build_id=cluster_environment_build_id,
+            )
+        )
+
     def list(self) -> None:
         """
         prints a non-exhaustive tabular list of information about non-deleted workspaces.
@@ -139,6 +179,7 @@ class WorkspaceController(BaseController):
         down: bool,
         rsync_filters: List[str],
         rsync_excludes: List[str],
+        delete_existing_files_in_destination: bool = False,
     ) -> None:
         """
         Args:
@@ -148,6 +189,7 @@ class WorkspaceController(BaseController):
             down: Whether to sync from 1. remote to local, or 2. local to remote.
             rsync_filters: A list of rsync filters to apply.
             rsync_excludes: A list of rsync excludes to apply.
+            delete_existing_files_in_destination: Whether to delete existing files in the destination.
         """
         workspace_command_context = self._get_workspace_command_context()
         ssh_info = workspace_command_context.ssh
@@ -172,9 +214,11 @@ class WorkspaceController(BaseController):
             "--rsh",
             subprocess.list2cmdline(base_ssh_command),
             "-rvzl",
-            "--delete",
         ]
 
+        if delete_existing_files_in_destination:
+            rsync_command.append("--delete")
+
         for rsync_exclude in rsync_excludes:
             rsync_command.extend(["--exclude", rsync_exclude])
 

anyscale-0.24.30/anyscale/version.py

@@ -0,0 +1 @@
+__version__ = "0.24.30"

{anyscale-0.24.29 → anyscale-0.24.30}/anyscale.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: anyscale
-Version: 0.24.29
+Version: 0.24.30
 Summary: Command Line Interface for Anyscale
 Author: Anyscale Inc.
 License: AS License

{anyscale-0.24.29 → anyscale-0.24.30}/anyscale.egg-info/SOURCES.txt

@@ -1866,6 +1866,7 @@ tests/commands/BUILD.bazel
 tests/commands/test_cloud_command.py
 tests/commands/test_login_commands.py
 tests/commands/test_service_commands.py
+tests/commands/test_workspace_commands.py
 tests/connect_utils/BUILD.bazel
 tests/connect_utils/test_prepare_cluster.py
 tests/connect_utils/test_project_block.py
@@ -1888,6 +1889,7 @@ tests/controllers/test_machine_pool_controller.py
 tests/controllers/test_project_controller.py
 tests/controllers/test_schedule_controller.py
 tests/controllers/test_service_controller.py
+tests/controllers/test_workspace_controller.py
 tests/gcp_responses/access_service_account_permissions.json
 tests/gcp_responses/add_firewall_policy_rule.json
 tests/gcp_responses/add_firewall_policy_rule_done.json

anyscale-0.24.30/tests/commands/test_workspace_commands.py

@@ -0,0 +1,163 @@
+from typing import Generator
+from unittest.mock import Mock, patch
+
+from click.testing import CliRunner
+import pytest
+
+from anyscale.commands.workspace_commands import create
+
+
+class MockWorkspaceController(Mock):
+    pass
+
+
+@pytest.fixture()
+def mock_workspace_controller() -> Generator[MockWorkspaceController, None, None]:
+    mock_workspace_controller = MockWorkspaceController()
+    mock_workspace_controller_cls = Mock(return_value=mock_workspace_controller,)
+
+    with patch(
+        "anyscale.commands.workspace_commands.WorkspaceController",
+        new=mock_workspace_controller_cls,
+    ):
+        yield mock_workspace_controller
+
+
+@pytest.fixture()
+def mock_workspace_sdk() -> Generator[MockWorkspaceController, None, None]:
+    mock_workspace_controller = MockWorkspaceController()
+    mock_workspace_controller_cls = Mock(return_value=mock_workspace_controller,)
+
+    with patch(
+        "anyscale.commands.workspace_commands.WorkspaceController",
+        new=mock_workspace_controller_cls,
+    ):
+        yield mock_workspace_controller
+
+
+def test_create(mock_workspace_controller: MockWorkspaceController):
+    """Tests the logic for creating new workspace
+    """
+    runner = CliRunner()
+
+    # test required options
+    runner.invoke(
+        create,
+        args=[
+            "--name",
+            "fake-workspace-name",
+            "--cloud-id",
+            "fake-cloud-id",
+            "--compute-config-id",
+            "fake-compute-config-id",
+            "--cluster-env-build-id",
+            "fake-cluster-environment-build-id",
+        ],
+    )
+    mock_workspace_controller.create.assert_called_once_with(
+        name="fake-workspace-name",
+        cloud_id="fake-cloud-id",
+        compute_config_id="fake-compute-config-id",
+        project_id=None,
+        cluster_environment_build_id="fake-cluster-environment-build-id",
+    )
+    mock_workspace_controller.reset_mock()
+
+    # test project_id option
+    runner.invoke(
+        create,
+        args=[
+            "--name",
+            "fake-workspace-name",
+            "--cloud-id",
+            "fake-cloud-id",
+            "--compute-config-id",
+            "fake-compute-config-id",
+            "--cluster-env-build-id",
+            "fake-cluster-environment-build-id",
+            "--project-id",
+            "fake-project-id",
+        ],
+    )
+    mock_workspace_controller.create.assert_called_once_with(
+        name="fake-workspace-name",
+        cloud_id="fake-cloud-id",
+        compute_config_id="fake-compute-config-id",
+        project_id="fake-project-id",
+        cluster_environment_build_id="fake-cluster-environment-build-id",
+    )
+    mock_workspace_controller.reset_mock()
+
+    # test mutual exclusive arguments (--docker and --cluster-env-build-id)
+    result = runner.invoke(
+        create,
+        args=[
+            "--name",
+            "fake-workspace-name",
+            "--cloud-id",
+            "fake-cloud-id",
+            "--compute-config-id",
+            "fake-compute-config-id",
+            "--project-id",
+            "fake-project-id",
+            "--cluster-env-build-id",
+            "fake-cluster-environment-build-id",
+            "--docker",
+            "fake-docker",
+        ],
+    )
+    assert result.exception is not None
+
+    # test mutual exclusive arguments (--docker and --python-version)
+    result = runner.invoke(
+        create,
+        args=[
+            "--name",
+            "fake-workspace-name",
+            "--cloud-id",
+            "fake-cloud-id",
+            "--compute-config-id",
+            "fake-compute-config-id",
+            "--project-id",
+            "fake-project-id",
+            "--python-version",
+            "fake-python-version",
+            "--docker",
+            "fake-docker",
+        ],
+    )
+    assert result.exception is not None
+
+    # test mutual exclusive arguments (--docker and --ray-version)
+    result = runner.invoke(
+        create,
+        args=[
+            "--name",
+            "fake-workspace-name",
+            "--cloud-id",
+            "fake-cloud-id",
+            "--compute-config-id",
+            "fake-compute-config-id",
+            "--project-id",
+            "fake-project-id",
+            "--ray-version",
+            "fake-ray-version",
+            "--docker",
+            "fake-docker",
+        ],
+    )
+    assert result.exception is not None
+
+    # test at least one of --docker or --cluster-env-build-id is required
+    result = runner.invoke(
+        create,
+        args=[
+            "--name",
+            "fake-workspace-name",
+            "--cloud-id",
+            "fake-cloud-id",
+            "--compute-config-id",
+            "fake-compute-config-id",
+        ],
+    )
+    assert result.exception is not None

{anyscale-0.24.29 → anyscale-0.24.30}/tests/controllers/test_cloud_controller.py

@@ -3077,5 +3077,9 @@ def test_update_cloud_config(enable_log_ingestion: Optional[bool]):
             cloud_controller.api_client.update_customer_aggregated_logs_config_api_v2_clouds_cloud_id_update_customer_aggregated_logs_config_put.assert_called_with(
                 cloud_id=cloud_id, is_enabled=enable_log_ingestion,
             )
+            cloud_controller.log.info.assert_called_with(
+                f"Successfully updated log ingestion configuration for cloud, {cloud_id} to {enable_log_ingestion}"
+            )
         else:
             cloud_controller.api_client.update_customer_aggregated_logs_config_api_v2_clouds_cloud_id_update_customer_aggregated_logs_config_put.assert_not_called()
+            cloud_controller.log.info.assert_not_called()