Force portable TestStream to use NesteCoder context throughout.

apache · Dec 9, 2024 · f2b88e9 · f2b88e9
1 parent 2344fdc
commit f2b88e9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.../java/core/src/main/java/org/apache/beam/sdk/util/construction/TestStreamTranslation.java b/.../java/core/src/main/java/org/apache/beam/sdk/util/construction/TestStreamTranslation.java
@@ -123,7 +123,7 @@ static <T> RunnerApi.TestStreamPayload.Event eventToProto(
                   .setTimestamp(element.getTimestamp().getMillis())
                   .setEncodedElement(
                       ByteString.copyFrom(
-                          CoderUtils.encodeToByteArray(coder, element.getValue()))));
+                          CoderUtils.encodeToByteArray(coder, element.getValue(), Coder.Context.NESTED))));
         }
         return RunnerApi.TestStreamPayload.Event.newBuilder().setElementEvent(builder).build();
       default:
@@ -149,7 +149,7 @@ static <T> TestStream.Event<T> eventFromProto(
             protoEvent.getElementEvent().getElementsList()) {
           decodedElements.add(
               TimestampedValue.of(
-                  CoderUtils.decodeFromByteArray(coder, element.getEncodedElement().toByteArray()),
+                  CoderUtils.decodeFromByteArray(coder, element.getEncodedElement().toByteArray(), Coder.Context.NESTED),
                   new Instant(element.getTimestamp())));
         }
         return TestStream.ElementEvent.add(decodedElements);