def test_instance_db_delete_missing_project_id(self, mock_hook, _check_if_db_exists): _check_if_db_exists.return_value = True op = CloudSQLDeleteInstanceDatabaseOperator(instance=INSTANCE_NAME, database=DB_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_database.assert_called_once_with( project_id=None, instance=INSTANCE_NAME, database=DB_NAME)
def test_instance_db_delete_should_abort_and_succeed_if_not_exists( self, mock_hook, _check_if_db_exists): _check_if_db_exists.return_value = False op = CloudSQLDeleteInstanceDatabaseOperator(project_id=PROJECT_ID, instance=INSTANCE_NAME, database=DB_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_database.assert_not_called()
def test_instance_db_delete(self, mock_hook, _check_if_db_exists): _check_if_db_exists.return_value = True op = CloudSQLDeleteInstanceDatabaseOperator(project_id=PROJECT_ID, instance=INSTANCE_NAME, database=DB_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_database.assert_called_once_with( project_id=PROJECT_ID, instance=INSTANCE_NAME, database=DB_NAME)