code: def test_create_cluster(self, create_cluster_mock, *_):
      operator = DataprocCreateClusterOperator(
          task_id="create_cluster",
          ssh_public_keys=SSH_PUBLIC_KEYS,
          folder_id=FOLDER_ID,
          subnet_id=SUBNET_ID,
          zone=AVAILABILITY_ZONE_ID,
          connection_id=CONNECTION_ID,
          s3_bucket=S3_BUCKET_NAME_FOR_LOGS,
          cluster_image_version=CLUSTER_IMAGE_VERSION,
          log_group_id=LOG_GROUP_ID,
      )
      context = {"task_instance": MagicMock()}
      operator.execute(context)
      create_cluster_mock.assert_called_once_with(
          cluster_description="",
          cluster_image_version="1.4",
          cluster_name=None,
          computenode_count=0,
          computenode_disk_size=None,
          computenode_disk_type=None,
          computenode_resource_preset=None,
          computenode_max_hosts_count=None,
          computenode_measurement_duration=None,
          computenode_warmup_duration=None,
          computenode_stabilization_duration=None,
          computenode_preemptible=False,
          computenode_cpu_utilization_target=None,
          computenode_decommission_timeout=None,
          datanode_count=1,
          datanode_disk_size=None,
          datanode_disk_type=None,
          datanode_resource_preset=None,
          folder_id="my_folder_id",
          masternode_disk_size=None,
          masternode_disk_type=None,
          masternode_resource_preset=None,
          s3_bucket="my_bucket_name",
          service_account_id=None,
          services=("HDFS", "YARN", "MAPREDUCE", "HIVE", "SPARK"),
          ssh_public_keys=[
              "ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAAAYQCxO38tKAJXIs9ivPxt7AYdfybgtAR1ow3Qkb9GPQ6wkFHQqcFD"
              "e6faKCxH6iDRteo4D8L8BxwzN42uZSB0nfmjkIxFTcEU3mFSXEbWByg78aoddMrAAjatyrhH1pON6P0="
          ],
          subnet_id="my_subnet_id",
          zone="ru-central1-c",
          log_group_id=LOG_GROUP_ID,
          properties=None,
          enable_ui_proxy=False,
          host_group_ids=None,
          security_group_ids=None,
          labels=None,
          initialization_actions=None,
      )
      context["task_instance"].xcom_push.assert_has_calls(
          [
              call(
                  key="cluster_id",
                  value=create_cluster_mock().response.id,
              ),
              call(
                  key="yandexcloud_connection_id", value=CONNECTION_ID
              ),
          ]
      )
  
file_path: airflow/tests/providers/yandex/operators/test_yandexcloud_dataproc.py
imports: from __future__ import annotations
  import datetime
  from unittest.mock import MagicMock, call, patch
  from airflow.models.dag import DAG
  from airflow.providers.yandex.operators.yandexcloud_dataproc import DataprocCreateClusterOperator, DataprocCreateHiveJobOperator, DataprocCreateMapReduceJobOperator, DataprocCreatePysparkJobOperator, DataprocCreateSparkJobOperator, DataprocDeleteClusterOperator
label_desc: Use image node_id 2 for calling the TestDataprocClusterCreateOperator obj's underlying member method code with example usage: obj.test_create_cluster(create_cluster_mock) without return types