From 5f914a4b1d3fb43f753abf39e8155f849eb47127 Mon Sep 17 00:00:00 2001
From: Daniel Mil <mladan@amazon.com>
Date: Thu, 27 Jul 2023 11:02:49 -0700
Subject: [PATCH] test: Add terraform v2 api integration test cases

---
 ...st_start_api_with_terraform_application.py |  15 +++-
 .../HelloWorldFunction.zip                    | Bin 0 -> 1079 bytes
 .../terraform-v2-api-quick-create/main.tf     |  49 +++++++++++++
 .../HelloWorldFunction.zip                    | Bin 0 -> 1079 bytes
 .../terraform/terraform-v2-api-simple/main.tf |  67 ++++++++++++++++++
 5 files changed, 129 insertions(+), 2 deletions(-)
 create mode 100644 tests/integration/testdata/start_api/terraform/terraform-v2-api-quick-create/HelloWorldFunction.zip
 create mode 100644 tests/integration/testdata/start_api/terraform/terraform-v2-api-quick-create/main.tf
 create mode 100644 tests/integration/testdata/start_api/terraform/terraform-v2-api-simple/HelloWorldFunction.zip
 create mode 100644 tests/integration/testdata/start_api/terraform/terraform-v2-api-simple/main.tf

diff --git a/tests/integration/local/start_api/test_start_api_with_terraform_application.py b/tests/integration/local/start_api/test_start_api_with_terraform_application.py
index e8ab21d01d..be98e8e76f 100644
--- a/tests/integration/local/start_api/test_start_api_with_terraform_application.py
+++ b/tests/integration/local/start_api/test_start_api_with_terraform_application.py
@@ -95,9 +95,20 @@ def tearDownClass(cls) -> None:
     "Skip Terraform test cases unless running in CI",
 )
 @pytest.mark.flaky(reruns=3)
+@parameterized_class(
+    [
+        {
+            "terraform_application": "terraform-v1-api-simple",
+        },
+        {
+            "terraform_application": "terraform-v2-api-simple",
+        },
+        {
+            "terraform_application": "terraform-v2-api-quick-create",
+        },
+    ]
+)
 class TestStartApiTerraformApplication(TerraformStartApiIntegrationBase):
-    terraform_application = "terraform-v1-api-simple"
-
     def setUp(self):
         self.url = "http://127.0.0.1:{}".format(self.port)
 
diff --git a/tests/integration/testdata/start_api/terraform/terraform-v2-api-quick-create/HelloWorldFunction.zip b/tests/integration/testdata/start_api/terraform/terraform-v2-api-quick-create/HelloWorldFunction.zip
new file mode 100644
index 0000000000000000000000000000000000000000..4daa6de02dd8d472c93e7be277639d6220ea20b3
GIT binary patch
literal 1079
zcmWIWW@Zs#-~hsiz!xD5NPwF`fgwIVGcU6wK3=b&GBkvjfn6f?L=p)9PQR8^TEWf0
z$nt}cfdOa$0|PTdfHyk_NIw(QkN~Vk*RwD%umO!uEGQtt)V;w6yKWl@?ES6H_H>d%
zj=rr_mB#73$cT&^d&Rnvd~@F3nr0<s>UsKKwd(1On*(k|vmU$l<Lu??{G7Yd&OG|(
zcp|qk&E3{=-`1)ua+;=%gZl5ilkP`ZuHeh;wJ&&}dnGNw=Hs$c3Ob5MHowVjPq0<A
z>C-Lv;C$Aq$o-7pq&E{M-q~^XxoMvJEv_l+m@7jTa9;hPw$6Vy`wtz-{!cZE_a|D;
zeVDjm-otwV`j)dFmQUoFvQj%^=GxV#jB;WW`yak~>TfKv^`&=a>12=7dUs@}Yn^z}
z<C?W6EW+x^o!d656K+=|t!sYYmh$Ck*R<bxq8Xn$)yu7x894H{>Fu?8)zcdJTqJ=t
z?SE6fiLK;bUk3&I4etb<+SUg09kOvSntbo~f?C7+ga0qrO)`tl_p0VQ<7}sSdh&*g
z$w~E2-%Miq^?$!E*jEv^Y;xM}6Fw>bvpuG-H<`PiX<4zU?Xji0HFNg=UgQ|py+YGt
zRrH$pt#d_Oa|5T`@Htzt>Z0C$pN~`I-W@BlkX^#O$1aogyad~GpBYa?Oit|OG%Q@r
zJne+QhMg=wGo;F|=4x6*KUUklYT^|>#^BDNRr{aKXiYwO@2*tqq9%TG^V?6a-dj<u
zaqRy7fPLHZ_I~MYkefT%H%CObY;NP?g?^oHt*>yoq`xwJ&A=1r>d(6HZAx$U$5MgY
z6aJ*VtGvDB-}X1=;G`9O{2n_nZGjRH_M`<$cLKnqRg_v-npu>Zo0?ZrtXEP|LQc|U
zWD;RU%@?p_4ayfVu%r>hA|aOqc%vDN<S9_DfPp2A(Lg3~c?82upiY=$LBRn7OB%NU
Zg-H#Z0B=?{kO|B{$j!*WAjS;h0RRw%Zs`C3

literal 0
HcmV?d00001

diff --git a/tests/integration/testdata/start_api/terraform/terraform-v2-api-quick-create/main.tf b/tests/integration/testdata/start_api/terraform/terraform-v2-api-quick-create/main.tf
new file mode 100644
index 0000000000..9c0bba204d
--- /dev/null
+++ b/tests/integration/testdata/start_api/terraform/terraform-v2-api-quick-create/main.tf
@@ -0,0 +1,49 @@
+provider "aws" {
+}
+
+resource "aws_iam_role" "iam_for_lambda" {
+  name = "iam_for_lambda"
+
+  assume_role_policy = <<EOF
+{
+  "Version": "2012-10-17",
+  "Statement": [
+    {
+      "Action": "sts:AssumeRole",
+      "Principal": {
+        "Service": "lambda.amazonaws.com"
+      },
+      "Effect": "Allow",
+      "Sid": ""
+    }
+  ]
+}
+EOF
+}
+
+resource "aws_s3_bucket" "lambda_code_bucket" {
+  bucket = "lambda_code_bucket"
+}
+
+resource "aws_s3_object" "s3_lambda_code" {
+  bucket = "lambda_code_bucket"
+  key    = "s3_lambda_code_key"
+  source = "HelloWorldFunction.zip"
+}
+
+resource "aws_lambda_function" "HelloWorldFunction" {
+  s3_bucket     = "lambda_code_bucket"
+  s3_key        = "s3_lambda_code_key"
+  handler       = "app.lambda_handler"
+  runtime       = "python3.8"
+  function_name = "HelloWorldFunction"
+  timeout       = 500
+  role          = aws_iam_role.iam_for_lambda.arn
+}
+
+resource "aws_apigatewayv2_api" "quick_create_api" {
+  name           = "quick_create_api"
+  protocol_type  = "HTTP"
+  target         = aws_lambda_function.HelloWorldFunction.invoke_arn
+  route_key      = "GET /hello"
+}
diff --git a/tests/integration/testdata/start_api/terraform/terraform-v2-api-simple/HelloWorldFunction.zip b/tests/integration/testdata/start_api/terraform/terraform-v2-api-simple/HelloWorldFunction.zip
new file mode 100644
index 0000000000000000000000000000000000000000..4daa6de02dd8d472c93e7be277639d6220ea20b3
GIT binary patch
literal 1079
zcmWIWW@Zs#-~hsiz!xD5NPwF`fgwIVGcU6wK3=b&GBkvjfn6f?L=p)9PQR8^TEWf0
z$nt}cfdOa$0|PTdfHyk_NIw(QkN~Vk*RwD%umO!uEGQtt)V;w6yKWl@?ES6H_H>d%
zj=rr_mB#73$cT&^d&Rnvd~@F3nr0<s>UsKKwd(1On*(k|vmU$l<Lu??{G7Yd&OG|(
zcp|qk&E3{=-`1)ua+;=%gZl5ilkP`ZuHeh;wJ&&}dnGNw=Hs$c3Ob5MHowVjPq0<A
z>C-Lv;C$Aq$o-7pq&E{M-q~^XxoMvJEv_l+m@7jTa9;hPw$6Vy`wtz-{!cZE_a|D;
zeVDjm-otwV`j)dFmQUoFvQj%^=GxV#jB;WW`yak~>TfKv^`&=a>12=7dUs@}Yn^z}
z<C?W6EW+x^o!d656K+=|t!sYYmh$Ck*R<bxq8Xn$)yu7x894H{>Fu?8)zcdJTqJ=t
z?SE6fiLK;bUk3&I4etb<+SUg09kOvSntbo~f?C7+ga0qrO)`tl_p0VQ<7}sSdh&*g
z$w~E2-%Miq^?$!E*jEv^Y;xM}6Fw>bvpuG-H<`PiX<4zU?Xji0HFNg=UgQ|py+YGt
zRrH$pt#d_Oa|5T`@Htzt>Z0C$pN~`I-W@BlkX^#O$1aogyad~GpBYa?Oit|OG%Q@r
zJne+QhMg=wGo;F|=4x6*KUUklYT^|>#^BDNRr{aKXiYwO@2*tqq9%TG^V?6a-dj<u
zaqRy7fPLHZ_I~MYkefT%H%CObY;NP?g?^oHt*>yoq`xwJ&A=1r>d(6HZAx$U$5MgY
z6aJ*VtGvDB-}X1=;G`9O{2n_nZGjRH_M`<$cLKnqRg_v-npu>Zo0?ZrtXEP|LQc|U
zWD;RU%@?p_4ayfVu%r>hA|aOqc%vDN<S9_DfPp2A(Lg3~c?82upiY=$LBRn7OB%NU
Zg-H#Z0B=?{kO|B{$j!*WAjS;h0RRw%Zs`C3

literal 0
HcmV?d00001

diff --git a/tests/integration/testdata/start_api/terraform/terraform-v2-api-simple/main.tf b/tests/integration/testdata/start_api/terraform/terraform-v2-api-simple/main.tf
new file mode 100644
index 0000000000..1a95397b2a
--- /dev/null
+++ b/tests/integration/testdata/start_api/terraform/terraform-v2-api-simple/main.tf
@@ -0,0 +1,67 @@
+provider "aws" {
+}
+
+resource "aws_iam_role" "iam_for_lambda" {
+  name = "iam_for_lambda"
+
+  assume_role_policy = <<EOF
+{
+  "Version": "2012-10-17",
+  "Statement": [
+    {
+      "Action": "sts:AssumeRole",
+      "Principal": {
+        "Service": "lambda.amazonaws.com"
+      },
+      "Effect": "Allow",
+      "Sid": ""
+    }
+  ]
+}
+EOF
+}
+
+resource "aws_s3_bucket" "lambda_code_bucket" {
+  bucket = "lambda_code_bucket"
+}
+
+resource "aws_s3_object" "s3_lambda_code" {
+  bucket = "lambda_code_bucket"
+  key    = "s3_lambda_code_key"
+  source = "HelloWorldFunction.zip"
+}
+
+resource "aws_lambda_function" "HelloWorldFunction" {
+  s3_bucket     = "lambda_code_bucket"
+  s3_key        = "s3_lambda_code_key"
+  handler       = "app.lambda_handler"
+  runtime       = "python3.8"
+  function_name = "HelloWorldFunction"
+  timeout       = 500
+  role          = aws_iam_role.iam_for_lambda.arn
+}
+
+resource "aws_apigatewayv2_api" "my_api" {
+  name           = "my_api"
+  protocol_type  = "HTTP"
+}
+
+resource "aws_apigatewayv2_route" "example" {
+  api_id         = aws_apigatewayv2_api.my_api.id
+  target         = aws_apigatewayv2_integration.example.id
+  route_key      = "GET /hello"
+  operation_name = "my_operation"
+}
+
+resource "aws_apigatewayv2_stage" "example" {
+  api_id          = aws_apigatewayv2_api.my_api.id
+  name            = "example-stage"
+}
+
+resource "aws_apigatewayv2_integration" "example" {
+  api_id                 = aws_apigatewayv2_api.my_api.id
+  integration_type       = "AWS_PROXY"
+  integration_method     = "POST"
+  integration_uri        = aws_lambda_function.HelloWorldFunction.invoke_arn
+  payload_format_version = "2.0"
+}