def test_instance_delete_should_abort_and_succeed_if_not_exists(
self, mock_hook, _check_if_instance_exists):
_check_if_instance_exists.return_value = False
op = CloudSqlInstanceDeleteOperator(project_id=PROJECT_ID,
instance=INSTANCE_NAME,
task_id="id")
result = op.execute(None)
self.assertTrue(result)
mock_hook.assert_called_once_with(api_version="v1beta4",
gcp_conn_id="google_cloud_default")
mock_hook.return_value.delete_instance.assert_not_called()
def test_instance_delete_missing_project_id(self, mock_hook,
_check_if_instance_exists):
_check_if_instance_exists.return_value = True
op = CloudSqlInstanceDeleteOperator(instance=INSTANCE_NAME,
task_id="id")
result = op.execute(None)
self.assertTrue(result)
mock_hook.assert_called_once_with(api_version="v1beta4",
gcp_conn_id="google_cloud_default")
mock_hook.return_value.delete_instance.assert_called_once_with(
project_id=None, instance=INSTANCE_NAME)
def test_instance_delete_missing_project_id(self, mock_hook, _check_if_instance_exists):
_check_if_instance_exists.return_value = True
op = CloudSqlInstanceDeleteOperator(
instance=INSTANCE_NAME,
task_id="id"
)
result = op.execute(None)
self.assertTrue(result)
mock_hook.assert_called_once_with(api_version="v1beta4",
gcp_conn_id="google_cloud_default")
mock_hook.return_value.delete_instance.assert_called_once_with(
project_id=None,
instance=INSTANCE_NAME
)
def test_instance_delete_should_abort_and_succeed_if_not_exists(
self,
mock_hook,
_check_if_instance_exists):
_check_if_instance_exists.return_value = False
op = CloudSqlInstanceDeleteOperator(
project_id=PROJECT_ID,
instance=INSTANCE_NAME,
task_id="id"
)
result = op.execute(None)
self.assertTrue(result)
mock_hook.assert_called_once_with(api_version="v1beta4",
gcp_conn_id="google_cloud_default")
mock_hook.return_value.delete_instance.assert_not_called()
)
# [END howto_operator_cloudsql_db_create]
# [START howto_operator_cloudsql_db_patch]
sql_db_patch_task = CloudSqlInstanceDatabasePatchOperator(
project_id=PROJECT_ID,
body=db_patch_body,
instance=INSTANCE_NAME,
database=DB_NAME,
task_id='sql_db_patch_task'
)
# [END howto_operator_cloudsql_db_patch]
# [START howto_operator_cloudsql_db_delete]
sql_db_delete_task = CloudSqlInstanceDatabaseDeleteOperator(
project_id=PROJECT_ID,
instance=INSTANCE_NAME,
database=DB_NAME,
task_id='sql_db_delete_task'
)
# [END howto_operator_cloudsql_db_delete]
# [START howto_operator_cloudsql_delete]
sql_instance_delete_task = CloudSqlInstanceDeleteOperator(
project_id=PROJECT_ID,
instance=INSTANCE_NAME,
task_id='sql_instance_delete_task'
)
# [END howto_operator_cloudsql_delete]
sql_instance_create_task >> sql_instance_patch_task \
>> sql_db_create_task >> sql_db_patch_task \
>> sql_db_delete_task >> sql_instance_delete_task
task_id='sql_db_delete_task')
sql_db_delete_task2 = CloudSqlInstanceDatabaseDeleteOperator(
instance=INSTANCE_NAME,
database=DB_NAME,
task_id='sql_db_delete_task2')
# [END howto_operator_cloudsql_db_delete]
prev_task = next_dep(sql_db_delete_task, prev_task)
prev_task = next_dep(sql_db_delete_task2, prev_task)
# ############################################## #
# ### INSTANCES TEAR DOWN ###################### #
# ############################################## #
# [START howto_operator_cloudsql_replicas_delete]
sql_instance_failover_replica_delete_task = CloudSqlInstanceDeleteOperator(
project_id=GCP_PROJECT_ID,
instance=FAILOVER_REPLICA_NAME,
task_id='sql_instance_failover_replica_delete_task')
sql_instance_read_replica_delete_task = CloudSqlInstanceDeleteOperator(
project_id=GCP_PROJECT_ID,
instance=READ_REPLICA_NAME,
task_id='sql_instance_read_replica_delete_task')
# [END howto_operator_cloudsql_replicas_delete]
prev_task = next_dep(sql_instance_failover_replica_delete_task, prev_task)
prev_task = next_dep(sql_instance_read_replica_delete_task, prev_task)
# [START howto_operator_cloudsql_delete]
sql_instance_delete_task = CloudSqlInstanceDeleteOperator(
project_id=GCP_PROJECT_ID,
instance=INSTANCE_NAME,
