node-lambda 0.18.0 → 0.21.0

package/.github/workflows/workflow.yml

@@ -9,13 +9,14 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        node-version: [10.x, 12.x, 14.x]
+        node-version: [12.x, 14.x, 16.x, 17.x]
         os: [ubuntu-latest, windows-latest, macos-latest]
     steps:
-    - uses: actions/checkout@master
-    - uses: actions/setup-node@master
+    - uses: actions/checkout@v2
+    - uses: actions/setup-node@v2
       with:
         node-version: ${{ matrix.node-version }}
+        cache: 'npm'
     - run: npm i -g npm
     - run: node -v
     - run: npm -v

package/.travis.yml

@@ -4,6 +4,8 @@ node_js:
   - "lts/*"
   - "node"
 
+install: npm i -g npm
+
 cache:
   directories:
     - node_modules

package/CHANGELOG.md

@@ -421,3 +421,26 @@ This project adheres to [Semantic Versioning](http://semver.org/).
 - Upgrade packages [#538](https://github.com/motdotla/node-lambda/pull/538)
 - Bump lodash from 4.17.15 to 4.17.19 [#536](https://github.com/motdotla/node-lambda/pull/536)
 - Add build badge in README [#534](https://github.com/motdotla/node-lambda/pull/534)
+
+## [0.19.0] - 2021-03-30
+### Features
+- feat: support `--no-optional` option in npm install [#557](https://github.com/motdotla/node-lambda/pull/557)
+
+## [0.19.1] - 2021-04-24
+### Bugfixes
+- Fix errors caused by old proxy-agent [#564](https://github.com/motdotla/node-lambda/pull/564)
+
+## [0.20.0] - 2021-09-25
+### Features
+- updated proxy-agent to 5.0.0 [#574](https://github.com/motdotla/node-lambda/pull/574)
+- show suggestions after an error for an unknown command or option [#572](https://github.com/motdotla/node-lambda/pull/572)
+- feat: drop nodejs10x from lambda runtime [#571](https://github.com/motdotla/node-lambda/pull/571)
+
+### Bugfixes
+- fix(_uploadExisting): fix function update errors [#575](https://github.com/motdotla/node-lambda/pull/575)
+- test: fix npm install test failing in some cases [#569](https://github.com/motdotla/node-lambda/pull/569)
+- Clean the tmp dir during `_archivePrebuilt` to match `_buildAndArchive` behavior [#518](https://github.com/motdotla/node-lambda/pull/518)
+
+## [0.21.0] - 2021-11-10
+### Features
+- feat: support for yarn [#581](https://github.com/motdotla/node-lambda/pull/581)

package/README.md

@@ -64,6 +64,7 @@ $ echo -e ".env\ndeploy.env\nevent.json\n.lambda" >> .gitignore
 ##### Deploy env variables
 
 ```
+PACKAGE_MANAGER            // (default: 'npm')
 AWS_ENVIRONMENT            // (default: '')
 AWS_ENDPOINT               // (default: '')
 CONFIG_FILE                // (default: '')
@@ -109,23 +110,21 @@ Runs your Amazon Lambda index.js file locally. Passes `event.json` data to the A
 
 ```
 $ node-lambda run --help
-
-Usage: run|execute [options]
+Usage: node-lambda run|execute [options]
 
 Run your Amazon Lambda application locally
 
-
 Options:
-  -h, --help                            output usage information
-  -H, --handler [index.handler]         Lambda Handler {index.handler}
-  -j, --eventFile [event.json]          Event JSON File
-  -u, --runtime [nodejs14.x]            Lambda Runtime
-  -t, --timeout [3]                     Lambda Timeout
-  -f, --configFile []                   Path to file holding secret environment variables (e.g. "deploy.env")
-  -x, --contextFile [context.json]      Context JSON File
-  -M, --enableRunMultipleEvents [true]  Enable run multiple events
-  -y, --proxy []                        Proxy server
-  --apiGateway                          Convert to API Gateway events
+  -H, --handler [AWS_HANDLER]                                 Lambda Handler {index.handler} (default: "index.handler")
+  -j, --eventFile [EVENT_FILE]                                Event JSON File (default: "event.json")
+  -u, --runtime [AWS_RUNTIME]                                 Lambda Runtime (default: "nodejs14.x")
+  -t, --timeout [AWS_RUN_TIMEOUT]                             Lambda Timeout (default: 3)
+  -f, --configFile [CONFIG_FILE]                              Path to file holding secret environment variables (e.g. "deploy.env") (default: "")
+  -x, --contextFile [CONTEXT_FILE]                            Context JSON File (default: "context.json")
+  -M, --enableRunMultipleEvents [ENABLE_RUN_MULTIPLE_EVENTS]  Enable run multiple events (default: true)
+  -y, --proxy [PROXY]                                         Proxy server (default: "")
+  --apiGateway                                                Convert to API Gateway events (default: false)
+  -h, --help                                                  display help for command
 ```
 
 #### package
@@ -134,23 +133,26 @@ Bundles your application into a local zip file.
 
 ```
 $ node-lambda package --help
-
-Usage: package|zip [options]
+Usage: node-lambda package|zip [options]
 
 Create zipped package for Amazon Lambda deployment
 
-
 Options:
-  -h, --help                        output usage information
-  -A, --packageDirectory [build]    Local Package Directory
-  -I, --dockerImage []              Docker image for npm install
-  -n, --functionName [node-lambda]  Lambda FunctionName
-  -H, --handler [index.handler]     Lambda Handler {index.handler}
-  -e, --environment [development]   Choose environment {dev, staging, production}
-  -x, --excludeGlobs [event.json]   Space-separated glob pattern(s) for additional exclude files (e.g. "event.json dotenv.sample")
-  -D, --prebuiltDirectory []        Prebuilt directory
-  -m, --keepNodeModules [false]     Keep the current node_modules and skip the npm install step
-  -v, --dockerVolumes []            Additional docker volumes to mount. Each volume definition has to be separated by a space (e.g. "$HOME/.gitconfig:/etc/gitconfig $HOME/.ssh:/root/.ssh"
+  --packageManager [PACKAGE_MANAGER]            Package manager used to install dependencies (default: "npm", options: "npm", "yarn")
+  -A, --packageDirectory [PACKAGE_DIRECTORY]    Local Package Directory
+  -I, --dockerImage [DOCKER_IMAGE]              Docker image for npm ci (default: "")
+  -n, --functionName [AWS_FUNCTION_NAME]        Lambda FunctionName (default: "node-lambda")
+  -H, --handler [AWS_HANDLER]                   Lambda Handler {index.handler} (default: "index.handler")
+  -e, --environment [AWS_ENVIRONMENT]           Choose environment {dev, staging, production} (default: "")
+  -x, --excludeGlobs [EXCLUDE_GLOBS]            Space-separated glob pattern(s) for additional exclude files (e.g.
+                                                "event.json dotenv.sample") (default: "")
+  -D, --prebuiltDirectory [PREBUILT_DIRECTORY]  Prebuilt directory (default: "")
+  -m, --keepNodeModules [KEEP_NODE_MODULES]     Keep the current node_modules directory. (default: false)
+  -v, --dockerVolumes [DOCKER_VOLUMES]          Additional docker volumes to mount. Each volume definition has to be
+                                                separated by a space (e.g. "$HOME/.gitconfig:/etc/gitconfig
+                                                $HOME/.ssh:/root/.ssh") (default: "")
+  --no-optionalDependencies                     Run `npm install` with `--no-optional`
+  -h, --help                                    display help for command
 ```
 
 #### deploy
@@ -159,50 +161,50 @@ Bundles and deploys your application up to Amazon Lambda.
 
 ```
 $ node-lambda deploy --help
-
-Usage: deploy [options]
+Usage: node-lambda deploy [options]
 
 Deploy your application to Amazon Lambda
 
-
 Options:
-  -h, --help                          output usage information
-  -e, --environment [development]     Choose environment {dev, staging, production}
-  -E, --endpoint [AWS_ENDPOINT]       Choose endpoint (e.g. localstack, "http://127.0.0.1:4574") (default: "")
-  -a, --accessKey [your_key]          AWS Access Key
-  -s, --secretKey [your_secret]       AWS Secret Key
-  -P, --profile []                    AWS Profile
-  -k, --sessionToken []               AWS Session Token
-  -r, --region [us-east-1]            AWS Region
-  -n, --functionName [node-lambda]    Lambda FunctionName
-  -H, --handler [index.handler]       Lambda Handler {index.handler}
-  -o, --role [your_amazon_role]       Amazon Role ARN
-  -m, --memorySize [128]              Lambda Memory Size
-  -t, --timeout [3]                   Lambda Timeout
-  -d, --description [missing]         Lambda Description
-  -u, --runtime [nodejs14.x]          Lambda Runtime
-  -p, --publish [false]               Lambda Publish
-  -L, --lambdaVersion []              Lambda Function Version
-  -b, --vpcSubnets []                 Lambda Function VPC Subnet IDs (comma delimited)
-  -g, --vpcSecurityGroups []          Lambda VPC Security Group IDs (comma delimited)
-  -K, --kmsKeyArn []                  Lambda KMS Key ARN
-  -Q, --deadLetterConfigTargetArn []  Lambda DLQ resource
-  -c, --tracingConfig []              Lambda tracing settings
-  -l, --layers []                     Lambda Layers settings (e.g. "ARN1,ARN2[,..])" (default: "")
-  -R, --retentionInDays []            CloudWatchLogs retentionInDays settings
-  -G, --sourceDirectory []            Path to lambda source Directory (e.g. "./some-lambda")
-  -I, --dockerImage []                Docker image for npm install
-  -f, --configFile []                 Path to file holding secret environment variables (e.g. "deploy.env")
-  -S, --eventSourceFile []            Path to file holding event source mapping variables (e.g. "event_sources.json")
-  -x, --excludeGlobs [event.json]     Space-separated glob pattern(s) for additional exclude files (e.g. "event.json dotenv.sample")
-  -D, --prebuiltDirectory []          Prebuilt directory
-  -T, --deployTimeout [120000]        Deploy Timeout
-  -z, --deployZipfile []              Deploy zipfile
-  -B, --deployUseS3 []                Use S3 to deploy.
-  -i, --imageUri []                   URI of a container image in the Amazon ECR registry. (default: "")
-  -y, --proxy []                      Proxy server
-  -A, --tags []                       Tags as key value pairs (e.g. "tagname1=tagvalue1,tagname2=tagvalue2") (default: "")
-  --silent                            Silent  or  quiet mode (default: false)
+  --packageManager [PACKAGE_MANAGER]                    Package manager used to install dependencies (default: "npm", options: "npm", "yarn")
+  -e, --environment [AWS_ENVIRONMENT]                   Choose environment {dev, staging, production} (default: "")
+  -E, --endpoint [AWS_ENDPOINT]                         Choose endpoint (e.g. localstack, "http://127.0.0.1:4574") (default: "")
+  -a, --accessKey [AWS_ACCESS_KEY_ID]                   AWS Access Key
+  -s, --secretKey [AWS_SECRET_ACCESS_KEY]               AWS Secret Key
+  -P, --profile [AWS_PROFILE]                           AWS Profile (default: "")
+  -k, --sessionToken [AWS_SESSION_TOKEN]                AWS Session Token (default: "")
+  -r, --region [AWS_REGION]                             AWS Region (default: "us-east-1,us-west-2,eu-west-1")
+  -n, --functionName [AWS_FUNCTION_NAME]                Lambda FunctionName (default: "node-lambda")
+  -H, --handler [AWS_HANDLER]                           Lambda Handler {index.handler} (default: "index.handler")
+  -o, --role [AWS_ROLE]                                 Amazon Role ARN (default: "missing")
+  -m, --memorySize [AWS_MEMORY_SIZE]                    Lambda Memory Size (default: 128)
+  -t, --timeout [AWS_TIMEOUT]                           Lambda Timeout (default: 60)
+  -d, --description [AWS_DESCRIPTION]                   Lambda Description (default: "Command line tool for locally running and remotely deploying your node.js applications to Amazon Lambda.")
+  -u, --runtime [AWS_RUNTIME]                           Lambda Runtime (default: "nodejs14.x")
+  -p, --publish [AWS_PUBLISH]                           Lambda Publish (default: false)
+  -L, --lambdaVersion [AWS_FUNCTION_VERSION]            Lambda Function Version (default: "")
+  -b, --vpcSubnets [AWS_VPC_SUBNETS]                    Lambda Function VPC Subnet IDs (comma delimited) (default: "")
+  -g, --vpcSecurityGroups [AWS_VPC_SECURITY_GROUPS]     Lambda VPC Security Group IDs (comma delimited) (default: "")
+  -K, --kmsKeyArn [AWS_KMS_KEY_ARN]                     Lambda KMS Key ARN (default: "")
+  -Q, --deadLetterConfigTargetArn [AWS_DLQ_TARGET_ARN]  Lambda DLQ resource
+  -c, --tracingConfig [AWS_TRACING_CONFIG]              Lambda tracing settings (default: "")
+  -l, --layers [AWS_LAYERS]                             Lambda Layers settings (e.g. "ARN1,ARN2[,..])" (default: "")
+  -R, --retentionInDays [AWS_LOGS_RETENTION_IN_DAYS]    CloudWatchLogs retentionInDays settings (default: "")
+  -G, --sourceDirectory [SRC_DIRECTORY]                 Path to lambda source Directory (e.g. "./some-lambda") (default: "")
+  -I, --dockerImage [DOCKER_IMAGE]                      Docker image for npm ci (default: "")
+  -f, --configFile [CONFIG_FILE]                        Path to file holding secret environment variables (e.g. "deploy.env") (default: "")
+  -S, --eventSourceFile [EVENT_SOURCE_FILE]             Path to file holding event source mapping variables (e.g. "event_sources.json") (default: "")
+  -x, --excludeGlobs [EXCLUDE_GLOBS]                    Space-separated glob pattern(s) for additional exclude files (e.g. "event.json dotenv.sample") (default: "")
+  -D, --prebuiltDirectory [PREBUILT_DIRECTORY]          Prebuilt directory (default: "")
+  -T, --deployTimeout [DEPLOY_TIMEOUT]                  Deploy Timeout (default: 120000)
+  -z, --deployZipfile [DEPLOY_ZIPFILE]                  Deploy zipfile (default: "")
+  -B, --deployUseS3 [DEPLOY_USE_S3]                     Use S3 to deploy. (default: false)
+  -i, --imageUri [IMAGE_URI]                            URI of a container image in the Amazon ECR registry. (default: "")
+  -y, --proxy [PROXY]                                   Proxy server (default: "")
+  -A, --tags [AWS_TAGS]                                 Tags as key value pairs (e.g. "tagname1=tagvalue1,tagname2=tagvalue2)" (default: "")
+  --silent                                              Silent  or  quiet mode (default: false)
+  --no-optionalDependencies                             Run `npm install` with `--no-optional`
+  -h, --help                                            display help for command
 ```
 
 If you are deploying to a custom endpoint you may also need to pass in an access key/secret. For localstack these can be anything, but cannot be blank:
@@ -217,7 +219,7 @@ AWS Lambda will let you set environment variables for your function. Use the sam
 
 ## Node.js Runtime Configuration
 
-AWS Lambda now supports Node.js14, Node.js 12 and Node.js 10. Please also check the [Building Lambda functions with Node.js](https://docs.aws.amazon.com/lambda/latest/dg/lambda-nodejs.html) page.
+AWS Lambda now supports Node.js14 and Node.js 12. Please also check the [Lambda runtimes](https://docs.aws.amazon.com/lambda/latest/dg/lambda-runtimes.html) page.
 
 ## To deploy a container image to Lambda
 

package/bin/node-lambda

@@ -11,6 +11,7 @@ const fs = require('fs')
 const packageJson = fs.existsSync(path.join(process.cwd(), 'package.json')) ? require(path.join(process.cwd(), 'package.json')) : {}
 const packageJsonName = packageJson.name || 'UnnamedFunction'
 
+const PACKAGE_MANAGER = process.env.PACKAGE_MANAGER || 'npm'
 const AWS_ENVIRONMENT = process.env.AWS_ENVIRONMENT || ''
 const AWS_ENDPOINT = process.env.AWS_ENDPOINT || ''
 const CONFIG_FILE = process.env.CONFIG_FILE || ''
@@ -61,9 +62,12 @@ const DOCKER_VOLUMES = process.env.DOCKER_VOLUMES || ''
 const AWS_TAGS = process.env.AWS_TAGS || ''
 const IMAGE_URI = process.env.IMAGE_URI || ''
 
+program.showSuggestionAfterError()
+
 program
   .command('deploy')
   .description('Deploy your application to Amazon Lambda')
+  .option('--packageManager [PACKAGE_MANAGER]', 'Package manager used to install dependencies', PACKAGE_MANAGER)
   .option('-e, --environment [AWS_ENVIRONMENT]', 'Choose environment {dev, staging, production}',
     AWS_ENVIRONMENT)
   .option('-E, --endpoint [AWS_ENDPOINT]', 'Choose endpoint (e.g. localstack, "http://127.0.0.1:4574")',
@@ -109,6 +113,7 @@ program
   .option('-y, --proxy [PROXY]', 'Proxy server', PROXY)
   .option('-A, --tags [AWS_TAGS]', 'Tags as key value pairs (e.g. "tagname1=tagvalue1,tagname2=tagvalue2)"', AWS_TAGS)
   .option('--silent', 'Silent  or  quiet mode', false)
+  .option('--no-optionalDependencies', 'Run `npm install` with `--no-optional`')
   .action((prg) => {
     if (prg.silent) {
       console.log = () => {}
@@ -120,6 +125,7 @@ program
   .command('package')
   .alias('zip')
   .description('Create zipped package for Amazon Lambda deployment')
+  .option('--packageManager [PACKAGE_MANAGER]', 'Package manager used to install dependencies', PACKAGE_MANAGER)
   .option('-A, --packageDirectory [PACKAGE_DIRECTORY]', 'Local Package Directory', PACKAGE_DIRECTORY)
   .option('-I, --dockerImage [DOCKER_IMAGE]', 'Docker image for npm ci', DOCKER_IMAGE)
   .option('-n, --functionName [AWS_FUNCTION_NAME]', 'Lambda FunctionName', AWS_FUNCTION_NAME)
@@ -131,6 +137,7 @@ program
   .option('-D, --prebuiltDirectory [PREBUILT_DIRECTORY]', 'Prebuilt directory', PREBUILT_DIRECTORY)
   .option('-m, --keepNodeModules [KEEP_NODE_MODULES]', 'Keep the current node_modules directory.', KEEP_NODE_MODULES)
   .option('-v, --dockerVolumes [DOCKER_VOLUMES]', 'Additional docker volumes to mount. Each volume definition has to be separated by a space (e.g. "$HOME/.gitconfig:/etc/gitconfig $HOME/.ssh:/root/.ssh")', DOCKER_VOLUMES)
+  .option('--no-optionalDependencies', 'Run `npm install` with `--no-optional`')
   .action((prg) => lambda.package(prg))
 
 program
@@ -157,12 +164,6 @@ program
   .option('-x, --contextFile [CONTEXT_FILE]', 'Context JSON File', CONTEXT_FILE)
   .action((prg) => lambda.setup(prg))
 
-program.on('command:*', (cmd) => {
-  console.log(`Unknown command ${cmd}`)
-  console.log()
-  program.help()
-})
-
 program
   .version(lambda.version)
   .parse(process.argv)

package/lib/main.js

@@ -51,7 +51,7 @@ event_sources.json and ${program.eventFile} files as needed.`)
   }
 
   run (program) {
-    if (!['nodejs10.x', 'nodejs12.x', 'nodejs14.x'].includes(program.runtime)) {
+    if (!['nodejs12.x', 'nodejs14.x'].includes(program.runtime)) {
       console.error(`Runtime [${program.runtime}] is not supported.`)
       process.exit(254)
     }
@@ -385,73 +385,108 @@ Emulate only the body of the API Gateway event.
     return fs.existsSync(path.join(codeDirectory, 'package-lock.json'))
   }
 
-  _npmInstall (program, codeDirectory) {
-    const installCommand = this._shouldUseNpmCi(codeDirectory)
-      ? 'ci'
-      : 'install'
-
-    const dockerBaseOptions = [
-      'run', '--rm',
-      '-v', `${fs.realpathSync(codeDirectory)}:/var/task`,
-      '-w', '/var/task'
+  _getNpmInstallCommand (program, codeDirectory) {
+    const installOptions = [
+      '-s',
+      this._shouldUseNpmCi(codeDirectory) ? 'ci' : 'install',
+      '--production',
+      '--no-audit'
     ]
 
-    let dockerVolumesOptions = []
-    program.dockerVolumes && program.dockerVolumes.split(' ').forEach((volume) => {
-      dockerVolumesOptions = dockerVolumesOptions.concat(['-v', volume])
-    })
+    if (program.optionalDependencies === false) {
+      installOptions.push('--no-optional')
+    }
 
-    const dockerCommand = [program.dockerImage, 'npm', '-s', installCommand, '--production', '--no-audit']
+    if (!program.dockerImage) {
+      installOptions.push('--prefix', codeDirectory)
+    }
 
-    const dockerOptions = dockerBaseOptions.concat(dockerVolumesOptions).concat(dockerCommand)
+    return {
+      packageManager: 'npm',
+      installOptions
+    }
+  }
 
-    const npmInstallBaseOptions = [
+  _getYarnInstallCommand (program, codeDirectory) {
+    const installOptions = [
       '-s',
-      installCommand,
-      '--production',
-      '--no-audit',
-      '--prefix', codeDirectory
+      'install',
+      '--production'
     ]
 
-    const params = (() => {
-      // reference: https://nodejs.org/api/child_process.html#child_process_spawning_bat_and_cmd_files_on_windows
+    if (program.optionalDependencies === false) {
+      installOptions.push('--ignore-optional')
+    }
+
+    if (!program.dockerImage) {
+      installOptions.push('--cwd', codeDirectory)
+    }
+
+    return {
+      packageManager: 'yarn',
+      installOptions
+    }
+  }
+
+  _packageInstall (program, codeDirectory) {
+    // Run on windows:
+    // https://nodejs.org/api/child_process.html#child_process_spawning_bat_and_cmd_files_on_windows
 
+    const { packageManager, installOptions } = (() => {
+      // default npm
+      if (program.packageManager === 'yarn') {
+        return this._getYarnInstallCommand(program, codeDirectory)
+      }
+      return this._getNpmInstallCommand(program, codeDirectory)
+    })()
+
+    const paramsOnContainer = (() => {
       // with docker
-      if (program.dockerImage) {
-        if (process.platform === 'win32') {
-          return {
-            command: 'cmd.exe',
-            options: ['/c', 'docker'].concat(dockerOptions)
-          }
-        }
+      let dockerVolumesOptions = []
+      program.dockerVolumes && program.dockerVolumes.split(' ').forEach((volume) => {
+        dockerVolumesOptions = dockerVolumesOptions.concat(['-v', volume])
+      })
+      const dockerCommand = [program.dockerImage, packageManager].concat(installOptions)
+      const dockerBaseOptions = [
+        'run', '--rm',
+        '-v', `${fs.realpathSync(codeDirectory)}:/var/task`,
+        '-w', '/var/task'
+      ]
+      const dockerOptions = dockerBaseOptions.concat(dockerVolumesOptions).concat(dockerCommand)
+      if (process.platform === 'win32') {
         return {
-          command: 'docker',
-          options: dockerOptions
+          command: 'cmd.exe',
+          options: ['/c', 'docker'].concat(dockerOptions)
         }
       }
+      return {
+        command: 'docker',
+        options: dockerOptions
+      }
+    })()
 
-      // simple npm install
+    const paramsOnHost = (() => {
+      // simple install
       if (process.platform === 'win32') {
         return {
           command: 'cmd.exe',
-          options: ['/c', 'npm']
-            .concat(npmInstallBaseOptions)
-            .concat(['--cwd', codeDirectory])
+          options: ['/c', packageManager].concat(installOptions)
         }
       }
       return {
-        command: 'npm',
-        options: npmInstallBaseOptions
+        command: packageManager,
+        options: installOptions
       }
     })()
 
+    const params = program.dockerImage ? paramsOnContainer : paramsOnHost
     return new Promise((resolve, reject) => {
       execFile(params.command, params.options, {
         maxBuffer: maxBufferSize,
         env: process.env
       }, (err) => {
         if (err) return reject(err)
-        resolve()
+        resolve(packageManager)
       })
     })
   }
@@ -519,7 +554,7 @@ Emulate only the body of the API Gateway event.
 
   _codeDirectory () {
     // Why realpathSync?:
-    // If tmpdir is symbolic link and npm>=7, `this._npmInstall()` may not work properly.
+    // If tmpdir is symbolic link and npm>=7, `this._packageInstall()` may not work properly.
     return path.join(fs.realpathSync(os.tmpdir()), `${path.basename(path.resolve('.'))}-lambda`)
   }
 
@@ -531,28 +566,27 @@ Emulate only the body of the API Gateway event.
           resolve()
         })
       })
-    } else {
-      return new Promise((resolve, reject) => {
-        fs.readdir(codeDirectory, (err, files) => {
-          if (err) return reject(err)
+    }
+    return new Promise((resolve, reject) => {
+      fs.readdir(codeDirectory, (err, files) => {
+        if (err) return reject(err)
 
-          Promise.all(files.map(file => {
-            return new Promise((resolve, reject) => {
-              if (keepNodeModules && file === 'node_modules') {
+        Promise.all(files.map(file => {
+          return new Promise((resolve, reject) => {
+            if (keepNodeModules && file === 'node_modules') {
+              resolve()
+            } else {
+              fs.remove(path.join(codeDirectory, file), err => {
+                if (err) return reject(err)
                 resolve()
-              } else {
-                fs.remove(path.join(codeDirectory, file), err => {
-                  if (err) return reject(err)
-                  resolve()
-                })
-              }
-            })
-          })).then(() => {
-            resolve()
+              })
+            }
           })
+        })).then(() => {
+          resolve()
         })
       })
-    }
+    })
   }
 
   _setRunTimeEnvironmentVars (program) {
@@ -568,7 +602,7 @@ Emulate only the body of the API Gateway event.
     }
   }
 
-  _uploadExisting (lambda, params) {
+  async _uploadExisting (lambda, params) {
     const functionCodeParams = Object.assign({
       FunctionName: params.FunctionName,
       Publish: params.Publish
@@ -596,29 +630,30 @@ Emulate only the body of the API Gateway event.
       delete functionConfigParams.Layers
     }
 
-    return new Promise((resolve, reject) => {
-      const updateConfigRequest = lambda.updateFunctionConfiguration(
-        functionConfigParams,
-        (err, configResponse) => {
-          if (err) return reject(err)
-
-          const updateCodeRequest = lambda.updateFunctionCode(functionCodeParams, (err) => {
-            if (err) return reject(err)
-            resolve(configResponse)
-          })
+    const updateConfigRequest = lambda.updateFunctionConfiguration(functionConfigParams)
+    updateConfigRequest.on('retry', (response) => {
+      console.log(response.error.message)
+      console.log('=> Retrying')
+    })
+    const updateConfigResponse = await updateConfigRequest.promise()
 
-          updateCodeRequest.on('retry', (response) => {
-            console.log(response.error.message)
-            console.log('=> Retrying')
-          })
-        }
-      )
+    // Wait for the `Configuration.LastUpdateStatus` to change from `InProgress` to `Successful`.
+    for (let i = 0; i < 10; i++) {
+      const data = await lambda.getFunction({ FunctionName: params.FunctionName }).promise()
+      if (data.Configuration.LastUpdateStatus === 'Successful') {
+        break
+      }
+      await new Promise((resolve) => setTimeout(resolve, 3000))
+    }
 
-      updateConfigRequest.on('retry', (response) => {
-        console.log(response.error.message)
-        console.log('=> Retrying')
-      })
+    const updateCodeRequest = lambda.updateFunctionCode(functionCodeParams)
+    updateCodeRequest.on('retry', (response) => {
+      console.log(response.error.message)
+      console.log('=> Retrying')
     })
+    await updateCodeRequest.promise()
+
+    return updateConfigResponse
   }
 
   _uploadNew (lambda, params) {
@@ -659,13 +694,17 @@ Emulate only the body of the API Gateway event.
   _archivePrebuilt (program) {
     const codeDirectory = this._codeDirectory()
 
-    return this._fileCopy(program, program.prebuiltDirectory, codeDirectory, false).then(() => {
-      console.log('=> Zipping deployment package')
-      return this._zip(program, codeDirectory)
+    return Promise.resolve().then(() => {
+      return this._cleanDirectory(codeDirectory, program.keepNodeModules)
+    }).then(() => {
+      return this._fileCopy(program, program.prebuiltDirectory, codeDirectory, false).then(() => {
+        console.log('=> Zipping deployment package')
+        return this._zip(program, codeDirectory)
+      })
     })
   }
 
-  _buildAndArchive (program) {
+  async _buildAndArchive (program) {
     if (!fs.existsSync('.env')) {
       console.warn('[Warning] `.env` file does not exist.')
       console.info('Execute `node-lambda setup` as necessary and set it up.')
@@ -684,24 +723,17 @@ they may not work as expected in the Lambda environment.
     const codeDirectory = this._codeDirectory()
     const lambdaSrcDirectory = program.sourceDirectory ? program.sourceDirectory.replace(/\/$/, '') : '.'
 
-    return Promise.resolve().then(() => {
-      return this._cleanDirectory(codeDirectory, program.keepNodeModules)
-    }).then(() => {
-      console.log('=> Moving files to temporary directory')
-      return this._fileCopy(program, lambdaSrcDirectory, codeDirectory, true)
-    }).then(() => {
-      if (program.keepNodeModules) {
-        return Promise.resolve()
-      } else {
-        console.log(`=> Running npm ${this._shouldUseNpmCi(codeDirectory) ? 'ci' : 'install'} --production`)
-        return this._npmInstall(program, codeDirectory)
-      }
-    }).then(() => {
-      return this._postInstallScript(program, codeDirectory)
-    }).then(() => {
-      console.log('=> Zipping deployment package')
-      return this._zip(program, codeDirectory)
-    })
+    await this._cleanDirectory(codeDirectory, program.keepNodeModules)
+    console.log('=> Moving files to temporary directory')
+    await this._fileCopy(program, lambdaSrcDirectory, codeDirectory, true)
+    if (!program.keepNodeModules) {
+      console.log('=> Running package install')
+      const usedPackageManager = await this._packageInstall(program, codeDirectory)
+      console.log(`(Package manager used was '${usedPackageManager}'.)`)
+    }
+    await this._postInstallScript(program, codeDirectory)
+    console.log('=> Zipping deployment package')
+    return this._zip(program, codeDirectory)
   }
 
   _listEventSourceMappings (lambda, params) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "node-lambda",
-  "version": "0.18.0",
+  "version": "0.21.0",
   "description": "Command line tool for locally running and remotely deploying your node.js applications to Amazon Lambda.",
   "main": "lib/main.js",
   "directories": {
@@ -32,25 +32,25 @@
   "author": "motdotla",
   "license": "BSD-2-Clause",
   "devDependencies": {
-    "aws-sdk-mock": "^5.1.0",
-    "chai": "^4.2.0",
-    "mocha": "^8.2.1",
+    "aws-sdk-mock": "^5.2.1",
+    "chai": "^4.3.4",
+    "mocha": "^9.0.1",
     "node-zip": "^1.1.1",
-    "standard": "^16.0.1"
+    "standard": "^16.0.3"
+  },
+  "engines": {
+    "node": ">= 10.0.0"
   },
   "dependencies": {
-    "archiver": "^5.0.0",
-    "aws-sdk": "^2.831.0",
-    "aws-xray-sdk-core": "^3.1.0",
-    "commander": "^7.0.0",
+    "archiver": "^5.3.0",
+    "aws-sdk": "^2.936.0",
+    "aws-xray-sdk-core": "^3.3.3",
+    "commander": "^8.2.0",
     "continuation-local-storage": "^3.2.1",
-    "dotenv": "^8.2.0",
-    "fs-extra": "^9.0.1",
+    "dotenv": "^10.0.0",
+    "fs-extra": "^10.0.0",
     "klaw": "^3.0.0",
     "minimatch": "^3.0.4",
-    "proxy-agent": "^3.1.1"
-  },
-  "engines": {
-    "node": ">= 10.0.0"
+    "proxy-agent": "^5.0.0"
   }
 }

package/test/main.js

@@ -10,6 +10,7 @@ const awsMock = require('aws-sdk-mock')
 awsMock.setSDK(path.resolve('node_modules/aws-sdk'))
 
 const originalProgram = {
+  packageManager: 'npm',
   environment: 'development',
   accessKey: 'key',
   secretKey: 'secret',
@@ -31,17 +32,18 @@ const originalProgram = {
   contextFile: 'context.json',
   deployTimeout: 120000,
   prebuiltDirectory: '',
-  proxy: ''
+  proxy: '',
+  optionalDependencies: true
 }
 
 let program = {}
 let codeDirectory = lambda._codeDirectory()
 
 const _timeout = function (params) {
-  // Even if timeout is set for the whole test for Windows,
+  // Even if timeout is set for the whole test for Windows and Mac,
   // if it is set in local it will be valid.
-  // For Windows, do not set it with local.
-  if (process.platform !== 'win32') {
+  // For Windows and Mac, do not set it with local.
+  if (!['win32', 'darwin'].includes(process.platform)) {
     params.this.timeout(params.sec * 1000)
   }
 }
@@ -51,7 +53,7 @@ const lambdaMockSettings = {
   addPermission: {},
   getFunction: {
     Code: {},
-    Configuration: {},
+    Configuration: { LastUpdateStatus: 'Successful' },
     FunctionArn: 'Lambda.getFunction.mock.FunctionArn'
   },
   createFunction: {
@@ -129,8 +131,8 @@ const _awsRestore = () => {
 
 /* global before, after, beforeEach, afterEach, describe, it */
 describe('lib/main', function () {
-  if (process.platform === 'win32') {
-    // It seems that it takes time for file operation in Windows.
+  if (['win32', 'darwin'].includes(process.platform)) {
+    // It seems that it takes time for file operation in Windows and Mac.
     // So set `timeout(60000)` for the whole test.
     this.timeout(60000)
   }
@@ -148,7 +150,7 @@ describe('lib/main', function () {
   })
 
   it('version should be set', () => {
-    assert.equal(lambda.version, '0.18.0')
+    assert.equal(lambda.version, '0.21.0')
   })
 
   describe('_codeDirectory', () => {
@@ -589,65 +591,251 @@ describe('lib/main', function () {
     })
   })
 
-  describe('_npmInstall', function () {
-    _timeout({ this: this, sec: 30 }) // ci should be faster than install
+  describe('_getNpmInstallCommand', () => {
+    describe('when package-lock.json exists', () => {
+      const codeDirectory = '.'
+
+      it('npm ci', () => {
+        const { packageManager, installOptions } = lambda._getNpmInstallCommand(program, codeDirectory)
+        assert.equal(packageManager, 'npm')
+        assert.deepEqual(installOptions, ['-s', 'ci', '--production', '--no-audit', '--prefix', codeDirectory])
+      })
+
+      it('npm ci with "--no-optional"', () => {
+        const { packageManager, installOptions } = lambda._getNpmInstallCommand(
+          {
+            ...program,
+            optionalDependencies: false
+          },
+          codeDirectory
+        )
+        assert.equal(packageManager, 'npm')
+        assert.deepEqual(
+          installOptions,
+          ['-s', 'ci', '--production', '--no-audit', '--no-optional', '--prefix', codeDirectory]
+        )
+      })
+
+      it('npm ci on docker', () => {
+        const { packageManager, installOptions } = lambda._getNpmInstallCommand(
+          {
+            ...program,
+            dockerImage: 'test'
+          },
+          codeDirectory
+        )
+        assert.equal(packageManager, 'npm')
+        assert.deepEqual(installOptions, ['-s', 'ci', '--production', '--no-audit'])
+      })
+    })
+
+    describe('when package-lock.json does not exist', () => {
+      const codeDirectory = './test'
+
+      it('npm install', () => {
+        const { packageManager, installOptions } = lambda._getNpmInstallCommand(program, './test')
+        assert.equal(packageManager, 'npm')
+        assert.deepEqual(installOptions, ['-s', 'install', '--production', '--no-audit', '--prefix', './test'])
+      })
+
+      it('npm install with "--no-optional"', () => {
+        const { packageManager, installOptions } = lambda._getNpmInstallCommand(
+          {
+            ...program,
+            optionalDependencies: false
+          },
+          codeDirectory
+        )
+        assert.equal(packageManager, 'npm')
+        assert.deepEqual(
+          installOptions,
+          ['-s', 'install', '--production', '--no-audit', '--no-optional', '--prefix', codeDirectory]
+        )
+      })
+
+      it('npm install on docker', () => {
+        const { packageManager, installOptions } = lambda._getNpmInstallCommand(
+          {
+            ...program,
+            dockerImage: 'test'
+          },
+          codeDirectory
+        )
+        assert.equal(packageManager, 'npm')
+        assert.deepEqual(installOptions, ['-s', 'install', '--production', '--no-audit'])
+      })
+    })
+  })
+
+  describe('_getYarnInstallCommand', () => {
+    const codeDirectory = '.'
+
+    it('yarn install', () => {
+      const { packageManager, installOptions } = lambda._getYarnInstallCommand(program, codeDirectory)
+      assert.equal(packageManager, 'yarn')
+      assert.deepEqual(installOptions, ['-s', 'install', '--production', '--cwd', codeDirectory])
+    })
+
+    it('yarn install with "--no-optional"', () => {
+      const { packageManager, installOptions } = lambda._getYarnInstallCommand(
+        {
+          ...program,
+          optionalDependencies: false
+        },
+        codeDirectory
+      )
+      assert.equal(packageManager, 'yarn')
+      assert.deepEqual(
+        installOptions,
+        ['-s', 'install', '--production', '--ignore-optional', '--cwd', codeDirectory]
+      )
+    })
+
+    it('yarn install on docker', () => {
+      const { packageManager, installOptions } = lambda._getYarnInstallCommand(
+        {
+          ...program,
+          dockerImage: 'test'
+        },
+        codeDirectory
+      )
+      assert.equal(packageManager, 'yarn')
+      assert.deepEqual(installOptions, ['-s', 'install', '--production'])
+    })
+  })
+
+  describe('_packageInstall', function () {
+    _timeout({ this: this, sec: 60 }) // ci should be faster than install
 
     // npm treats files as packages when installing, and so removes them.
     // Test with `devDependencies` packages that are not installed with the `--production` option.
     const nodeModulesMocha = path.join(codeDirectory, 'node_modules', 'mocha')
 
-    beforeEach(() => {
-      return lambda._cleanDirectory(codeDirectory).then(() => {
-        fs.copySync(
-          path.join('node_modules', 'aws-sdk'),
-          path.join(codeDirectory, 'node_modules', 'aws-sdk')
-        )
-        return lambda._fileCopy(program, '.', codeDirectory, true)
-      })
+    const testCleanAndInstall = async (packageManager) => {
+      const beforeAwsSdkStat = fs.statSync(path.join(codeDirectory, 'node_modules', 'aws-sdk'))
+
+      const usedPackageManager = await lambda._packageInstall(
+        {
+          ...program,
+          packageManager
+        },
+        codeDirectory
+      )
+      assert.equal(usedPackageManager, packageManager)
+
+      const contents = fs.readdirSync(path.join(codeDirectory, 'node_modules'))
+      assert.include(contents, 'dotenv')
+
+      // To remove and then install.
+      // beforeAwsSdkStat.ctimeMs < afterAwsSdkStat.ctimeMs
+      const afterAwsSdkStat = fs.statSync(path.join(codeDirectory, 'node_modules', 'aws-sdk'))
+      assert.isBelow(beforeAwsSdkStat.ctimeMs, afterAwsSdkStat.ctimeMs)
+
+      // Not installed with the `--production` option.
+      assert.isFalse(fs.existsSync(nodeModulesMocha))
+    }
+
+    const beforeEachOptionalDependencies = () => {
+      const packageJsonPath = path.join(codeDirectory, 'package.json')
+      const packageJson = require(packageJsonPath)
+      packageJson.optionalDependencies = { npm: '*' }
+      fs.writeFileSync(packageJsonPath, JSON.stringify(packageJson))
+    }
+
+    const testOptionalDependenciesIsInstalled = async (packageManager) => {
+      const usedPackageManager = await lambda._packageInstall(
+        {
+          ...program,
+          packageManager
+        },
+        codeDirectory
+      )
+      assert.equal(usedPackageManager, packageManager)
+
+      const contents = fs.readdirSync(path.join(codeDirectory, 'node_modules'))
+      assert.include(contents, 'npm')
+    }
+
+    const testOptionalDependenciesIsNotInstalled = async (packageManager) => {
+      const params = {
+        ...program,
+        packageManager,
+        optionalDependencies: false
+      }
+      const usedPackageManager = await lambda._packageInstall(params, codeDirectory)
+      assert.equal(usedPackageManager, packageManager)
+
+      const contents = fs.readdirSync(path.join(codeDirectory, 'node_modules'))
+      assert.notInclude(contents, 'npm')
+    }
+
+    beforeEach(async () => {
+      await lambda._cleanDirectory(codeDirectory)
+      await lambda._fileCopy(program, '.', codeDirectory, false)
     })
 
-    describe('when package-lock.json does exist', () => {
-      it('should use "npm ci"', () => {
-        const beforeAwsSdkStat = fs.statSync(path.join(codeDirectory, 'node_modules', 'aws-sdk'))
-        return lambda._npmInstall(program, codeDirectory).then(() => {
-          const contents = fs.readdirSync(path.join(codeDirectory, 'node_modules'))
-          assert.include(contents, 'dotenv')
+    describe('Use npm', () => {
+      describe('when package-lock.json does exist', () => {
+        it('should use "npm ci"', () => testCleanAndInstall('npm'))
+      })
+
+      describe('when package-lock.json does not exist', () => {
+        beforeEach(() => {
+          return fs.removeSync(path.join(codeDirectory, 'package-lock.json'))
+        })
+
+        it('should use "npm install"', () => {
+          const beforeAwsSdkStat = fs.statSync(path.join(codeDirectory, 'node_modules', 'aws-sdk'))
+          return lambda._packageInstall(program, codeDirectory).then((usedPackageManager) => {
+            assert.equal(usedPackageManager, 'npm')
 
-          // To remove and then install.
-          // beforeAwsSdkStat.ctimeMs < afterAwsSdkStat.ctimeMs
-          const afterAwsSdkStat = fs.statSync(path.join(codeDirectory, 'node_modules', 'aws-sdk'))
-          assert.isBelow(beforeAwsSdkStat.ctimeMs, afterAwsSdkStat.ctimeMs)
+            const contents = fs.readdirSync(path.join(codeDirectory, 'node_modules'))
+            assert.include(contents, 'dotenv')
 
-          // Not installed with the `--production` option.
-          assert.isFalse(fs.existsSync(nodeModulesMocha))
+            // Installed packages will remain intact.
+            // beforeAwsSdkStat.ctimeMs === afterAwsSdkStat.ctimeMs
+            const afterAwsSdkStat = fs.statSync(path.join(codeDirectory, 'node_modules', 'aws-sdk'))
+            assert.equal(beforeAwsSdkStat.ctimeMs, afterAwsSdkStat.ctimeMs)
+
+            // Not installed with the `--production` option.
+            assert.isFalse(fs.existsSync(nodeModulesMocha))
+          })
         })
       })
-    })
 
-    describe('when package-lock.json does not exist', () => {
-      beforeEach(() => {
-        return fs.removeSync(path.join(codeDirectory, 'package-lock.json'))
+      describe('optionalDependencies', () => {
+        beforeEach(beforeEachOptionalDependencies)
+
+        describe('No `--no-optionalDependencies`', () => {
+          it('optionalDependencies is installed', () => testOptionalDependenciesIsInstalled('npm'))
+        })
+
+        describe('With `--no-optionalDependencies`', () => {
+          it('optionalDependency is NOT installed', () => testOptionalDependenciesIsNotInstalled('npm'))
+        })
       })
+    })
 
-      it('should use "npm install"', () => {
-        const beforeAwsSdkStat = fs.statSync(path.join(codeDirectory, 'node_modules', 'aws-sdk'))
-        return lambda._npmInstall(program, codeDirectory).then(() => {
-          const contents = fs.readdirSync(path.join(codeDirectory, 'node_modules'))
-          assert.include(contents, 'dotenv')
+    describe('Use yarn', () => {
+      it('should use "yarn install"', () => testCleanAndInstall('yarn'))
 
-          // Installed packages will remain intact.
-          // beforeAwsSdkStat.ctimeMs === afterAwsSdkStat.ctimeMs
-          const afterAwsSdkStat = fs.statSync(path.join(codeDirectory, 'node_modules', 'aws-sdk'))
-          assert.equal(beforeAwsSdkStat.ctimeMs, afterAwsSdkStat.ctimeMs)
+      describe('optionalDependencies', () => {
+        beforeEach(beforeEachOptionalDependencies)
+
+        describe('No `--ignore-optionalDependencies`', () => {
+          it('optionalDependencies is installed', () => testOptionalDependenciesIsInstalled('yarn'))
+        })
 
-          // Not installed with the `--production` option.
-          assert.isFalse(fs.existsSync(nodeModulesMocha))
+        describe('With `--ignore-optionalDependencies`', () => {
+          it('optionalDependency is NOT installed', () => testOptionalDependenciesIsNotInstalled('yarn'))
         })
       })
     })
   })
 
-  describe('_npmInstall (When codeDirectory contains characters to be escaped)', () => {
+  describe('_packageInstall (When codeDirectory contains characters to be escaped)', function () {
+    _timeout({ this: this, sec: 30 }) // give it time to build the node modules
+
     beforeEach(() => {
       // Since '\' can not be included in the file or directory name in Windows
       const directoryName = process.platform === 'win32'
@@ -664,14 +852,21 @@ describe('lib/main', function () {
       codeDirectory = lambda._codeDirectory()
     })
 
-    it('_npm adds node_modules', function () {
-      _timeout({ this: this, sec: 30 }) // give it time to build the node modules
+    const testFunc = async (packageManager) => {
+      const usedPackageManager = await lambda._packageInstall(
+        {
+          ...program,
+          packageManager
+        },
+        codeDirectory
+      )
+      assert.equal(usedPackageManager, packageManager)
+      const contents = fs.readdirSync(codeDirectory)
+      assert.include(contents, 'node_modules')
+    }
 
-      return lambda._npmInstall(program, codeDirectory).then(() => {
-        const contents = fs.readdirSync(codeDirectory)
-        assert.include(contents, 'node_modules')
-      })
-    })
+    it('npm adds node_modules', () => testFunc('npm'))
+    it('yarn adds node_modules', () => testFunc('yarn'))
   })
 
   describe('_postInstallScript', () => {
@@ -741,53 +936,70 @@ describe('lib/main', function () {
     })
   })
 
-  describe('_zip', () => {
-    beforeEach(function () {
-      _timeout({ this: this, sec: 30 }) // give it time to build the node modules
-      return Promise.resolve().then(() => {
-        return lambda._cleanDirectory(codeDirectory)
-      }).then(() => {
-        return lambda._fileCopy(program, '.', codeDirectory, true)
-      }).then(() => {
-        return lambda._npmInstall(program, codeDirectory)
-      }).then(() => {
-        if (process.platform !== 'win32') {
-          fs.symlinkSync(
-            path.join(__dirname, '..', 'bin', 'node-lambda'),
-            path.join(codeDirectory, 'node-lambda-link')
-          )
-        }
-      })
-    })
+  describe('_zip', function () {
+    _timeout({ this: this, sec: 60 }) // give it time to zip
 
-    it('Compress the file. `index.js` and `bin/node-lambda` are included and the permission is also preserved.', function () {
-      _timeout({ this: this, sec: 30 }) // give it time to zip
+    const beforeTask = async (packageManager) => {
+      await lambda._cleanDirectory(codeDirectory)
+      await lambda._fileCopy(program, '.', codeDirectory, true)
+      const usedPackageManager = await lambda._packageInstall(
+        {
+          ...program,
+          packageManager
+        },
+        codeDirectory
+      )
+      assert.equal(usedPackageManager, packageManager)
+      if (process.platform !== 'win32') {
+        fs.symlinkSync(
+          path.join(__dirname, '..', 'bin', 'node-lambda'),
+          path.join(codeDirectory, 'node-lambda-link')
+        )
+      }
+    }
 
-      return lambda._zip(program, codeDirectory).then((data) => {
-        const archive = new Zip(data)
-        assert.include(archive.files['index.js'].name, 'index.js')
-        assert.include(archive.files['bin/node-lambda'].name, 'bin/node-lambda')
-
-        if (process.platform !== 'win32') {
-          const indexJsStat = fs.lstatSync('index.js')
-          const binNodeLambdaStat = fs.lstatSync(path.join('bin', 'node-lambda'))
-          assert.equal(
-            archive.files['index.js'].unixPermissions,
-            indexJsStat.mode
-          )
-          assert.equal(
-            archive.files['bin/node-lambda'].unixPermissions,
-            binNodeLambdaStat.mode
-          )
-
-          // isSymbolicLink
-          assert.include(archive.files['node-lambda-link'].name, 'node-lambda-link')
-          assert.equal(
-            archive.files['node-lambda-link'].unixPermissions & fs.constants.S_IFMT,
-            fs.constants.S_IFLNK
-          )
-        }
-      })
+    const testFunc = async (packageManager) => {
+      // setup
+      await beforeTask(packageManager)
+
+      // tests
+      const data = await lambda._zip(program, codeDirectory)
+      const archive = new Zip(data)
+      assert.include(archive.files['index.js'].name, 'index.js')
+      assert.include(archive.files['bin/node-lambda'].name, 'bin/node-lambda')
+
+      if (process.platform !== 'win32') {
+        const indexJsStat = fs.lstatSync('index.js')
+        const binNodeLambdaStat = fs.lstatSync(path.join('bin', 'node-lambda'))
+        assert.equal(
+          archive.files['index.js'].unixPermissions,
+          indexJsStat.mode
+        )
+        assert.equal(
+          archive.files['bin/node-lambda'].unixPermissions,
+          binNodeLambdaStat.mode
+        )
+
+        // isSymbolicLink
+        assert.include(archive.files['node-lambda-link'].name, 'node-lambda-link')
+        assert.equal(
+          archive.files['node-lambda-link'].unixPermissions & fs.constants.S_IFMT,
+          fs.constants.S_IFLNK
+        )
+      }
+    }
+
+    describe('Use npm', () => {
+      it(
+        'Compress the file. `index.js` and `bin/node-lambda` are included and the permission is also preserved.',
+        () => testFunc('npm')
+      )
+    })
+    describe('Use yarn', () => {
+      it(
+        'Compress the file. `index.js` and `bin/node-lambda` are included and the permission is also preserved.',
+        () => testFunc('yarn')
+      )
     })
   })
 
@@ -833,6 +1045,24 @@ describe('lib/main', function () {
         })
       })
     })
+
+    it('cleans the temporary directory before running `_archivePrebuilt`', function () {
+      _timeout({ this: this, sec: 30 }) // give it time to zip
+      const buildDir = '.build_' + Date.now()
+      const codeDir = lambda._codeDirectory()
+      const tmpFile = path.join(codeDir, 'deleteme')
+      after(() => fs.removeSync(buildDir))
+
+      fs.mkdirSync(codeDir, { recursive: true })
+      fs.writeFileSync(tmpFile, '...')
+      fs.mkdirSync(buildDir)
+      fs.writeFileSync(path.join(buildDir, 'test'), '...')
+
+      program.prebuiltDirectory = buildDir
+      return lambda._archive(program).then((_data) => {
+        assert.isNotTrue(fs.existsSync(tmpFile))
+      })
+    })
   })
 
   describe('_readArchive', () => {