def test_assert_valid_hook_call(self, mock_hook): task = CloudMemorystoreImportOperator( task_id=TEST_TASK_ID, location=TEST_LOCATION, instance=TEST_INSTANCE_NAME, input_config=TEST_INPUT_CONFIG, 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.import_instance.assert_called_once_with( location=TEST_LOCATION, instance=TEST_INSTANCE_NAME, input_config=TEST_INPUT_CONFIG, project_id=TEST_PROJECT_ID, retry=TEST_RETRY, timeout=TEST_TIMEOUT, metadata=TEST_METADATA, )
task_id="export-instance", location="europe-north1", instance=INSTANCE_NAME, output_config={"gcs_destination": { "uri": EXPORT_GCS_URL }}, project_id=GCP_PROJECT_ID, ) # [END howto_operator_export_instance] # [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(