Examples with KafkaMergedFlushExFW io.aklivity.zilla.specs.binding.kafka.internal.types.stream.KafkaMergedFlushExFW used on opensource projects

Example 1 with KafkaMergedFlushExFW

use of io.aklivity.zilla.specs.binding.kafka.internal.types.stream.KafkaMergedFlushExFW in project zilla by aklivity.

the class KafkaFunctionsTest method shouldGenerateMergedFlushExtension.

@Test
public void shouldGenerateMergedFlushExtension() {
    byte[] build = KafkaFunctions.flushEx().typeId(0x01).merged().progress(0, 1L).filter().key("match").build().filter().header("name", "value").build().build().build();
    DirectBuffer buffer = new UnsafeBuffer(build);
    KafkaFlushExFW flushEx = new KafkaFlushExFW().wrap(buffer, 0, buffer.capacity());
    assertEquals(0x01, flushEx.typeId());
    final KafkaMergedFlushExFW mergedFlushEx = flushEx.merged();
    final MutableInteger partitionsCount = new MutableInteger();
    mergedFlushEx.progress().forEach(f -> partitionsCount.value++);
    assertEquals(1, partitionsCount.value);
    assertNotNull(mergedFlushEx.progress().matchFirst(p -> p.partitionId() == 0 && p.partitionOffset() == 1L));
    final MutableInteger filterCount = new MutableInteger();
    mergedFlushEx.filters().forEach(f -> filterCount.value++);
    assertEquals(2, filterCount.value);
    assertNotNull(mergedFlushEx.filters().matchFirst(f -> f.conditions().matchFirst(c -> c.kind() == KEY.value() && "match".equals(c.key().value().get((b, o, m) -> b.getStringWithoutLengthUtf8(o, m - o)))) != null));
    assertNotNull(mergedFlushEx.filters().matchFirst(f -> f.conditions().matchFirst(c -> c.kind() == HEADER.value() && "name".equals(c.header().name().get((b, o, m) -> b.getStringWithoutLengthUtf8(o, m - o))) && "value".equals(c.header().value().get((b, o, m) -> b.getStringWithoutLengthUtf8(o, m - o)))) != null));
}