Skip to content

Commit 0efa79b

Browse files
committed
fix: criteria for event stream member detection
1 parent 0fa94c2 commit 0efa79b

File tree

2 files changed

+2
-10
lines changed

2 files changed

+2
-10
lines changed

smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/integration/EventStreamGenerator.java

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -90,9 +90,7 @@ public static MemberShape getEventStreamMember(GenerationContext context, Struct
9090
Shape target = context.getModel().expectShape(shape.getTarget());
9191
boolean targetStreaming = target.hasTrait(StreamingTrait.class);
9292
boolean targetUnion = target.isUnionShape();
93-
boolean memberStreaming = shape.hasTrait(StreamingTrait.class);
94-
boolean memberPayload = shape.hasTrait(HttpPayloadTrait.class);
95-
return memberPayload && targetUnion && (targetStreaming || memberStreaming);
93+
return targetUnion && targetStreaming;
9694
}).toList();
9795

9896
if (eventStreamMembers.isEmpty()) {
@@ -454,7 +452,7 @@ private void generateEventStreamDeserializer(GenerationContext context, UnionSha
454452
});
455453
});
456454
});
457-
writer.write("return {$$unknown: output};");
455+
writer.write("return {$$unknown: event};");
458456
});
459457
});
460458
});

smithy-typescript-codegen/src/test/java/software/amazon/smithy/typescript/codegen/integration/EventStreamGeneratorTest.java

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -35,8 +35,6 @@ void getEventStreamMember(
3535

3636
when(streamingTarget1.hasTrait(StreamingTrait.class)).thenReturn(true);
3737
when(streamingTarget1.isUnionShape()).thenReturn(true);
38-
when(eventStreamMember1.hasTrait(StreamingTrait.class)).thenReturn(false);
39-
when(eventStreamMember1.hasTrait(HttpPayloadTrait.class)).thenReturn(true);
4038

4139
MemberShape eventStreamMember = EventStreamGenerator.getEventStreamMember(
4240
context,
@@ -90,15 +88,11 @@ void getEventStreamMemberTooMany(
9088
when(model.expectShape(streamingMember1ShapeId)).thenReturn(streamingTarget1);
9189
when(streamingTarget1.hasTrait(StreamingTrait.class)).thenReturn(true);
9290
when(streamingTarget1.isUnionShape()).thenReturn(true);
93-
when(eventStreamMember1.hasTrait(StreamingTrait.class)).thenReturn(false);
94-
when(eventStreamMember1.hasTrait(HttpPayloadTrait.class)).thenReturn(true);
9591

9692
when(eventStreamMember2.getTarget()).thenReturn(streamingMember2ShapeId);
9793
when(model.expectShape(streamingMember2ShapeId)).thenReturn(streamingTarget2);
9894
when(streamingTarget2.hasTrait(StreamingTrait.class)).thenReturn(true);
9995
when(streamingTarget2.isUnionShape()).thenReturn(true);
100-
when(eventStreamMember2.hasTrait(StreamingTrait.class)).thenReturn(false);
101-
when(eventStreamMember2.hasTrait(HttpPayloadTrait.class)).thenReturn(true);
10296

10397
try {
10498
MemberShape eventStreamMember = EventStreamGenerator.getEventStreamMember(

0 commit comments

Comments
 (0)