Skip to content

Commit 931d514

Browse files
committed
Test if any broken test to change TestStream coder context.
1 parent 5485467 commit 931d514

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

sdks/java/core/src/main/java/org/apache/beam/sdk/util/construction/TestStreamTranslation.java

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,8 @@ static <T> RunnerApi.TestStreamPayload.Event eventToProto(
123123
.setTimestamp(element.getTimestamp().getMillis())
124124
.setEncodedElement(
125125
ByteString.copyFrom(
126-
CoderUtils.encodeToByteArray(coder, element.getValue()))));
126+
CoderUtils.encodeToByteArray(
127+
coder, element.getValue(), Coder.Context.NESTED))));
127128
}
128129
return RunnerApi.TestStreamPayload.Event.newBuilder().setElementEvent(builder).build();
129130
default:
@@ -149,7 +150,8 @@ static <T> TestStream.Event<T> eventFromProto(
149150
protoEvent.getElementEvent().getElementsList()) {
150151
decodedElements.add(
151152
TimestampedValue.of(
152-
CoderUtils.decodeFromByteArray(coder, element.getEncodedElement().toByteArray()),
153+
CoderUtils.decodeFromByteArray(
154+
coder, element.getEncodedElement().toByteArray(), Coder.Context.NESTED),
153155
new Instant(element.getTimestamp())));
154156
}
155157
return TestStream.ElementEvent.add(decodedElements);

0 commit comments

Comments
 (0)