/*
* Copyright 2011 LMAX Ltd., modified by Jamie Allen to use Scala port.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.lmax.disruptor;
import static org.hamcrest.core.Is.is;
import static org.junit.Assert.assertThat;
import org.junit.Test;
import com.jamieallen.sdisruptor.Consumer;
import com.jamieallen.sdisruptor.ConsumerBarrier;
import com.jamieallen.sdisruptor.NoOpConsumer;
import com.jamieallen.sdisruptor.RingBuffer;
import com.jamieallen.sdisruptor.SequenceBatch;
import com.lmax.disruptor.support.StubEntry;
public final class BatchProducerTest
{
private final RingBuffer<StubEntry> ringBuffer = new RingBuffer<StubEntry>(StubEntry.ENTRY_FACTORY, 20, null, null);
private final ConsumerBarrier<StubEntry> consumerBarrier = ringBuffer.createConsumerBarrier(new Consumer[0]);
{
final NoOpConsumer<StubEntry> noOpConsumer = new NoOpConsumer<StubEntry>(ringBuffer);
final NoOpConsumer[] noOpConsumers = new NoOpConsumer[] { noOpConsumer };
ringBuffer.consumersToTrack_(noOpConsumers);
}
@Test
public void shouldClaimBatchAndCommitBack() throws Exception
{
final int batchSize = 5;
final SequenceBatch sequenceBatch = new SequenceBatch(batchSize);
ringBuffer.nextEntries(sequenceBatch);
assertThat(Long.valueOf(sequenceBatch.getStart()), is(Long.valueOf(0L)));
assertThat(Long.valueOf(sequenceBatch.end()), is(Long.valueOf(4L)));
assertThat(Long.valueOf(ringBuffer.cursor()), is(Long.valueOf(-1L)));
ringBuffer.commit(sequenceBatch);
assertThat(Long.valueOf(ringBuffer.cursor()), is(Long.valueOf(batchSize - 1L)));
assertThat(Long.valueOf(consumerBarrier.waitFor(0L)), is(Long.valueOf(batchSize - 1L)));
}
}