boolean wasSuccessful(GetQueryExecutionResponse getQueryExecutionResponse) {
        QueryExecutionState state = getQueryExecutionResponse.queryExecution().status().state();
        return state == QueryExecutionState.SUCCEEDED;
    }
@Test
    public void wasSuccessful() {
        Athena2QueryHelper helper = defaultAthena2QueryHelper();

        assertTrue(helper.wasSuccessful(newGetQueryExecutionResponse(QueryExecutionState.SUCCEEDED)));

        assertFalse(helper.wasSuccessful(newGetQueryExecutionResponse(QueryExecutionState.FAILED)));
        assertFalse(helper.wasSuccessful(newGetQueryExecutionResponse(QueryExecutionState.CANCELLED)));
        assertFalse(helper.wasSuccessful(newGetQueryExecutionResponse(QueryExecutionState.UNKNOWN_TO_SDK_VERSION)));
        assertFalse(helper.wasSuccessful(newGetQueryExecutionResponse(QueryExecutionState.QUEUED)));
        assertFalse(helper.wasSuccessful(newGetQueryExecutionResponse(QueryExecutionState.RUNNING)));
    }