From ebfc7012658ed086066def144e674200f082e245 Mon Sep 17 00:00:00 2001
From: Greg Thole <gthole@gmail.com>
Date: Thu, 5 May 2016 16:14:51 -0400
Subject: [PATCH] provider/aws: AWS API Gateway request parameters json (#6501)

* Update docs with new parameters

* Add request parameters as JSON

* Update function name and error statements
---
 .../resource_aws_api_gateway_integration.go   | 18 ++++-
 ...ce_aws_api_gateway_integration_response.go |  2 +-
 ...source_aws_api_gateway_integration_test.go | 18 +++++
 .../aws/resource_aws_api_gateway_method.go    | 26 ++++--
 ...esource_aws_api_gateway_method_response.go |  2 +-
 ...ce_aws_api_gateway_method_response_test.go | 20 ++---
 .../resource_aws_api_gateway_method_test.go   | 80 +++++++++++++++++++
 builtin/providers/aws/structure.go            |  6 +-
 .../r/api_gateway_integration.html.markdown   |  3 +
 .../aws/r/api_gateway_method.html.markdown    |  2 +
 10 files changed, 154 insertions(+), 23 deletions(-)

diff --git a/builtin/providers/aws/resource_aws_api_gateway_integration.go b/builtin/providers/aws/resource_aws_api_gateway_integration.go
index c1dedd8777..68f9c50bf9 100644
--- a/builtin/providers/aws/resource_aws_api_gateway_integration.go
+++ b/builtin/providers/aws/resource_aws_api_gateway_integration.go
@@ -1,6 +1,7 @@
 package aws
 
 import (
+	"encoding/json"
 	"fmt"
 	"log"
 	"time"
@@ -73,6 +74,11 @@ func resourceAwsApiGatewayIntegration() *schema.Resource {
 				Optional: true,
 				Elem:     schema.TypeString,
 			},
+
+			"request_parameters_in_json": &schema.Schema{
+				Type:     schema.TypeString,
+				Optional: true,
+			},
 		},
 	}
 }
@@ -93,6 +99,13 @@ func resourceAwsApiGatewayIntegrationCreate(d *schema.ResourceData, meta interfa
 		templates[k] = v.(string)
 	}
 
+	parameters := make(map[string]string)
+	if v, ok := d.GetOk("request_parameters_in_json"); ok {
+		if err := json.Unmarshal([]byte(v.(string)), &parameters); err != nil {
+			return fmt.Errorf("Error unmarshaling request_parameters_in_json: %s", err)
+		}
+	}
+
 	var credentials *string
 	if val, ok := d.GetOk("credentials"); ok {
 		credentials = aws.String(val.(string))
@@ -105,8 +118,8 @@ func resourceAwsApiGatewayIntegrationCreate(d *schema.ResourceData, meta interfa
 		Type:       aws.String(d.Get("type").(string)),
 		IntegrationHttpMethod: integrationHttpMethod,
 		Uri: uri,
-		// TODO implement once [GH-2143](https://github.com/hashicorp/terraform/issues/2143) has been implemented
-		RequestParameters:  nil,
+		// TODO reimplement once [GH-2143](https://github.com/hashicorp/terraform/issues/2143) has been implemented
+		RequestParameters:  aws.StringMap(parameters),
 		RequestTemplates:   aws.StringMap(templates),
 		Credentials:        credentials,
 		CacheNamespace:     nil,
@@ -149,6 +162,7 @@ func resourceAwsApiGatewayIntegrationRead(d *schema.ResourceData, meta interface
 	d.Set("credentials", integration.Credentials)
 	d.Set("type", integration.Type)
 	d.Set("uri", integration.Uri)
+	d.Set("request_parameters_in_json", aws.StringValueMap(integration.RequestParameters))
 
 	return nil
 }
diff --git a/builtin/providers/aws/resource_aws_api_gateway_integration_response.go b/builtin/providers/aws/resource_aws_api_gateway_integration_response.go
index 588ef84b9b..e5f17abd78 100644
--- a/builtin/providers/aws/resource_aws_api_gateway_integration_response.go
+++ b/builtin/providers/aws/resource_aws_api_gateway_integration_response.go
@@ -75,7 +75,7 @@ func resourceAwsApiGatewayIntegrationResponseCreate(d *schema.ResourceData, meta
 	parameters := make(map[string]string)
 	if v, ok := d.GetOk("response_parameters_in_json"); ok {
 		if err := json.Unmarshal([]byte(v.(string)), &parameters); err != nil {
-			return fmt.Errorf("Error unmarshaling request_parameters_in_json: %s", err)
+			return fmt.Errorf("Error unmarshaling response_parameters_in_json: %s", err)
 		}
 	}
 
diff --git a/builtin/providers/aws/resource_aws_api_gateway_integration_test.go b/builtin/providers/aws/resource_aws_api_gateway_integration_test.go
index 74f87bb6c6..090bcf0a94 100644
--- a/builtin/providers/aws/resource_aws_api_gateway_integration_test.go
+++ b/builtin/providers/aws/resource_aws_api_gateway_integration_test.go
@@ -59,6 +59,9 @@ func testAccCheckAWSAPIGatewayMockIntegrationAttributes(conf *apigateway.Integra
 		if *conf.Type != "MOCK" {
 			return fmt.Errorf("Wrong Type: %q", *conf.Type)
 		}
+		if *conf.RequestParameters["integration.request.header.X-Authorization"] != "'updated'" {
+			return fmt.Errorf("wrong updated RequestParameters for header.X-Authorization")
+		}
 		return nil
 	}
 }
@@ -80,6 +83,9 @@ func testAccCheckAWSAPIGatewayIntegrationAttributes(conf *apigateway.Integration
 		if *conf.RequestTemplates["application/xml"] != "#set($inputRoot = $input.path('$'))\n{ }" {
 			return fmt.Errorf("wrong RequestTemplate for application/xml")
 		}
+		if *conf.RequestParameters["integration.request.header.X-Authorization"] != "'static'" {
+			return fmt.Errorf("wrong RequestParameters for header.X-Authorization")
+		}
 		return nil
 	}
 }
@@ -178,6 +184,12 @@ resource "aws_api_gateway_integration" "test" {
     "application/xml" = "#set($inputRoot = $input.path('$'))\n{ }"
   }
 
+  request_parameters_in_json = <<PARAMS
+  {
+	  "integration.request.header.X-Authorization": "'static'"
+  }
+  PARAMS
+
   type = "HTTP"
   uri = "https://www.google.de"
   integration_http_method = "GET"
@@ -211,6 +223,12 @@ resource "aws_api_gateway_integration" "test" {
   resource_id = "${aws_api_gateway_resource.test.id}"
   http_method = "${aws_api_gateway_method.test.http_method}"
 
+  request_parameters_in_json = <<PARAMS
+  {
+	  "integration.request.header.X-Authorization": "'updated'"
+  }
+  PARAMS
+
   type = "MOCK"
 }
 `
diff --git a/builtin/providers/aws/resource_aws_api_gateway_method.go b/builtin/providers/aws/resource_aws_api_gateway_method.go
index 31e90640d1..ad9b327af8 100644
--- a/builtin/providers/aws/resource_aws_api_gateway_method.go
+++ b/builtin/providers/aws/resource_aws_api_gateway_method.go
@@ -1,6 +1,7 @@
 package aws
 
 import (
+	"encoding/json"
 	"fmt"
 	"log"
 	"time"
@@ -55,6 +56,11 @@ func resourceAwsApiGatewayMethod() *schema.Resource {
 				Optional: true,
 				Elem:     schema.TypeString,
 			},
+
+			"request_parameters_in_json": &schema.Schema{
+				Type:     schema.TypeString,
+				Optional: true,
+			},
 		},
 	}
 }
@@ -68,10 +74,9 @@ func resourceAwsApiGatewayMethodCreate(d *schema.ResourceData, meta interface{})
 	}
 
 	parameters := make(map[string]bool)
-	if parameterData, ok := d.GetOk("request_parameters"); ok {
-		params := parameterData.(*schema.Set).List()
-		for k := range params {
-			parameters[params[k].(string)] = true
+	if v, ok := d.GetOk("request_parameters_in_json"); ok {
+		if err := json.Unmarshal([]byte(v.(string)), &parameters); err != nil {
+			return fmt.Errorf("Error unmarshaling request_parameters_in_json: %s", err)
 		}
 	}
 
@@ -81,8 +86,8 @@ func resourceAwsApiGatewayMethodCreate(d *schema.ResourceData, meta interface{})
 		ResourceId:        aws.String(d.Get("resource_id").(string)),
 		RestApiId:         aws.String(d.Get("rest_api_id").(string)),
 		RequestModels:     aws.StringMap(models),
-		// TODO implement once [GH-2143](https://github.com/hashicorp/terraform/issues/2143) has been implemented
-		RequestParameters: nil,
+		// TODO reimplement once [GH-2143](https://github.com/hashicorp/terraform/issues/2143) has been implemented
+		RequestParameters: aws.BoolMap(parameters),
 		ApiKeyRequired:    aws.Bool(d.Get("api_key_required").(bool)),
 	})
 	if err != nil {
@@ -113,6 +118,7 @@ func resourceAwsApiGatewayMethodRead(d *schema.ResourceData, meta interface{}) e
 	}
 	log.Printf("[DEBUG] Received API Gateway Method: %s", out)
 	d.SetId(fmt.Sprintf("agm-%s-%s-%s", d.Get("rest_api_id").(string), d.Get("resource_id").(string), d.Get("http_method").(string)))
+	d.Set("request_parameters_in_json", aws.BoolValueMap(out.RequestParameters))
 
 	return nil
 }
@@ -134,6 +140,14 @@ func resourceAwsApiGatewayMethodUpdate(d *schema.ResourceData, meta interface{})
 		operations = append(operations, expandApiGatewayRequestResponseModelOperations(d, "request_models", "requestModels")...)
 	}
 
+	if d.HasChange("request_parameters_in_json") {
+		ops, err := expandApiGatewayMethodParametersJSONOperations(d, "request_parameters_in_json", "requestParameters")
+		if err != nil {
+			return err
+		}
+		operations = append(operations, ops...)
+	}
+
 	method, err := conn.UpdateMethod(&apigateway.UpdateMethodInput{
 		HttpMethod:      aws.String(d.Get("http_method").(string)),
 		ResourceId:      aws.String(d.Get("resource_id").(string)),
diff --git a/builtin/providers/aws/resource_aws_api_gateway_method_response.go b/builtin/providers/aws/resource_aws_api_gateway_method_response.go
index c9c69d72a5..01c07bee04 100644
--- a/builtin/providers/aws/resource_aws_api_gateway_method_response.go
+++ b/builtin/providers/aws/resource_aws_api_gateway_method_response.go
@@ -130,7 +130,7 @@ func resourceAwsApiGatewayMethodResponseUpdate(d *schema.ResourceData, meta inte
 	}
 
 	if d.HasChange("response_parameters_in_json") {
-		ops, err := expandApiGatewayMethodResponseParametersJSONOperations(d, "response_parameters_in_json", "responseParameters")
+		ops, err := expandApiGatewayMethodParametersJSONOperations(d, "response_parameters_in_json", "responseParameters")
 		if err != nil {
 			return err
 		}
diff --git a/builtin/providers/aws/resource_aws_api_gateway_method_response_test.go b/builtin/providers/aws/resource_aws_api_gateway_method_response_test.go
index 5d136010ce..2c73f37931 100644
--- a/builtin/providers/aws/resource_aws_api_gateway_method_response_test.go
+++ b/builtin/providers/aws/resource_aws_api_gateway_method_response_test.go
@@ -184,11 +184,11 @@ resource "aws_api_gateway_method_response" "error" {
     "application/json" = "Error"
   }
 
-	response_parameters_in_json = <<PARAMS
-	{
-		"method.response.header.Content-Type": true
-	}
-	PARAMS
+  response_parameters_in_json = <<PARAMS
+  {
+    "method.response.header.Content-Type": true
+  }
+  PARAMS
 }
 `
 
@@ -224,11 +224,11 @@ resource "aws_api_gateway_method_response" "error" {
     "application/json" = "Empty"
   }
 
-	response_parameters_in_json = <<PARAMS
-	{
-		"method.response.header.Host": true
-	}
-	PARAMS
+  response_parameters_in_json = <<PARAMS
+  {
+    "method.response.header.Host": true
+  }
+  PARAMS
 
 }
 `
diff --git a/builtin/providers/aws/resource_aws_api_gateway_method_test.go b/builtin/providers/aws/resource_aws_api_gateway_method_test.go
index 0795e746d6..23b2593ae3 100644
--- a/builtin/providers/aws/resource_aws_api_gateway_method_test.go
+++ b/builtin/providers/aws/resource_aws_api_gateway_method_test.go
@@ -32,6 +32,14 @@ func TestAccAWSAPIGatewayMethod_basic(t *testing.T) {
 						"aws_api_gateway_method.test", "request_models.application/json", "Error"),
 				),
 			},
+
+			resource.TestStep{
+				Config: testAccAWSAPIGatewayMethodConfigUpdate,
+				Check: resource.ComposeTestCheckFunc(
+					testAccCheckAWSAPIGatewayMethodExists("aws_api_gateway_method.test", &conf),
+					testAccCheckAWSAPIGatewayMethodAttributesUpdate(&conf),
+				),
+			},
 		},
 	})
 }
@@ -44,6 +52,42 @@ func testAccCheckAWSAPIGatewayMethodAttributes(conf *apigateway.Method) resource
 		if *conf.AuthorizationType != "NONE" {
 			return fmt.Errorf("Wrong Authorization: %q", *conf.AuthorizationType)
 		}
+
+		if val, ok := conf.RequestParameters["method.request.header.Content-Type"]; !ok {
+			return fmt.Errorf("missing Content-Type RequestParameters")
+		} else {
+			if *val != false {
+				return fmt.Errorf("wrong Content-Type RequestParameters value")
+			}
+		}
+		if val, ok := conf.RequestParameters["method.request.querystring.page"]; !ok {
+			return fmt.Errorf("missing page RequestParameters")
+		} else {
+			if *val != true {
+				return fmt.Errorf("wrong query string RequestParameters value")
+			}
+		}
+
+		return nil
+	}
+}
+
+func testAccCheckAWSAPIGatewayMethodAttributesUpdate(conf *apigateway.Method) resource.TestCheckFunc {
+	return func(s *terraform.State) error {
+		if *conf.HttpMethod != "GET" {
+			return fmt.Errorf("Wrong HttpMethod: %q", *conf.HttpMethod)
+		}
+		if conf.RequestParameters["method.request.header.Content-Type"] != nil {
+			return fmt.Errorf("Content-Type RequestParameters shouldn't exist")
+		}
+		if val, ok := conf.RequestParameters["method.request.querystring.page"]; !ok {
+			return fmt.Errorf("missing updated page RequestParameters")
+		} else {
+			if *val != false {
+				return fmt.Errorf("wrong query string RequestParameters updated value")
+			}
+		}
+
 		return nil
 	}
 }
@@ -130,5 +174,41 @@ resource "aws_api_gateway_method" "test" {
   request_models = {
     "application/json" = "Error"
   }
+
+  request_parameters_in_json = <<PARAMS
+  {
+    "method.request.header.Content-Type": false,
+	"method.request.querystring.page": true
+  }
+  PARAMS
+}
+`
+
+const testAccAWSAPIGatewayMethodConfigUpdate = `
+resource "aws_api_gateway_rest_api" "test" {
+  name = "test"
+}
+
+resource "aws_api_gateway_resource" "test" {
+  rest_api_id = "${aws_api_gateway_rest_api.test.id}"
+  parent_id = "${aws_api_gateway_rest_api.test.root_resource_id}"
+  path_part = "test"
+}
+
+resource "aws_api_gateway_method" "test" {
+  rest_api_id = "${aws_api_gateway_rest_api.test.id}"
+  resource_id = "${aws_api_gateway_resource.test.id}"
+  http_method = "GET"
+  authorization = "NONE"
+
+  request_models = {
+    "application/json" = "Error"
+  }
+
+  request_parameters_in_json = <<PARAMS
+  {
+	"method.request.querystring.page": false
+  }
+  PARAMS
 }
 `
diff --git a/builtin/providers/aws/structure.go b/builtin/providers/aws/structure.go
index b0c84b9ec5..9074c6b0b4 100644
--- a/builtin/providers/aws/structure.go
+++ b/builtin/providers/aws/structure.go
@@ -948,7 +948,7 @@ func expandApiGatewayRequestResponseModelOperations(d *schema.ResourceData, key
 	return operations
 }
 
-func expandApiGatewayMethodResponseParametersJSONOperations(d *schema.ResourceData, key string, prefix string) ([]*apigateway.PatchOperation, error) {
+func expandApiGatewayMethodParametersJSONOperations(d *schema.ResourceData, key string, prefix string) ([]*apigateway.PatchOperation, error) {
 	operations := make([]*apigateway.PatchOperation, 0)
 
 	oldParameters, newParameters := d.GetChange(key)
@@ -956,12 +956,12 @@ func expandApiGatewayMethodResponseParametersJSONOperations(d *schema.ResourceDa
 	newParametersMap := make(map[string]interface{})
 
 	if err := json.Unmarshal([]byte(oldParameters.(string)), &oldParametersMap); err != nil {
-		err := fmt.Errorf("Error unmarshaling old response_parameters_in_json: %s", err)
+		err := fmt.Errorf("Error unmarshaling old %s: %s", key, err)
 		return operations, err
 	}
 
 	if err := json.Unmarshal([]byte(newParameters.(string)), &newParametersMap); err != nil {
-		err := fmt.Errorf("Error unmarshaling new response_parameters_in_json: %s", err)
+		err := fmt.Errorf("Error unmarshaling new %s: %s", key, err)
 		return operations, err
 	}
 
diff --git a/website/source/docs/providers/aws/r/api_gateway_integration.html.markdown b/website/source/docs/providers/aws/r/api_gateway_integration.html.markdown
index 028d6b28cc..fc36efcc07 100644
--- a/website/source/docs/providers/aws/r/api_gateway_integration.html.markdown
+++ b/website/source/docs/providers/aws/r/api_gateway_integration.html.markdown
@@ -56,3 +56,6 @@ The following arguments are supported:
   Not all methods are compatible with all `AWS` integrations.
   e.g. Lambda function [can only be invoked](https://github.com/awslabs/aws-apigateway-importer/issues/9#issuecomment-129651005) via `POST`.
 * `request_templates` - (Optional) A map of the integration's request templates.
+* `request_parameters_in_json` - (Optional) A map written as a JSON string specifying
+  the request query string parameters and headers that should be passed to the
+  backend responder
diff --git a/website/source/docs/providers/aws/r/api_gateway_method.html.markdown b/website/source/docs/providers/aws/r/api_gateway_method.html.markdown
index d61d334e6f..54e15183b2 100644
--- a/website/source/docs/providers/aws/r/api_gateway_method.html.markdown
+++ b/website/source/docs/providers/aws/r/api_gateway_method.html.markdown
@@ -44,3 +44,5 @@ The following arguments are supported:
 * `request_models` - (Optional) A map of the API models used for the request's content type
   where key is the content type (e.g. `application/json`)
   and value is either `Error`, `Empty` (built-in models) or `aws_api_gateway_model`'s `name`.
+* `request_parameters_in_json` - (Optional) A map written as a JSON string specifying
+  the request query string parameters and headers that should be passed to the integration