def test_assert_valid_hook_call(self, mock_hook):
task = CloudMemorystoreFailoverInstanceOperator(
task_id=TEST_TASK_ID,
location=TEST_LOCATION,
instance=TEST_INSTANCE_NAME,
data_protection_mode=TEST_DATA_PROTECTION_MODE,
project_id=TEST_PROJECT_ID,
retry=TEST_RETRY,
timeout=TEST_TIMEOUT,
metadata=TEST_METADATA,
gcp_conn_id=TEST_GCP_CONN_ID,
impersonation_chain=TEST_IMPERSONATION_CHAIN,
)
task.execute(context=mock.MagicMock())
mock_hook.assert_called_once_with(
gcp_conn_id=TEST_GCP_CONN_ID,
impersonation_chain=TEST_IMPERSONATION_CHAIN,
)
mock_hook.return_value.failover_instance.assert_called_once_with(
location=TEST_LOCATION,
instance=TEST_INSTANCE_NAME,
data_protection_mode=TEST_DATA_PROTECTION_MODE,
project_id=TEST_PROJECT_ID,
retry=TEST_RETRY,
timeout=TEST_TIMEOUT,
metadata=TEST_METADATA,
)
project_id=GCP_PROJECT_ID,
do_xcom_push=True,
)
# [END howto_operator_get_instance]
# [START howto_operator_get_instance_result]
get_instance_result = BashOperator(
task_id="get-instance-result",
bash_command="echo \"{{ task_instance.xcom_pull('get-instance') }}\"")
# [END howto_operator_get_instance_result]
# [START howto_operator_failover_instance]
failover_instance = CloudMemorystoreFailoverInstanceOperator(
task_id="failover-instance",
location="europe-north1",
instance=INSTANCE_NAME_2,
data_protection_mode=FailoverInstanceRequest.DataProtectionMode.
LIMITED_DATA_LOSS,
project_id=GCP_PROJECT_ID,
)
# [END howto_operator_failover_instance]
# [START howto_operator_list_instances]
list_instances = CloudMemorystoreListInstancesOperator(
task_id="list-instances",
location="-",
page_size=100,
project_id=GCP_PROJECT_ID)
# [END howto_operator_list_instances]
# [START howto_operator_list_instances_result]
list_instances_result = BashOperator(
