Adding a test case that verifies runner-managed sharding for DirectRu…

…nner
pabloem · Aug 31, 2023 · 924a325 · 924a325
1 parent 913f000
commit 924a325
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/sdks/java/core/src/test/java/org/apache/beam/sdk/io/TextIOWriteTest.java b/sdks/java/core/src/test/java/org/apache/beam/sdk/io/TextIOWriteTest.java
@@ -727,6 +727,22 @@ public void testWindowedWritesWithOnceTrigger() throws Throwable {
         .apply(Values.create());
   }
 
+  @Test
+  @Category(NeedsRunner.class)
+  public void testWindowedWritesWithStreamingRunnerDeterminedSharding() throws Throwable {
+    PCollection<String> data =
+        p.apply(
+                TestStream.create(StringUtf8Coder.of())
+                    .addElements("0", "1", "2")
+                    .advanceWatermarkToInfinity())
+            .apply(Window.<String>into(FixedWindows.of(Duration.standardSeconds(1))));
+    data.apply(
+        TextIO.write()
+            .to(new File(tempFolder.getRoot(), "windowed-writes-noshards").getAbsolutePath())
+            .withWindowedWrites());
+    p.run().waitUntilFinish();
+  }
+
   @Test
   @Category(NeedsRunner.class)
   public void testWriteViaSink() throws Exception {