From 980c8fe15d43ea34f2d48ca02abc9043ebcde83e Mon Sep 17 00:00:00 2001
From: Shunping Huang <shunping@google.com>
Date: Mon, 6 Oct 2025 15:09:48 -0400
Subject: [PATCH] Test if any broken test to change TestStream coder context.

---
 .../beam/sdk/util/construction/TestStreamTranslation.java   | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/sdks/java/core/src/main/java/org/apache/beam/sdk/util/construction/TestStreamTranslation.java b/sdks/java/core/src/main/java/org/apache/beam/sdk/util/construction/TestStreamTranslation.java
index db1a2f875c90..9de85778e3a4 100644
--- a/sdks/java/core/src/main/java/org/apache/beam/sdk/util/construction/TestStreamTranslation.java
+++ b/sdks/java/core/src/main/java/org/apache/beam/sdk/util/construction/TestStreamTranslation.java
@@ -123,7 +123,8 @@ 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 +150,8 @@ 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);