Fix e2e tests

Author: exanubes

authorizer/authorizer.go

@@ -9,7 +9,7 @@ type Authorizer interface {
 
 type AuthorizeCommandInput struct {
 	Channel string
-	Payload []byte
+	Payload [][]byte
 }
 
 type AuthorizeCommandOutput struct {

authorizer/iam.go

@@ -48,7 +48,10 @@ func (authorizer *iam_authorizer) Authorize(ctx context.Context, input Authorize
 	}
 
 	if input.Payload != nil {
-		canonical.Payload = []string{string(input.Payload)}
+		canonical.Payload = make([]string, len(input.Payload))
+		for index, payload := range input.Payload {
+			canonical.Payload[index] = string(payload)
+		}
 	}
 
 	payload, err := json.Marshal(canonical)

client.go

@@ -30,10 +30,15 @@ func (client *appsync_client) Publish(ctx context.Context, input PublishCommandI
 		authz = authorizer.NewInternalAdapter(input.Authorizer)
 	}
 	var output PublishCommandOutput
+	payload := make([]app.Payload, len(input.Events))
+
+	for index, event := range input.Events {
+		payload[index] = app.Payload(event)
+	}
 
 	result, err := client.usecases.Publish.Publish(ctx, publish.PublishCommandInput{
 		Destination: input.Channel,
-		Payload:     input.Payload,
+		Events:      payload,
 		Frame:       events.Publish(),
 		Authorizer:  authz,
 	})

e2e/e2e_test.go

@@ -126,9 +126,9 @@ func TestAppSyncAuthorizers(t *testing.T) {
 				}
 			}()
 
-			if err := client.Publish(ctx, appsync.PublishCommandInput{
+			if _, err := client.Publish(ctx, appsync.PublishCommandInput{
 				Channel: channel,
-				Payload: payload,
+				Events:  [][]byte{payload},
 			}); err != nil {
 				t.Fatalf("publish: %v", err)
 			}
@@ -284,9 +284,9 @@ func TestPerRequestAuthorizerOverride(t *testing.T) {
 		}
 	}()
 
-	if err := client.Publish(ctx, appsync.PublishCommandInput{
+	if _, err := client.Publish(ctx, appsync.PublishCommandInput{
 		Channel:    channel,
-		Payload:    payload,
+		Events:     [][]byte{payload},
 		Authorizer: iam,
 	}); err != nil {
 		t.Fatalf("publish: %v", err)
@@ -356,9 +356,9 @@ func TestConnectionSpecificAuthorizers(t *testing.T) {
 		}
 	}()
 
-	if err := client.Publish(ctx, appsync.PublishCommandInput{
+	if _, err := client.Publish(ctx, appsync.PublishCommandInput{
 		Channel: channel,
-		Payload: payload,
+		Events:  [][]byte{payload},
 	}); err != nil {
 		t.Fatalf("publish: %v", err)
 	}
@@ -435,9 +435,9 @@ func TestMixedConnectionLevelAuthorizers(t *testing.T) {
 		}
 	}()
 
-	if err := client.Publish(ctx, appsync.PublishCommandInput{
+	if _, err := client.Publish(ctx, appsync.PublishCommandInput{
 		Channel: channel,
-		Payload: payload,
+		Events:  [][]byte{payload},
 	}); err != nil {
 		t.Fatalf("publish: %v", err)
 	}

internal/infrastructure/authorizer/public-adapter.go

@@ -18,9 +18,14 @@ func NewInternalAdapter(public pub.Authorizer) *InternalAuthorizerAdapter {
 }
 
 func (authorizer *InternalAuthorizerAdapter) Authorize(ctx context.Context, input app.AuthorizeCommandInput) (app.Signature, error) {
+	payload := make([][]byte, len(input.Payload))
+	for index := range input.Payload {
+		payload[index] = []byte(input.Payload[index])
+	}
+
 	output, err := authorizer.public.Authorize(ctx, pub.AuthorizeCommandInput{
 		Channel: input.Channel,
-		Payload: input.Payload,
+		Payload: payload,
 	})
 
 	if err != nil {

internal/infrastructure/events/builder.go

@@ -71,11 +71,18 @@ func (builder *FrameBuilder) Build() app.Frame {
 	} else {
 		id = generate_id()
 	}
+
+	payload := make([]string, len(builder.batch))
+
+	for index, event := range builder.batch {
+		payload[index] = string(event)
+	}
+
 	return Frame{
 		Id:        id,
 		Topic:     builder.channel,
 		Signature: builder.signature,
-		Payload:   []string{string(builder.payload)},
+		Payload:   payload,
 		Type:      builder._type,
 	}
 }

ports.go

@@ -63,6 +63,7 @@ type Backpressure struct {
 type PublishCommandInput struct {
 	Channel    string
 	Payload    []byte
+	Events     [][]byte
 	Authorizer authorizer.Authorizer
 }