Adding structures for iot custom authorizer request/response

Author: ynori7

events/apigw.go

@@ -115,9 +115,3 @@ type APIGatewayCustomAuthorizerPolicy struct {
 	Version   string
 	Statement []IAMPolicyStatement
 }
-
-type IAMPolicyStatement struct {
-	Action   []string
-	Effect   string
-	Resource []string
-}

events/iam.go

@@ -0,0 +1,7 @@
+package events
+
+type IAMPolicyStatement struct {
+	Action   []string
+	Effect   string
+	Resource []string
+}

events/iot.go

@@ -0,0 +1,23 @@
+package events
+
+// IoTCustomAuthorizerRequest contains data coming in to a custom IoT device gateway authorizer function.
+type IoTCustomAuthorizerRequest struct {
+	AuthorizationToken string `json:"token"`
+}
+
+// IoTCustomAuthorizerResponse represents the expected format of an IoT device gateway authorization response.
+type IoTCustomAuthorizerResponse struct {
+	IsAuthenticated          bool                        `json:"isAuthenticated"`
+	PrincipalID              string                      `json:"principalId"`
+	DisconnectAfterInSeconds int32                       `json:"disconnectAfterInSeconds"`
+	RefreshAfterInSeconds    int32                       `json:"refreshAfterInSeconds"`
+	PolicyDocuments          []IoTCustomAuthorizerPolicy `json:"policyDocuments"`
+	Context                  map[string]interface{}      `json:"context,omitempty"`
+}
+
+// IoTCustomAuthorizerPolicy represents an IAM policy
+type IoTCustomAuthorizerPolicy struct {
+	Version   string
+	Statement []IAMPolicyStatement
+}
+

events/iot_test.go

@@ -0,0 +1,63 @@
+package events
+
+import (
+	"encoding/json"
+	"io/ioutil"
+	"testing"
+
+	"github.com/aws/aws-lambda-go/events/test"
+)
+
+func TestIoTCustomAuthorizerRequestMarshaling(t *testing.T) {
+
+	// read json from file
+	inputJSON, err := ioutil.ReadFile("./testdata/iot-custom-auth-request.json")
+	if err != nil {
+		t.Errorf("could not open test file. details: %v", err)
+	}
+
+	// de-serialize into Go object
+	var inputEvent IoTCustomAuthorizerRequest
+	if err := json.Unmarshal(inputJSON, &inputEvent); err != nil {
+		t.Errorf("could not unmarshal event. details: %v", err)
+	}
+
+	// serialize to json
+	outputJSON, err := json.Marshal(inputEvent)
+	if err != nil {
+		t.Errorf("could not marshal event. details: %v", err)
+	}
+
+	test.AssertJsonsEqual(t, inputJSON, outputJSON)
+}
+
+func TestIoTCustomAuthorizerRequestMalformedJson(t *testing.T) {
+	test.TestMalformedJson(t, IoTCustomAuthorizerRequest{})
+}
+
+func TestIoTCustomAuthorizerResponseMarshaling(t *testing.T) {
+
+	// read json from file
+	inputJSON, err := ioutil.ReadFile("./testdata/iot-custom-auth-response.json")
+	if err != nil {
+		t.Errorf("could not open test file. details: %v", err)
+	}
+
+	// de-serialize into Go object
+	var inputEvent IoTCustomAuthorizerResponse
+	if err := json.Unmarshal(inputJSON, &inputEvent); err != nil {
+		t.Errorf("could not unmarshal event. details: %v", err)
+	}
+
+	// serialize to json
+	outputJSON, err := json.Marshal(inputEvent)
+	if err != nil {
+		t.Errorf("could not marshal event. details: %v", err)
+	}
+
+	test.AssertJsonsEqual(t, inputJSON, outputJSON)
+}
+
+func TestIoTCustomAuthorizerResponseMalformedJson(t *testing.T) {
+	test.TestMalformedJson(t, IoTCustomAuthorizerResponse{})
+}

events/testdata/iot-custom-auth-request.json

@@ -0,0 +1,3 @@
+{
+  "token":"allow"
+}

events/testdata/iot-custom-auth-response.json

@@ -0,0 +1,21 @@
+{
+  "isAuthenticated":true,
+  "principalId": "xxxxxxxx",
+  "disconnectAfterInSeconds": 86400,
+  "refreshAfterInSeconds": 300,
+  "policyDocuments": [
+    { "Version": "2012-10-17", "Statement": [
+        {
+          "Action": ["iot:Subscribe"],
+          "Effect": "Allow",
+          "Resource": ["*"]
+        }
+      ]
+    }
+  ],
+  "context": {
+    "username" : "johnDoe123",
+    "city" : "Seattle",
+    "country" : "USA"
+  }
+}