boto3: setting use_docker to false

Signed-off-by: Varsha GS <[email protected]>

Author: GSVarsha

tests/clients/boto3/test_boto3_lambda.py

@@ -21,58 +21,29 @@
 from ...helpers import get_first_span_by_filter
 
 class TestLambda(unittest.TestCase):
-    def _get_role(self):
-        iam = boto3.client("iam", region_name=self.lambda_region)
-        return iam.create_role(
-            RoleName="my-role",
-            AssumeRolePolicyDocument="some policy"
-        )["Role"]["Arn"]
-        
-    def _process_lambda(self, func_str):
-        zip_output = BytesIO()
-        with ZipFile(zip_output, "w") as zip_file:
-            zip_file.writestr("lambda_function.py", func_str)
-        return zip_output.getvalue()
-    
-    def _get_test_zip_file(self):
-        pfunc = """
-def lambda_handler(event, context):
-    print("custom log event")
-    return {"message": "success"}
-"""
-        return self._process_lambda(pfunc)
-
     def setUp(self):
         """ Clear all spans before a test run """
         self.recorder = tracer.recorder
         self.recorder.clear_spans()
-        self.mock = mock_aws()
+        self.mock = mock_aws(config={"lambda": {"use_docker": False}})
         self.mock.start()
         self.lambda_region = "us-east-1"
         self.aws_lambda = boto3.client('lambda', region_name=self.lambda_region)
         self.function_name = "myfunc"
-        self.aws_lambda.create_function(
-            FunctionName=self.function_name,
-            Runtime="python3.9",
-            Role=self._get_role(),
-            Handler="lambda_function.lambda_handler",
-            Code={"ZipFile": self._get_test_zip_file()}
-        )
 
     def tearDown(self):
         # Stop Moto after each test
         self.mock.stop()
 
 
-    @pytest.mark.skip("Lambda mocking requires docker")
     def test_lambda_invoke(self):
         with tracer.start_active_span('test'):
-            result = self.aws_lambda.invoke(FunctionName=self.function_name)
+            result = self.aws_lambda.invoke(FunctionName=self.function_name, Payload=json.dumps({"message": "success"}))
 
         self.assertEqual(result["StatusCode"], 200)
-        payload = json.loads(result["Payload"].read().decode("utf-8"))
-        self.assertIn("message", payload)
-        self.assertEqual("success", payload["message"])
+        result_payload = json.loads(result["Payload"].read().decode("utf-8"))
+        self.assertIn("message", result_payload)
+        self.assertEqual("success", result_payload["message"])
 
         spans = tracer.recorder.queued_spans()
         self.assertEqual(2, len(spans))
@@ -94,7 +65,7 @@ def test_lambda_invoke(self):
         self.assertEqual(boto_span.data['boto3']['op'], 'Invoke')
         endpoint = f'https://lambda.{self.lambda_region}.amazonaws.com'
         self.assertEqual(boto_span.data['boto3']['ep'], endpoint)
-        self.assertEqual(boto_span.data['boto3']['reg'], 'us-east-1')
+        self.assertEqual(boto_span.data['boto3']['reg'], self.lambda_region)
         self.assertIn('FunctionName', boto_span.data['boto3']['payload'])
         self.assertEqual(boto_span.data['boto3']['payload']['FunctionName'], self.function_name)
         self.assertEqual(boto_span.data['http']['status'], 200)