Merge branch 'main' into stylecheck

Author: bmoffatt

events/appsync.go

@@ -38,3 +38,28 @@ const (
 	// OperationBatchInvoke instructs AWS AppSync to batch requests for the current GraphQL field
 	OperationBatchInvoke AppSyncOperation = "BatchInvoke"
 )
+
+// AppSyncLambdaAuthorizerRequest contains an authorization request from AppSync.
+type AppSyncLambdaAuthorizerRequest struct {
+	AuthorizationToken string                                `json:"authorizationToken"`
+	RequestContext     AppSyncLambdaAuthorizerRequestContext `json:"requestContext"`
+}
+
+// AppSyncLambdaAuthorizerRequestContext contains the parameters of the AppSync invocation which triggered
+// this authorization request.
+type AppSyncLambdaAuthorizerRequestContext struct {
+	APIID         string                 `json:"apiId"`
+	AccountID     string                 `json:"accountId"`
+	RequestID     string                 `json:"requestId"`
+	QueryString   string                 `json:"queryString"`
+	OperationName string                 `json:"operationName"`
+	Variables     map[string]interface{} `json:"variables"`
+}
+
+// AppSyncLambdaAuthorizerResponse represents the expected format of an authorization response to AppSync.
+type AppSyncLambdaAuthorizerResponse struct {
+	IsAuthorized    bool                   `json:"isAuthorized"`
+	ResolverContext map[string]interface{} `json:"resolverContext,omitempty"`
+	DeniedFields    []string               `json:"deniedFields,omitempty"`
+	TTLOverride     *int                   `json:"ttlOverride,omitempty"`
+}

events/appsync_test.go

@@ -5,6 +5,7 @@ import (
 	"io/ioutil"
 	"testing"
 
+	"github.com/aws/aws-lambda-go/events/test"
 	"github.com/stretchr/testify/assert"
 )
 
@@ -85,3 +86,49 @@ func TestAppSyncIdentity_Cognito(t *testing.T) {
 
 	assert.JSONEq(t, string(inputJSON), string(outputJSON))
 }
+
+func TestAppSyncLambdaAuthorizerRequestMarshalling(t *testing.T) {
+	inputJSON, err := ioutil.ReadFile("./testdata/appsync-lambda-auth-request.json")
+	if err != nil {
+		t.Errorf("could not open test file. details: %v", err)
+	}
+
+	var inputEvent AppSyncLambdaAuthorizerRequest
+	if err := json.Unmarshal(inputJSON, &inputEvent); err != nil {
+		t.Errorf("could not unmarshal event. details: %v", err)
+	}
+
+	outputJSON, err := json.Marshal(inputEvent)
+	if err != nil {
+		t.Errorf("could not marshal event. details: %v", err)
+	}
+
+	assert.JSONEq(t, string(inputJSON), string(outputJSON))
+}
+
+func TestAppSyncLambdaAuthorizerRequestMalformedJson(t *testing.T) {
+	test.TestMalformedJson(t, AppSyncLambdaAuthorizerRequest{})
+}
+
+func TestAppSyncLambdaAuthorizerResponseMarshalling(t *testing.T) {
+	inputJSON, err := ioutil.ReadFile("./testdata/appsync-lambda-auth-response.json")
+	if err != nil {
+		t.Errorf("could not open test file. details: %v", err)
+	}
+
+	var inputEvent AppSyncLambdaAuthorizerResponse
+	if err := json.Unmarshal(inputJSON, &inputEvent); err != nil {
+		t.Errorf("could not unmarshal event. details: %v", err)
+	}
+
+	outputJSON, err := json.Marshal(inputEvent)
+	if err != nil {
+		t.Errorf("could not marshal event. details: %v", err)
+	}
+
+	assert.JSONEq(t, string(inputJSON), string(outputJSON))
+}
+
+func TestAppSyncLambdaAuthorizerResponseMalformedJson(t *testing.T) {
+	test.TestMalformedJson(t, AppSyncLambdaAuthorizerResponse{})
+}

events/testdata/appsync-lambda-auth-request.json

@@ -0,0 +1,11 @@
+{
+    "authorizationToken": "ExampleAUTHtoken123123123",
+    "requestContext": {
+        "apiId": "aaaaaa123123123example123",
+        "accountId": "111122223333",
+        "requestId": "f4081827-1111-4444-5555-5cf4695f339f",
+        "queryString": "mutation CreateEvent {...}\n\nquery MyQuery {...}\n",
+        "operationName": "MyQuery",
+        "variables": {}
+    }
+}

events/testdata/appsync-lambda-auth-response.json

@@ -0,0 +1,7 @@
+{
+    "isAuthorized": true,
+    "resolverContext": {
+        "banana": "very yellow",
+        "apple": "very green"
+    }
+}