Examples of info.archinnov.achilles.internal.statement.wrapper.BatchStatementWrapper


    protected ListenableFuture<ResultSet> executeBatch(BatchStatement.Type batchType, List<AbstractStatementWrapper> statementWrappers) {
        ListenableFuture<ResultSet> resultSetFuture = null;
        final int size = statementWrappers.size();
        if (size > 1) {
            final BatchStatementWrapper batchStatementWrapper = new BatchStatementWrapper(batchType, statementWrappers, fromNullable(consistencyLevel),serialConsistencyLevel);
            resultSetFuture = daoContext.execute(batchStatementWrapper);
        } else if (size == 1) {
            AbstractStatementWrapper wrapper;
            if (batchType == BatchStatement.Type.LOGGED) {
                wrapper = new BatchStatementWrapper(batchType, statementWrappers, fromNullable(consistencyLevel),serialConsistencyLevel);
            } else {
                wrapper = statementWrappers.get(0);
            }
            resultSetFuture = daoContext.execute(wrapper);
        }

        //Then
        assertThat(futureResultSets).isSameAs(achillesEmpty);
        assertThat(futureResultSetsCaptor.getValue()).containsExactly(futureResultSet1, futureResultSet2);


        final List<AbstractStatementWrapper> wrappers = statementWrapperCaptor.getAllValues();
        BatchStatementWrapper batchWrapper = (BatchStatementWrapper) wrappers.get(0);
        assertThat(Whitebox.<List<AbstractStatementWrapper>>getInternalState(batchWrapper, "statementWrappers")).containsExactly(wrapper1, wrapper2);
        assertThat(wrappers.get(1)).isSameAs(wrapper3);


        assertThat(batchWrapper.getStatement().getConsistencyLevel()).isSameAs(com.datastax.driver.core.ConsistencyLevel.LOCAL_QUORUM);
        assertThat(batchWrapper.getStatement().getSerialConsistencyLevel()).isNull();


        final Function<List<ResultSet>, Empty> applyTriggers = applyTriggersCaptor.getValue();
        applyTriggers.apply(asList(mock(ResultSet.class)));


        verify(eventHolder).triggerInterception();

Examples of info.archinnov.achilles.internal.statement.wrapper.BatchStatementWrapper

Related Classes of info.archinnov.achilles.internal.statement.wrapper.BatchStatementWrapper