def test_assert_valid_hook_call(self, mock_hook):
task = CloudMemorystoreUpdateInstanceOperator(
task_id=TEST_TASK_ID,
update_mask=TEST_UPDATE_MASK,
instance=TEST_INSTANCE,
location=TEST_LOCATION,
instance_id=TEST_INSTANCE_ID,
project_id=TEST_PROJECT_ID,
retry=TEST_RETRY,
timeout=TEST_TIMEOUT,
metadata=TEST_METADATA,
gcp_conn_id=TEST_GCP_CONN_ID,
)
task.execute(mock.MagicMock())
mock_hook.assert_called_once_with(gcp_conn_id=TEST_GCP_CONN_ID)
mock_hook.return_value.update_instance.assert_called_once_with(
update_mask=TEST_UPDATE_MASK,
instance=TEST_INSTANCE,
location=TEST_LOCATION,
instance_id=TEST_INSTANCE_ID,
project_id=TEST_PROJECT_ID,
retry=TEST_RETRY,
timeout=TEST_TIMEOUT,
metadata=TEST_METADATA,
)
location="-",
page_size=100,
project_id=GCP_PROJECT_ID)
# [END howto_operator_list_instances]
# [START howto_operator_list_instances_result]
list_instances_result = BashOperator(
task_id="list-instances-result",
bash_command="echo \"{{ task_instance.xcom_pull('get-instance') }}\"")
# [END howto_operator_list_instances_result]
# [START howto_operator_update_instance]
update_instance = CloudMemorystoreUpdateInstanceOperator(
task_id="update-instance",
location="europe-north1",
instance_id=INSTANCE_NAME,
project_id=GCP_PROJECT_ID,
update_mask={"paths": ["memory_size_gb"]},
instance={"memory_size_gb": 2},
)
# [END howto_operator_update_instance]
# [START howto_operator_set_acl_permission]
set_acl_permission = GCSBucketCreateAclEntryOperator(
task_id="gcs-set-acl-permission",
bucket=BUCKET_NAME,
entity=
"user-{{ task_instance.xcom_pull('get-instance')['persistenceIamIdentity']"
".split(':', 2)[1] }}",
role="OWNER",
)
# [END howto_operator_set_acl_permission]
