codegen: Fix eventstream require params and test generation (#3463)

Fixes the code generation for eventstream APIs to correctly not generated required parameter validation, and fix test WriteEvents code generation.
aws · Aug 10, 2020 · 34f092c · 34f092c
1 parent 1d7c049
commit 34f092c
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 1 deletion.
diff --git a/private/model/api/api.go b/private/model/api/api.go
@@ -855,7 +855,7 @@ func resolveShapeValidations(s *Shape, ancestry ...*Shape) {
 		ref := s.MemberRefs[name]
 		// Since this is a grab bag we will just continue since
 		// we can't validate because we don't know the valued shape.
-		if ref.JSONValue {
+		if ref.JSONValue || (s.UsedAsInput && ref.Shape.IsEventStream) {
 			continue
 		}
 

diff --git a/private/model/api/eventstream_tmpl_writertests.go b/private/model/api/eventstream_tmpl_writertests.go
@@ -201,6 +201,13 @@ var eventStreamWriterTestTmpl = template.Must(
 			{{- end }}
 		}
 
+		var marshalers request.HandlerList
+		marshalers.PushBackNamed({{ $.API.ProtocolPackage }}.BuildHandler)
+		payloadMarshaler := protocol.HandlerPayloadMarshal{
+			Marshalers: marshalers,
+		}
+		_ = payloadMarshaler
+
 		eventMsgs := []eventstream.Message{
 			{{- range $idx, $event := $.InputStream.Events }}
 				{{- template "set event message" Map "idx" $idx "parentShape" $event.Shape "eventName" $event.Name }}

diff --git a/service/transcribestreamingservice/eventstream_test.go b/service/transcribestreamingservice/eventstream_test.go