private static final Schema INPUT =
Schema.recordOf("input",
Schema.Field.of("a", Schema.of(Schema.Type.STRING)),
Schema.Field.of("b", Schema.of(Schema.Type.STRING)),
Schema.Field.of("c", Schema.of(Schema.Type.STRING)),
Schema.Field.of("d", Schema.of(Schema.Type.STRING)),
Schema.Field.of("e", Schema.of(Schema.Type.STRING))
);
@Test
public void testMyTransform() throws Exception {
ExampleTransformPlugin.Config config
= new ExampleTransformPlugin.Config("SomeValue", null, INPUT.toString());
Transform<StructuredRecord, StructuredRecord> transform
= new ExampleTransformPlugin(config);
transform.initialize(null);
MockEmitter<StructuredRecord> emitter = new MockEmitter<>();
transform.transform(StructuredRecord.builder(INPUT)
.set("a", "1")
.set("b", "2")
.set("c", "3")
.set("d", "4")
.set("e", "5").build(), emitter);
Assert.assertEquals("1", emitter.getEmitted().get(0).get("a"));
Assert.assertEquals("2", emitter.getEmitted().get(0).get("b"));
Assert.assertEquals("3", emitter.getEmitted().get(0).get("c"));
Assert.assertEquals("4", emitter.getEmitted().get(0).get("d"));
Assert.assertEquals("5", emitter.getEmitted().get(0).get("e"));
} |