Ejemplo n.º 1
0
 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 = CloudSQLDeleteInstanceOperator(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()
Ejemplo n.º 2
0
 def test_instance_delete_missing_project_id(self, mock_hook,
                                             _check_if_instance_exists):
     _check_if_instance_exists.return_value = True
     op = CloudSQLDeleteInstanceOperator(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)
Ejemplo n.º 3
0
 def test_instance_delete(self, mock_hook, _check_if_instance_exists):
     _check_if_instance_exists.return_value = True
     op = CloudSQLDeleteInstanceOperator(project_id=PROJECT_ID,
                                         instance=INSTANCE_NAME,
                                         task_id="id")
     result = op.execute(None)
     assert result
     mock_hook.assert_called_once_with(
         api_version="v1beta4",
         gcp_conn_id="google_cloud_default",
         impersonation_chain=None,
     )
     mock_hook.return_value.delete_instance.assert_called_once_with(
         project_id=PROJECT_ID, instance=INSTANCE_NAME)
Ejemplo n.º 4
0
        instance=INSTANCE_NAME,
        database=DB_NAME,
        task_id='sql_db_delete_task')
    sql_db_delete_task2 = CloudSQLDeleteInstanceDatabaseOperator(
        instance=INSTANCE_NAME,
        database=DB_NAME,
        task_id='sql_db_delete_task2')
    # [END howto_operator_cloudsql_db_delete]

    # ############################################## #
    # ### INSTANCES TEAR DOWN ###################### #
    # ############################################## #

    # [START howto_operator_cloudsql_replicas_delete]
    sql_instance_failover_replica_delete_task = CloudSQLDeleteInstanceOperator(
        project_id=GCP_PROJECT_ID,
        instance=FAILOVER_REPLICA_NAME,
        task_id='sql_instance_failover_replica_delete_task')

    sql_instance_read_replica_delete_task = CloudSQLDeleteInstanceOperator(
        project_id=GCP_PROJECT_ID,
        instance=READ_REPLICA_NAME,
        task_id='sql_instance_read_replica_delete_task')
    # [END howto_operator_cloudsql_replicas_delete]

    # [START howto_operator_cloudsql_delete]
    sql_instance_delete_task = CloudSQLDeleteInstanceOperator(
        project_id=GCP_PROJECT_ID,
        instance=INSTANCE_NAME,
        task_id='sql_instance_delete_task')
    sql_instance_delete_task2 = CloudSQLDeleteInstanceOperator(
        instance=INSTANCE_NAME2, task_id='sql_instance_delete_task2')