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,