improved detection of first creation of lambda function in LocalStack

Author: DinisCruz

osbot_serverless_s3_browser/utils/deploy/Deploy__Serverless_S3_Browser.py

@@ -1,5 +1,6 @@
 from osbot_aws.AWS_Config                                                               import AWS_Config
 from osbot_aws.deploy.Deploy_Lambda                                                     import Deploy_Lambda
+from osbot_aws.helpers.Lambda_Upload_Package import Lambda_Upload_Package
 from osbot_utils.decorators.methods.cache_on_self                                       import cache_on_self
 from osbot_utils.helpers.Safe_Id                                                        import Safe_Id
 from osbot_utils.type_safe.Type_Safe                                                    import Type_Safe
@@ -29,9 +30,16 @@ def deploy_lambda(self):
     def deploy(self):
         with self.deploy_lambda() as _:
             result = _.update()
-            if result != "Successful":
-                raise Exception(f"Lambda update failed: {result}")
-            return True
+            print(f"***** : {result}")
+            if result == "Successful":
+                return True
+            if result == 'Pending':
+                print(">>>>>>> Pending result >>>> going to wait a bit more")
+                wait_result = self.lambda_function().wait_for_function_update_to_complete()
+                print(f">>>>>>>> WAIT RESULT: {wait_result}")                           # todo add check here
+                return True
+            raise Exception(f"Lambda update failed: {result}")
+
 
     def lambda_name(self):
         return f'{BASE__LAMBDA_NAME}__{self.stage}'
@@ -55,3 +63,9 @@ def setup_aws_environment(self):
                 if result.get('status') == 'ok':
                     _.bucket__osbot_lambdas__exists = True
         return aws_setup
+
+    def upload_lambda_dependencies_to_s3(self):
+        packages = ['fastapi']
+        lambda_upload_package = Lambda_Upload_Package()
+        for package in packages:
+            return lambda_upload_package.install_locally(package)

tests/unit/utils/deploy/test_Deploy__Serverless_S3_Browser.py

@@ -1,10 +1,9 @@
-from unittest                                                               import TestCase
-from osbot_aws.deploy.Deploy_Lambda                                         import Deploy_Lambda
-from osbot_utils.utils.Objects                                              import __
-from osbot_serverless_s3_browser.utils.deploy.Deploy__Serverless_S3_Browser import Deploy__Serverless_S3_Browser
-from osbot_serverless_s3_browser.utils.deploy.Schema__AWS_Setup__Serverless_S3_Browser import \
-    Schema__AWS_Setup__Serverless_S3_Browser
-from tests.s3_browser__objs_for_tests                                       import setup_local_stack, S3_BROWSER__TEST__AWS_ACCOUNT_ID, S3_BROWSER__TEST__AWS_DEFAULT_REGION
+from unittest                                                                           import TestCase
+from osbot_aws.deploy.Deploy_Lambda                                                     import Deploy_Lambda
+from osbot_utils.utils.Objects                                                          import __
+from osbot_serverless_s3_browser.utils.deploy.Deploy__Serverless_S3_Browser             import Deploy__Serverless_S3_Browser
+from osbot_serverless_s3_browser.utils.deploy.Schema__AWS_Setup__Serverless_S3_Browser  import Schema__AWS_Setup__Serverless_S3_Browser
+from tests.s3_browser__objs_for_tests                                                   import setup_local_stack, S3_BROWSER__TEST__AWS_ACCOUNT_ID, S3_BROWSER__TEST__AWS_DEFAULT_REGION
 
 
 class test_Deploy__Serverless_S3_Browser(TestCase):
@@ -29,7 +28,9 @@ def test_1__setup_aws_environment(self):
                                  bucket__osbot_lambdas__name   = '000022220000--osbot-lambdas--eu-west-2',
                                  current_aws_region            = 'eu-west-2'                             )
 
-    def test_2__deploy(self):
+    def test_2_upload_lambda_dependencies_to_s3(self):
+        pass
+    def test_3__deploy(self):
         with self.deploy_s3_browser as _:
             assert _.lambda_function().exists() is False
             result = _.deploy()