misc: enum and int enum as event header tests

Author: 0marperez

tests/codegen/event-stream/src/test/kotlin/aws/sdk/kotlin/tests/codegen/eventstream/HttpEventStreamTests.kt

@@ -188,6 +188,8 @@ class HttpEventStreamTests {
                 long = 9_000_000_000L
                 string = "a tay is a hammer"
                 timestamp = Instant.fromEpochSeconds(5)
+                enum = aws.sdk.kotlin.test.restjson1.model.Enum.fromValue("DIAMOND")
+                intEnum = IntEnum.fromValue(1)
             },
         )
 
@@ -204,6 +206,8 @@ class HttpEventStreamTests {
         assertEquals(9_000_000_000L, headers["long"]?.expectInt64())
         assertEquals("a tay is a hammer", headers["string"]?.expectString())
         assertEquals(Instant.fromEpochSeconds(5), headers["timestamp"]?.expectTimestamp())
+        assertEquals(aws.sdk.kotlin.test.restjson1.model.Enum.fromValue("DIAMOND").value, headers["enum"]?.expectString())
+        assertEquals(IntEnum.fromValue(1).value, headers["intEnum"]?.expectInt32())
 
         val deserialized = deserializedEvent(message)
         assertIs<TestStream.MessageWithHeaders>(deserialized)

tests/codegen/event-stream/src/test/resources/event-stream-model-template.smithy

@@ -1,3 +1,5 @@
+$version: "2"
+
 namespace aws.sdk.kotlin.test
 
 use aws.protocols#restJson1
@@ -54,6 +56,8 @@ structure MessageWithHeaders {
     @eventHeader short: Short,
     @eventHeader string: String,
     @eventHeader timestamp: Timestamp,
+    @eventHeader enum: Enum,
+    @eventHeader intEnum: IntEnum,
 }
 structure MessageWithHeaderAndPayload {
     @eventHeader header: String,
@@ -81,3 +85,18 @@ union TestStream {
     MessageWithUnboundPayloadTraits: MessageWithUnboundPayloadTraits,
     SomeError: SomeError,
 }
+
+enum Enum {
+    DIAMOND
+    CLUB
+    HEART
+    SPADE
+}
+
+intEnum IntEnum {
+    JACK = 1
+    QUEEN = 2
+    KING = 3
+    ACE = 4
+    JOKER = 5
+}