Adding additional E2E tests

Author: Vandita2020

apis/v1alpha1/ack-generate-metadata.yaml

@@ -1,9 +1,9 @@
 ack_generate_info:
-  build_date: "2023-12-22T06:14:56Z"
-  build_hash: 1f16813c807af6889060b4ce7ded2a69dc027d8c
-  go_version: go1.21.5
-  version: v0.28.0
-api_directory_checksum: 1c73101b17769d9e1165fbb915bdeeb3dcd8b075
+  build_date: "2024-01-04T19:34:06Z"
+  build_hash: 1563ceb2b15ff29ee692b4178d4de5f2cbe0b0f0
+  go_version: go1.19
+  version: v0.28.0-12-g1563ceb
+api_directory_checksum: 6d988b03425fad3e88d887c1cd4510a07e4be281
 api_version: v1alpha1
 aws_sdk_go_version: v1.44.181
 generator_config_info:

test/e2e/tests/test_version.py

@@ -17,6 +17,8 @@
 import pytest
 import time
 import logging
+import hashlib
+import base64
 
 from acktest.resources import random_suffix_name
 from acktest.aws.identity import get_region
@@ -25,7 +27,7 @@
 from e2e import service_marker, CRD_GROUP, CRD_VERSION, load_lambda_resource
 from e2e.replacement_values import REPLACEMENT_VALUES
 from e2e.bootstrap_resources import get_bootstrap_resources
-from e2e.service_bootstrap import LAMBDA_FUNCTION_FILE_ZIP
+from e2e.service_bootstrap import LAMBDA_FUNCTION_FILE_ZIP, LAMBDA_FUNCTION_FILE_PATH_ZIP
 from e2e.tests.helper import LambdaValidator
 
 RESOURCE_PLURAL = "versions"
@@ -88,7 +90,7 @@ def test_smoke(self, lambda_client, lambda_function):
         replacements["AWS_REGION"] = get_region()
         replacements["FUNCTION_NAME"] = lambda_function_name
         replacements["VERSION_NAME"] = resource_name
-
+        
         # Load Lambda CR
         resource_data = load_lambda_resource(
             "version",
@@ -126,3 +128,60 @@ def test_smoke(self, lambda_client, lambda_function):
 
         # Check function version doesn't exist
         assert not lambda_validator.version_exists(lambda_function_name, version_number)
+
+    def test_version_with_check(self, lambda_client, lambda_function):
+        (_, function_resource) = lambda_function
+        lambda_function_name = function_resource["spec"]["name"]
+
+        resource_name = random_suffix_name("lambda-version", 24)
+
+        archive_1 = open(LAMBDA_FUNCTION_FILE_PATH_ZIP, 'rb') 
+        readFile_1 = archive_1.read() 
+        hash_1 = hashlib.sha256(readFile_1) 
+        binary_hash_1 = hash_1.digest() 
+        base64_hash_1 = base64.b64encode(binary_hash_1).decode('utf-8')
+
+        replacements = REPLACEMENT_VALUES.copy()
+        replacements["AWS_REGION"] = get_region()
+        replacements["FUNCTION_NAME"] = lambda_function_name
+        replacements["VERSION_NAME"] = resource_name
+        replacements["HASH"] = base64_hash_1
+        replacements["REVISION_ID"] = ""
+
+        # Load Lambda CR
+        resource_data = load_lambda_resource(
+            "version_with_check",
+            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)
+
+        cr = k8s.wait_resource_consumed_by_controller(ref)
+
+        lambda_validator = LambdaValidator(lambda_client)
+
+        version_number = cr['status']['version']
+
+        # Check version exists
+        assert lambda_validator.version_exists(lambda_function_name, version_number)
+
+        # Delete k8s resource
+        _, deleted = k8s.delete_custom_resource(ref)
+        assert deleted is True
+
+        time.sleep(DELETE_WAIT_AFTER_SECONDS)
+
+        # Check function version doesn't exist
+        assert not lambda_validator.version_exists(lambda_function_name, version_number)