Adding e2e tests for ScalingConfig

Author: Vandita2020

apis/v1alpha1/ack-generate-metadata.yaml

@@ -1,5 +1,5 @@
 ack_generate_info:
-  build_date: "2023-01-29T21:40:15Z"
+  build_date: "2023-02-01T00:38:06Z"
   build_hash: 12246c7da82841b351ec7a9e1f139f9338f2784b
   go_version: go1.19
   version: v0.20.1-14-g12246c7

test/e2e/tests/test_event_source_mapping.py

@@ -90,6 +90,7 @@ def test_smoke_sqs_queue_stream(self, lambda_client, lambda_function):
         replacements["EVENT_SOURCE_MAPPING_NAME"] = resource_name
         replacements["BATCH_SIZE"] = "10"
         replacements["FUNCTION_NAME"] = lambda_function_name
+        replacements["RESERVED_CONCURRENT_EXECUTIONS"] = "10"
         replacements["EVENT_SOURCE_ARN"] = resources.ESMQueue.arn
         replacements["MAXIMUM_BATCHING_WINDOW_IN_SECONDS"] = "1"
 
@@ -128,6 +129,7 @@ def test_smoke_sqs_queue_stream(self, lambda_client, lambda_function):
                 },
             ]
         }
+        cr["spec"]["scalingConfig"] = {"maximumConcurrency": 4}
 
         # Patch k8s resource
         k8s.patch_custom_resource(ref, cr)
@@ -142,6 +144,7 @@ def test_smoke_sqs_queue_stream(self, lambda_client, lambda_function):
                 "Pattern": "{\"controller-version\":[\"v1\"]}"
             },
         ]
+        assert esm["ScalingConfig"]["MaximumConcurrency"] == 4
 
         # Delete the filterCriteria field
         cr = k8s.wait_resource_consumed_by_controller(ref)
@@ -319,4 +322,69 @@ def test_smoke_dynamodb_table_stream(self, lambda_client, lambda_function):
         time.sleep(DELETE_WAIT_AFTER_SECONDS)
 
         # Check ESM doesn't exist
-        assert not lambda_validator.event_source_mapping_exists(esm_uuid)
+        assert not lambda_validator.event_source_mapping_exists(esm_uuid)
+    
+    def test_scaling_config(self, lambda_client, lambda_function):
+
+        (_, function_resource) = lambda_function
+        lambda_function_name = function_resource["spec"]["name"]
+
+        resource_name = random_suffix_name("lambda-esm", 24)
+        resources = get_bootstrap_resources()
+
+        replacements = REPLACEMENT_VALUES.copy()
+        replacements["AWS_REGION"] = get_region()
+        replacements["EVENT_SOURCE_MAPPING_NAME"] = resource_name
+        replacements["BATCH_SIZE"] = "10"
+        replacements["RESERVED_CONCURRENT_EXECUTIONS"] = "10"
+        replacements["FUNCTION_NAME"] = lambda_function_name
+        replacements["EVENT_SOURCE_ARN"] = resources.ESMQueue.arn
+        replacements["MAXIMUM_BATCHING_WINDOW_IN_SECONDS"] = "1"
+
+        # Load ESM CR
+        resource_data = load_lambda_resource(
+            "event_source_mapping_sqs",
+            additional_replacements=replacements,
+        )
+        logging.debug(resource_data)
+
+        # Create k8s resource
+        ref = k8s.CustomResourceReference(
+            CRD_GROUP, CRD_VERSION, RESOURCE_PLURAL,
+            resource_name, namespace="default",
+        )
+        k8s.create_custom_resource(ref, resource_data)
+        cr = k8s.wait_resource_consumed_by_controller(ref)
+
+        assert cr is not None
+        assert k8s.get_resource_exists(ref)
+
+        time.sleep(CREATE_WAIT_AFTER_SECONDS)
+
+        esm_uuid = cr['status']['uuid']
+
+        lambda_validator = LambdaValidator(lambda_client)
+        # Check ESM exists
+        assert lambda_validator.event_source_mapping_exists(esm_uuid)
+
+        # Update cr
+        cr["spec"]["scalingConfig"] = {"maximumConcurrency": 4}
+
+        # Patch k8s resource
+        k8s.patch_custom_resource(ref, cr)
+        time.sleep(UPDATE_WAIT_AFTER_SECONDS)
+
+        # Check ESM batch size
+        esm = lambda_validator.get_event_source_mapping(esm_uuid)
+        assert esm is not None
+        assert esm["ScalingConfig"]["MaximumConcurrency"] == 4
+
+        # Delete k8s resource
+        _, deleted = k8s.delete_custom_resource(ref)
+        assert deleted
+
+        time.sleep(DELETE_WAIT_AFTER_SECONDS)
+
+        # Check ESM doesn't exist
+        assert not lambda_validator.event_source_mapping_exists(esm_uuid)
+