Test your AI Agents with the all-new Agent to Agent Testing Platform.Learn More

How to use store_and_get_lambda_code_archive method in localstack

Best Python code snippet using localstack_python

lambda_api.py

Source:lambda_api.py

...611            plugin.init_function_code(lambda_function)612    # unzipping can take some time - limit the execution time to avoid client/network timeout issues613    run_for_max_seconds(config.LAMBDA_CODE_EXTRACT_TIME, _set_and_configure)614    return {"FunctionName": lambda_function.name()}615def store_and_get_lambda_code_archive(616    lambda_function: LambdaFunction, zip_file_content: bytes = None617) -> Optional[Tuple[str, str, bytes]]:618    """Store the Lambda code referenced in the LambdaFunction details to disk as a zip file,619    and return the Lambda CWD, file name, and zip bytes content. May optionally return None620    in case this is a Lambda with the special bucket marker __local__, used for code mounting."""621    code_passed = lambda_function.code622    is_local_mount = code_passed.get("S3Bucket") == config.BUCKET_MARKER_LOCAL623    lambda_cwd = lambda_function.cwd624    if code_passed:625        lambda_cwd = lambda_cwd or set_archive_code(code_passed, lambda_function.name())626        if not zip_file_content and not is_local_mount:627            # Save the zip file to a temporary file that the lambda executors can reference628            zip_file_content = get_zip_bytes(code_passed)629    else:630        lambda_details = LambdaRegion.get().lambdas[lambda_function.arn()]631        lambda_cwd = lambda_cwd or lambda_details.cwd632    if not lambda_cwd:633        return634    # construct archive name635    archive_file = os.path.join(lambda_cwd, LAMBDA_ZIP_FILE_NAME)636    if not zip_file_content:637        zip_file_content = load_file(archive_file, mode="rb")638    else:639        # override lambda archive with fresh code if we got an update640        save_file(archive_file, zip_file_content)641    return lambda_cwd, archive_file, zip_file_content642def do_set_function_code(lambda_function: LambdaFunction):643    """Main function that creates the local zip archive for the given Lambda function, and644    optionally creates the handler function references (for LAMBDA_EXECUTOR=local)"""645    def generic_handler(*_):646        raise ClientError(647            (648                'Unable to find executor for Lambda function "%s". Note that '649                + "Node.js, Golang, and .Net Core Lambdas currently require LAMBDA_EXECUTOR=docker"650            )651            % lambda_name652        )653    lambda_name = lambda_function.name()654    arn = lambda_function.arn()655    runtime = get_lambda_runtime(lambda_function)656    lambda_environment = lambda_function.envvars657    handler_name = lambda_function.handler = lambda_function.handler or LAMBDA_DEFAULT_HANDLER658    code_passed = lambda_function.code659    is_local_mount = code_passed.get("S3Bucket") == config.BUCKET_MARKER_LOCAL660    # cleanup any left-over Lambda executor instances661    LAMBDA_EXECUTOR.cleanup(arn)662    # get local Lambda code archive path663    _result = store_and_get_lambda_code_archive(lambda_function)664    if not _result:665        return666    lambda_cwd, archive_file, zip_file_content = _result667    # Set the appropriate Lambda handler.668    lambda_handler = generic_handler669    is_java = lambda_executors.is_java_lambda(runtime)670    if is_java:671        # The Lambda executors for Docker subclass LambdaExecutorContainers, which672        # runs Lambda in Docker by passing all *.jar files in the function working673        # directory as part of the classpath. Obtain a Java handler function below.674        try:675            lambda_handler = get_java_handler(676                zip_file_content, archive_file, lambda_function=lambda_function677            )...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.