diff --git a/sdks/java/core/src/test/java/org/apache/beam/sdk/transforms/MetadataPropagationTest.java b/sdks/java/core/src/test/java/org/apache/beam/sdk/transforms/MetadataPropagationTest.java index 5cfc3d3a1eb6..f41c3297ae7e 100644 --- a/sdks/java/core/src/test/java/org/apache/beam/sdk/transforms/MetadataPropagationTest.java +++ b/sdks/java/core/src/test/java/org/apache/beam/sdk/transforms/MetadataPropagationTest.java @@ -29,56 +29,53 @@ import org.junit.runner.RunWith; import org.junit.runners.JUnit4; +@RunWith(JUnit4.class) public class MetadataPropagationTest { - @RunWith(JUnit4.class) - public static class MiscTest { + /** Tests for metadata propagation. */ + @Rule public final transient TestPipeline pipeline = TestPipeline.create(); - /** Tests for metadata propagation. */ - @Rule public final transient TestPipeline pipeline = TestPipeline.create(); - - static class CausedByDrainSettingDoFn extends DoFn { - @ProcessElement - public void process(OutputReceiver r) { - r.builder("value").setCausedByDrain(CausedByDrain.CAUSED_BY_DRAIN).output(); - } + static class CausedByDrainSettingDoFn extends DoFn { + @ProcessElement + public void process(OutputReceiver r) { + r.builder("value").setCausedByDrain(CausedByDrain.CAUSED_BY_DRAIN).output(); } + } - static class CausedByDrainExtractingDoFn extends DoFn { - @ProcessElement - public void process(ProcessContext pc, OutputReceiver r) { - r.output(pc.causedByDrain().toString()); - } + static class CausedByDrainExtractingDoFn extends DoFn { + @ProcessElement + public void process(ProcessContext pc, OutputReceiver r) { + r.output(pc.causedByDrain().toString()); } + } - @Test - @Category(NeedsRunner.class) - public void testMetadataPropagationAcrossShuffleParameter() { - WindowedValues.WindowedValueCoder.setMetadataSupported(); - PCollection results = - pipeline - .apply(Create.of(1)) - .apply(ParDo.of(new CausedByDrainSettingDoFn())) - .apply(Redistribute.arbitrarily()) - .apply(ParDo.of(new CausedByDrainExtractingDoFn())); + @Test + @Category(NeedsRunner.class) + public void testMetadataPropagationAcrossShuffleParameter() { + WindowedValues.WindowedValueCoder.setMetadataSupported(); + PCollection results = + pipeline + .apply(Create.of(1)) + .apply(ParDo.of(new CausedByDrainSettingDoFn())) + .apply(Redistribute.arbitrarily()) + .apply(ParDo.of(new CausedByDrainExtractingDoFn())); - PAssert.that(results).containsInAnyOrder("CAUSED_BY_DRAIN"); + PAssert.that(results).containsInAnyOrder("CAUSED_BY_DRAIN"); - pipeline.run(); - } + pipeline.run(); + } - @Test - @Category(NeedsRunner.class) - public void testMetadataPropagationParameter() { - PCollection results = - pipeline - .apply(Create.of(1)) - .apply(ParDo.of(new CausedByDrainSettingDoFn())) - .apply(ParDo.of(new CausedByDrainExtractingDoFn())); + @Test + @Category(NeedsRunner.class) + public void testMetadataPropagationParameter() { + PCollection results = + pipeline + .apply(Create.of(1)) + .apply(ParDo.of(new CausedByDrainSettingDoFn())) + .apply(ParDo.of(new CausedByDrainExtractingDoFn())); - PAssert.that(results).containsInAnyOrder("CAUSED_BY_DRAIN"); + PAssert.that(results).containsInAnyOrder("CAUSED_BY_DRAIN"); - pipeline.run(); - } + pipeline.run(); } }