test(NODE-4160): add evg run for lambda example

Author: durran

.evergreen/config.in.yml

@@ -818,6 +818,40 @@ functions:
         args:
           - "${PROJECT_DIRECTORY}/.evergreen/run-snappy-version-test.sh"
 
+  "run lambda tests":
+    - command: subprocess.exec
+      params:
+        working_dir: "src"
+        timeout_secs: 60
+        env:
+          MONGODB_URI: ${MONGODB_URI}
+          PROJECT_DIRECTORY: ${PROJECT_DIRECTORY}
+        binary: bash
+        args:
+          - "${PROJECT_DIRECTORY}/.evergreen/run-lambda-tests.sh"
+
+  "run lambda tests with aws auth":
+    - command: shell.exec
+      type: test
+      params:
+        working_dir: "src"
+        silent: true
+        script: |
+          cd ${DRIVERS_TOOLS}/.evergreen/auth_aws
+          ${MONGODB_BINARIES}/mongo --verbose aws_e2e_regular_aws.js
+          cd -
+          cat <<EOF > "${PROJECT_DIRECTORY}/prepare_mongodb_aws.sh"
+            export AWS_ACCESS_KEY_ID=${iam_auth_ecs_account}
+            export AWS_SECRET_ACCESS_KEY=${iam_auth_ecs_secret_access_key}
+          EOF
+    - command: shell.exec
+      type: test
+      params:
+        working_dir: "src"
+        script: |
+          ${PREPARE_SHELL}
+          ${PROJECT_DIRECTORY}/.evergreen/run-lambda-aws-tests.sh
+
   "run bson-ext test":
     - command: subprocess.exec
       type: test

.evergreen/config.yml

@@ -782,6 +782,38 @@ functions:
         binary: bash
         args:
           - ${PROJECT_DIRECTORY}/.evergreen/run-snappy-version-test.sh
+  run lambda tests:
+    - command: subprocess.exec
+      params:
+        working_dir: src
+        timeout_secs: 60
+        env:
+          MONGODB_URI: ${MONGODB_URI}
+          PROJECT_DIRECTORY: ${PROJECT_DIRECTORY}
+        binary: bash
+        args:
+          - ${PROJECT_DIRECTORY}/.evergreen/run-lambda-tests.sh
+  run lambda tests with aws auth:
+    - command: shell.exec
+      type: test
+      params:
+        working_dir: src
+        silent: true
+        script: |
+          cd ${DRIVERS_TOOLS}/.evergreen/auth_aws
+          ${MONGODB_BINARIES}/mongo --verbose aws_e2e_regular_aws.js
+          cd -
+          cat <<EOF > "${PROJECT_DIRECTORY}/prepare_mongodb_aws.sh"
+            export AWS_ACCESS_KEY_ID=${iam_auth_ecs_account}
+            export AWS_SECRET_ACCESS_KEY=${iam_auth_ecs_secret_access_key}
+          EOF
+    - command: shell.exec
+      type: test
+      params:
+        working_dir: src
+        script: |
+          ${PREPARE_SHELL}
+          ${PROJECT_DIRECTORY}/.evergreen/run-lambda-aws-tests.sh
   run bson-ext test:
     - command: subprocess.exec
       type: test
@@ -1253,6 +1285,32 @@ tasks:
           AUTH: auth
           COMPRESSOR: snappy
       - func: run-compression-tests
+  - name: test-lambda-example
+    tags:
+      - latest
+      - lambda
+    commands:
+      - func: install dependencies
+      - func: bootstrap mongo-orchestration
+        vars:
+          VERSION: latest
+          TOPOLOGY: server
+      - func: run lambda tests
+  - name: test-lambda-aws-auth-example
+    tags:
+      - latest
+      - lambda
+    commands:
+      - func: install dependencies
+      - func: bootstrap mongo-orchestration
+        vars:
+          VERSION: latest
+          AUTH: auth
+          ORCHESTRATION_FILE: auth-aws.json
+          TOPOLOGY: server
+      - func: add aws auth variables to file
+      - func: setup aws env
+      - func: run lambda tests with aws auth
   - name: test-tls-support-latest
     tags:
       - tls-support
@@ -2162,6 +2220,8 @@ buildvariants:
       - test-socks5-tls
       - test-zstd-compression
       - test-snappy-compression
+      - test-lambda-example
+      - test-lambda-aws-auth-example
       - test-tls-support-latest
       - test-tls-support-6.0
       - test-tls-support-5.0
@@ -2221,6 +2281,8 @@ buildvariants:
       - test-socks5-tls
       - test-zstd-compression
       - test-snappy-compression
+      - test-lambda-example
+      - test-lambda-aws-auth-example
       - test-tls-support-latest
       - test-tls-support-6.0
       - test-tls-support-5.0
@@ -2274,6 +2336,8 @@ buildvariants:
       - test-socks5-tls
       - test-zstd-compression
       - test-snappy-compression
+      - test-lambda-example
+      - test-lambda-aws-auth-example
       - test-tls-support-latest
       - test-tls-support-6.0
       - test-tls-support-5.0

.evergreen/generate_evergreen_tasks.js

@@ -247,6 +247,44 @@ TASKS.push(
   });
 });
 
+// Add task for testing lambda example without aws auth.
+TASKS.push({
+  name: 'test-lambda-example',
+  tags: ['latest', 'lambda'],
+  commands: [
+    { func: 'install dependencies' },
+    {
+      func: 'bootstrap mongo-orchestration',
+      vars: {
+        VERSION: 'latest',
+        TOPOLOGY: 'server'
+      }
+    },
+    { func: 'run lambda tests' }
+  ]
+});
+
+// Add task for testing lambda example with aws auth.
+TASKS.push({
+  name: 'test-lambda-aws-auth-example',
+  tags: ['latest', 'lambda'],
+  commands: [
+    { func: 'install dependencies' },
+    {
+      func: 'bootstrap mongo-orchestration',
+      vars: {
+        VERSION: 'latest',
+        AUTH: 'auth',
+        ORCHESTRATION_FILE: 'auth-aws.json',
+        TOPOLOGY: 'server'
+      }
+    },
+    { func: 'add aws auth variables to file' },
+    { func: 'setup aws env' },
+    { func: 'run lambda tests with aws auth' }
+  ]
+});
+
 TLS_VERSIONS.forEach(VERSION => {
   TASKS.push({
     name: `test-tls-support-${VERSION}`,

.evergreen/run-lambda-aws-tests.sh

@@ -0,0 +1,24 @@
+#!/bin/bash
+# set -o xtrace   # Write all commands first to stderr
+set -o errexit  # Exit the script with error if any of the commands fail
+
+MONGODB_URI=${MONGODB_URI:-}
+
+# ensure no secrets are printed in log files
+set +x
+
+# load node.js environment
+source "${PROJECT_DIRECTORY}/.evergreen/init-nvm.sh"
+
+# the default connection string, may be overridden by the environment script
+export MONGODB_URI="mongodb://localhost:27017/aws"
+
+# load the script
+shopt -s expand_aliases # needed for `urlencode` alias
+[ -s "$PROJECT_DIRECTORY/prepare_mongodb_aws.sh" ] && source "$PROJECT_DIRECTORY/prepare_mongodb_aws.sh"
+
+# revert to show test output
+set -x
+
+npm install aws4
+npm run check:lambda:aws

.evergreen/run-lambda-tests.sh

@@ -0,0 +1,13 @@
+#!/bin/bash
+# set -o xtrace   # Write all commands first to stderr
+set -o errexit  # Exit the script with error if any of the commands fail
+
+MONGODB_URI=${MONGODB_URI:-}
+
+# ensure no secrets are printed in log files
+set +x
+
+# load node.js environment
+source "${PROJECT_DIRECTORY}/.evergreen/init-nvm.sh"
+
+npm run check:lambda

package.json

@@ -98,7 +98,8 @@
     "check:bench": "node test/benchmarks/driverBench",
     "check:coverage": "nyc npm run test:all",
     "check:integration-coverage": "nyc npm run check:test",
-    "check:lambda": "mocha --config test/examples/mocha_lambda.json test/examples/lambda.test.js",
+    "check:lambda": "mocha --config test/examples/mocha_lambda.json test/examples/handler.test.js",
+    "check:lambda:aws": "mocha --config test/examples/mocha_lambda.json test/examples/aws_handler.test.js",
     "check:lint": "npm run build:dts && npm run check:dts && npm run check:eslint && npm run check:tsd",
     "check:eslint": "eslint -v && eslint --max-warnings=0 --ext '.js,.ts' src test",
     "check:tsd": "tsd --version && tsd",

test/examples/aws_handler.js

@@ -0,0 +1,26 @@
+// begin lambda connection
+const { MongoClient } = require('mongodb');
+
+// Get the URI for the cluster then set AWS_ACCESS_KEY_ID as the username in the
+// URI and AWS_SECRET_ACCESS_KEY as the password, then set the appropriate
+// options.
+const connectionString = new URL(process.env.MONGODB_URI);
+connectionString.username = process.env.AWS_ACCESS_KEY_ID;
+connectionString.password = encodeURIComponent(process.env.AWS_SECRET_ACCESS_KEY);
+connectionString.searchParams.set('authSource', '$external');
+connectionString.searchParams.set('authMechanism', 'MONGODB-AWS');
+
+// MongoClient now auto-connects so no need to store the connect()
+// promise anywhere and reference it.
+const client = new MongoClient(connectionString.toString());
+
+module.exports.handler = async function () {
+  const databases = await client.db('admin').command({ listDatabases: 1 });
+  return {
+    statusCode: 200,
+    databases: databases
+  };
+};
+// end lambda connection
+
+module.exports.client = client;

test/examples/aws_handler.test.js

@@ -0,0 +1,26 @@
+const { expect } = require('chai');
+const { client, handler } = require('./aws_handler');
+
+describe('AWS Lambda Examples', function () {
+  describe('#handler', function () {
+    context('when using aws environment variable authentication', function () {
+      let response;
+
+      before(async function () {
+        response = await handler();
+      });
+
+      after(async function () {
+        await client.close();
+      });
+
+      it('returns the databases', async function () {
+        expect(response.databases).to.exist;
+      });
+
+      it('returns the status code', async function () {
+        expect(response.statusCode).to.equal(200);
+      });
+    });
+  });
+});

test/examples/lambda.test.js renamed to test/examples/handler.test.js

@@ -1,8 +1,12 @@
 const path = require('path');
 const Module = require('module');
+
 const loader = Module._load;
 
-Module._load = function(request, loc) {
+// This little hack is to make require('mongodb') in our own project
+// during this specific test run to resolve to /lib so we can do
+// const { MongoClient } = require('mongodb');
+Module._load = function (request) {
   if (request === 'mongodb') {
     arguments[0] = path.join(__dirname, '..', '..', 'lib');
   }