Examples with ConnectHeaders org.apache.kafka.connect.header.ConnectHeaders used on opensource projects

Example 16 with ConnectHeaders

use of org.apache.kafka.connect.header.ConnectHeaders in project kafka by apache.

the class HeaderFromTest method schemaless.

@ParameterizedTest
@MethodSource("data")
public void schemaless(String description, boolean keyTransform, RecordBuilder originalBuilder, List<String> transformFields, List<String> headers1, HeaderFrom.Operation operation, RecordBuilder expectedBuilder) {
    HeaderFrom<SourceRecord> xform = keyTransform ? new HeaderFrom.Key<>() : new HeaderFrom.Value<>();
    xform.configure(config(headers1, transformFields, operation));
    ConnectHeaders headers = new ConnectHeaders();
    headers.addString("existing", "existing-value");
    SourceRecord originalRecord = originalBuilder.schemaless(keyTransform);
    SourceRecord expectedRecord = expectedBuilder.schemaless(keyTransform);
    SourceRecord xformed = xform.apply(originalRecord);
    assertSameRecord(expectedRecord, xformed);
}

Example 17 with ConnectHeaders

use of org.apache.kafka.connect.header.ConnectHeaders in project kafka by apache.

the class DropHeadersTest method dropNonExistingHeader.

@Test
public void dropNonExistingHeader() {
    xform.configure(config("to-drop"));
    ConnectHeaders expected = new ConnectHeaders();
    expected.addString("existing", "existing-value");
    ConnectHeaders headers = expected.duplicate();
    SourceRecord original = sourceRecord(headers);
    SourceRecord xformed = xform.apply(original);
    assertNonHeaders(original, xformed);
    assertEquals(expected, xformed.headers());
}

Example 18 with ConnectHeaders

use of org.apache.kafka.connect.header.ConnectHeaders in project kafka by apache.

the class InsertHeaderTest method insertionWithByteHeader.

@Test
public void insertionWithByteHeader() {
    xform.configure(config("inserted", "1"));
    ConnectHeaders headers = new ConnectHeaders();
    headers.addString("existing", "existing-value");
    Headers expect = headers.duplicate().addByte("inserted", (byte) 1);
    SourceRecord original = sourceRecord(headers);
    SourceRecord xformed = xform.apply(original);
    assertNonHeaders(original, xformed);
    assertEquals(expect, xformed.headers());
}

Example 19 with ConnectHeaders

use of org.apache.kafka.connect.header.ConnectHeaders in project kafka by apache.

the class DropHeadersTest method dropExistingHeader.

@Test
public void dropExistingHeader() {
    xform.configure(config("to-drop"));
    ConnectHeaders expected = new ConnectHeaders();
    expected.addString("existing", "existing-value");
    ConnectHeaders headers = expected.duplicate();
    headers.addString("to-drop", "existing-value");
    SourceRecord original = sourceRecord(headers);
    SourceRecord xformed = xform.apply(original);
    assertNonHeaders(original, xformed);
    assertEquals(expected, xformed.headers());
}

Example 20 with ConnectHeaders

use of org.apache.kafka.connect.header.ConnectHeaders in project kafka by apache.

the class DropHeadersTest method dropExistingHeaderWithMultipleValues.

@Test
public void dropExistingHeaderWithMultipleValues() {
    xform.configure(config("to-drop"));
    ConnectHeaders expected = new ConnectHeaders();
    expected.addString("existing", "existing-value");
    ConnectHeaders headers = expected.duplicate();
    headers.addString("to-drop", "existing-value");
    headers.addString("to-drop", "existing-other-value");
    SourceRecord original = sourceRecord(headers);
    SourceRecord xformed = xform.apply(original);
    assertNonHeaders(original, xformed);
    assertEquals(expected, xformed.headers());
}