aromanenko-dev commented on a change in pull request #13282:
URL: https://github.com/apache/beam/pull/13282#discussion_r522185225



##########
File path: 
sdks/java/io/kafka/src/test/java/org/apache/beam/sdk/io/kafka/KafkaIOIT.java
##########
@@ -116,15 +117,54 @@ public static void setup() throws IOException {
   }
 
   @Test
-  public void testKafkaIOReadsAndWritesCorrectly() throws IOException {
+  public void testKafkaIOReadsAndWritesCorrectlyInStreaming() throws 
IOException {
+    // Use batch pipeline to write records.
+    writePipeline
+        .apply("Generate records", Read.from(new 
SyntheticBoundedSource(sourceOptions)))
+        .apply("Measure write time", ParDo.of(new TimeMonitor<>(NAMESPACE, 
WRITE_TIME_METRIC_NAME)))
+        .apply("Write to Kafka", writeToKafka());
+
+    // Use streaming pipeline to read Kafka records.
+    readPipeline.getOptions().as(Options.class).setStreaming(true);
+    readPipeline
+        .apply("Read from Runner V2 Kafka", readFromKafka())

Review comment:
       nit: Please, remove "Runner V2"




----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

For queries about this service, please contact Infrastructure at:
[email protected]


Reply via email to