From 82960e4249fe5b2ccb0200a16905f3f3f008a89b Mon Sep 17 00:00:00 2001
From: Kishan B <kishancs46@gmail.com>
Date: Fri, 1 Dec 2023 19:26:20 +0530
Subject: [PATCH] Add support for aws_api_gateway_integration

---
 pkg/convertor.go      |  2 +-
 pkg/convertor_test.go | 16 ++++++++++++++++
 2 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/pkg/convertor.go b/pkg/convertor.go
index 77be0f0..6c66ea2 100644
--- a/pkg/convertor.go
+++ b/pkg/convertor.go
@@ -26,7 +26,7 @@ func computeResourceID(resource parser.TerraformResource) string {
 		return computeResourceIDForAWSSecurityGroupRole(resource)
 	case "aws_api_gateway_resource":
 		return fmt.Sprintf("%s/%s", getValue("rest_api_id"), getValue("id"))
-	case "aws_api_gateway_method":
+	case "aws_api_gateway_method", "aws_api_gateway_integration":
 		return fmt.Sprintf("%s/%s/%s", getValue("rest_api_id"), getValue("resource_id"), getValue("http_method"))
 	default:
 		return getValue("id")
diff --git a/pkg/convertor_test.go b/pkg/convertor_test.go
index 4b48305..428105f 100644
--- a/pkg/convertor_test.go
+++ b/pkg/convertor_test.go
@@ -111,6 +111,22 @@ func Test_ComputeTerraformImportForResource(t *testing.T) {
 				ResourceID:      "rest_api_id/resource_id/http_method",
 			},
 		},
+		{
+			name: "For aws_api_gateway_integration",
+			terraformResource: parser.TerraformResource{
+				Address: "aws_api_gateway_integration.test",
+				Type:    "aws_api_gateway_integration",
+				AttributeValues: map[string]any{
+					"http_method": "http_method",
+					"resource_id": "resource_id",
+					"rest_api_id": "rest_api_id",
+				},
+			},
+			expected: TerraformImport{
+				ResourceAddress: "aws_api_gateway_integration.test",
+				ResourceID:      "rest_api_id/resource_id/http_method",
+			},
+		},
 		{
 			name: "For everything else",
 			terraformResource: parser.TerraformResource{