reformatting streaming benchmark

Author: Rachel Gagnon

Diff for: examples/dataflow-examples/java/dataflow-streaming-benchmark/src/main/java/com/google/cloud/pso/pipeline/StreamingBenchmark.java

@@ -36,7 +36,6 @@
 import java.util.ArrayList;
 import java.util.List;
 import java.util.Map;
-
 import org.apache.beam.sdk.Pipeline;
 import org.apache.beam.sdk.PipelineResult;
 import org.apache.beam.sdk.io.FileSystems;
@@ -52,8 +51,8 @@
 import org.apache.beam.sdk.transforms.DoFn;
 import org.apache.beam.sdk.transforms.ParDo;
 import org.joda.time.Duration;
-import org.slf4j.LoggerFactory;
 import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 
 /**
  * The {@link StreamingBenchmark} is a streaming pipeline which generates messages at a specified

Diff for: examples/dataflow-examples/java/dataflow-streaming-benchmark/src/test/java/com/google/cloud/pso/pipeline/StreamingBenchmarkTest.java

@@ -21,8 +21,8 @@
 import static org.hamcrest.CoreMatchers.notNullValue;
 import static org.hamcrest.MatcherAssert.assertThat;
 
-import com.google.cloud.pso.pipeline.StreamingBenchmark.MessageGeneratorFn;
 import com.google.cloud.pso.pipeline.StreamingBenchmark.MalformedSchemaException;
+import com.google.cloud.pso.pipeline.StreamingBenchmark.MessageGeneratorFn;
 import com.google.common.io.ByteStreams;
 import java.io.ByteArrayInputStream;
 import java.io.File;
@@ -119,7 +119,8 @@ public void testMessageGenerator() throws IOException {
     PCollection<PubsubMessage> results =
         pipeline
             .apply("CreateInput", Create.of(0L))
-            .apply("GenerateMessage", ParDo.of(new MessageGeneratorFn(file.getAbsolutePath(), true)));
+            .apply(
+                "GenerateMessage", ParDo.of(new MessageGeneratorFn(file.getAbsolutePath(), true)));
 
     // Assert
     //
@@ -153,12 +154,16 @@ public void testInvalidSchemaThrowsException() throws IOException {
     PCollection<PubsubMessage> results =
         pipeline
             .apply("CreateInput", Create.of(0L))
-            .apply("GenerateMessage", ParDo.of(new MessageGeneratorFn(file.getAbsolutePath(), true)));
+            .apply(
+                "GenerateMessage", ParDo.of(new MessageGeneratorFn(file.getAbsolutePath(), true)));
 
     pipeline.run();
   }
 
-  /** Tests the {@link MessageGeneratorFn} should not fails when given invalid schema with validateSchema set to false. */
+  /**
+   * Tests the {@link MessageGeneratorFn} should not fails when given invalid schema with
+   * validateSchema set to false.
+   */
   @Test
   public void testInvalidSchemaIgnoringValidation() throws IOException {
     // Arrange
@@ -171,21 +176,25 @@ public void testInvalidSchemaIgnoringValidation() throws IOException {
     // Act
     //
     PCollection<PubsubMessage> results =
-            pipeline
-                    .apply("CreateInput", Create.of(0L))
-                    .apply("GenerateMessage", ParDo.of(new MessageGeneratorFn(file.getAbsolutePath(), false)));
+        pipeline
+            .apply("CreateInput", Create.of(0L))
+            .apply(
+                "GenerateMessage", ParDo.of(new MessageGeneratorFn(file.getAbsolutePath(), false)));
 
     // Assert
     //
-    PAssert.that(results).satisfies(input -> {
-      PubsubMessage message = input.iterator().next();
-      assertThat(message, is(notNullValue()));
-      assertThat(new String(message.getPayload()), is(equalTo(schema)));
-      return null;
-    });
+    PAssert.that(results)
+        .satisfies(
+            input -> {
+              PubsubMessage message = input.iterator().next();
+              assertThat(message, is(notNullValue()));
+              assertThat(new String(message.getPayload()), is(equalTo(schema)));
+              return null;
+            });
 
     pipeline.run();
   }
+
   /**
    * Helper to generate files for testing.
    *