Examples with BatchElementCounter com.google.api.gax.rpc.Batch.BatchElementCounter used on opensource projects

Example 1 with BatchElementCounter

use of com.google.api.gax.rpc.Batch.BatchElementCounter in project gax-java by googleapis.

the class BatchTest method testBatchElementCounter.

@Test
public void testBatchElementCounter() {
    ElementCounter<Batch<LabeledIntList, List<Integer>>> counter = new BatchElementCounter<>(SQUARER_BATCHING_DESC);
    LabeledIntList request = new LabeledIntList("lable", 3);
    Batch<LabeledIntList, List<Integer>> batch = createBatch(request);
    Truth.assertThat(counter.count(batch)).isEqualTo(1);
}