def test_assert_valid_hook_call(self, mock_hook):
task = CloudMemorystoreCreateInstanceAndImportOperator(
task_id=TEST_TASK_ID,
location=TEST_LOCATION,
instance_id=TEST_INSTANCE_ID,
instance=TEST_INSTANCE,
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,
)
task.execute(mock.MagicMock())
mock_hook.assert_has_calls([
mock.call(gcp_conn_id=TEST_GCP_CONN_ID),
mock.call().create_instance(
location=TEST_LOCATION,
instance_id=TEST_INSTANCE_ID,
instance=TEST_INSTANCE,
project_id=TEST_PROJECT_ID,
retry=TEST_RETRY,
timeout=TEST_TIMEOUT,
metadata=TEST_METADATA,
),
mock.call().import_instance(
input_config=TEST_INPUT_CONFIG,
instance=TEST_INSTANCE,
location=TEST_LOCATION,
metadata=TEST_METADATA,
project_id=TEST_PROJECT_ID,
retry=TEST_RETRY,
timeout=TEST_TIMEOUT,
),
])
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",
instance_id=INSTANCE_NAME_3,
instance=FIRST_INSTANCE,
input_config={"gcs_source": {
"uri": EXPORT_GCS_URL
}},
project_id=GCP_PROJECT_ID,
)
# [START howto_operator_create_instance_and_import]
# [START howto_operator_scale_instance]
scale_instance = CloudMemorystoreScaleInstanceOperator(
task_id="scale-instance",
location="europe-north1",
instance_id=INSTANCE_NAME_3,
project_id=GCP_PROJECT_ID,
memory_size_gb=3,
)
# [END howto_operator_scale_instance]
