def test_assert_valid_hook_call(self, mock_hook): task = CloudMemorystoreDeleteInstanceOperator( task_id=TEST_TASK_ID, location=TEST_LOCATION, instance=TEST_INSTANCE_NAME, 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(mock.MagicMock()) mock_hook.assert_called_once_with( gcp_conn_id=TEST_GCP_CONN_ID, impersonation_chain=TEST_IMPERSONATION_CHAIN, ) mock_hook.return_value.delete_instance.assert_called_once_with( location=TEST_LOCATION, instance=TEST_INSTANCE_NAME, project_id=TEST_PROJECT_ID, retry=TEST_RETRY, timeout=TEST_TIMEOUT, metadata=TEST_METADATA, )
# [START howto_operator_import_instance] import_instance = CloudMemorystoreImportOperator( task_id="import-instance", location="europe-north1", instance=INSTANCE_NAME_2, input_config={"gcs_source": { "uri": EXPORT_GCS_URL }}, project_id=GCP_PROJECT_ID, ) # [END howto_operator_import_instance] # [START howto_operator_delete_instance] delete_instance = CloudMemorystoreDeleteInstanceOperator( task_id="delete-instance", location="europe-north1", instance=INSTANCE_NAME, project_id=GCP_PROJECT_ID) # [END howto_operator_delete_instance] delete_instance_2 = CloudMemorystoreDeleteInstanceOperator( task_id="delete-instance-2", location="europe-north1", instance=INSTANCE_NAME_2, project_id=GCP_PROJECT_ID, ) # [END howto_operator_create_instance_and_import] create_instance_and_import = CloudMemorystoreCreateInstanceAndImportOperator( task_id="create-instance-and-import", location="europe-north1",