chore(cli-repl): add async-rewriter library integration test (#767)

Load lodash (all of it) and moment.js into the shell and perform
a basic test that verifies that they can be used with the new
async rewriter.

Author: addaleax

packages/cli-repl/package-lock.json

@@ -75,7 +75,9 @@
     "@types/read": "^0.0.28",
     "@types/text-table": "^0.2.1",
     "@types/yargs-parser": "^15.0.0",
-    "chai-as-promised": "^7.1.1"
+    "chai-as-promised": "^7.1.1",
+    "lodash": "^4.17.21",
+    "moment": "^2.29.1"
   },
   "optionalDependencies": {
     "macos-export-certificate-and-key": "^1.0.0",

packages/cli-repl/package.json

@@ -364,6 +364,34 @@ describe('e2e', function() {
       expect(result).to.include('{ _id: 1, value: 4 }');
     });
 
+    it('rewrites async properly for common libraries', async function() {
+      this.timeout(120_000);
+      // This is being run under the new async rewriter because the old one
+      // did not support these libraries at all (for various reasons).
+      // Once the new async rewriter is the default, this block can be removed.
+      shell = TestShell.start({
+        args: [ await testServer.connectionString() ],
+        env: { ...process.env, MONGOSH_ASYNC_REWRITER2: '1' }
+      });
+      await shell.waitForPrompt();
+      shell.assertNoErrors();
+
+      await shell.executeLine('Date.now = () => new Date().getTime()'); // MONGOSH-597 hack
+      await shell.executeLine(`use ${dbName}`);
+      await shell.executeLine('db.test.insertOne({ d: new Date("2021-04-07T11:24:54+02:00") })');
+      shell.writeInputLine(`load(${JSON.stringify(require.resolve('lodash'))})`);
+      shell.writeInputLine(`load(${JSON.stringify(require.resolve('moment'))})`);
+      shell.writeInputLine('print("loaded" + "scripts")');
+      await eventually(() => {
+        // Use eventually explicitly to get a bigger timeout, lodash is
+        // quite “big” in terms of async rewriting
+        shell.assertContainsOutput('loadedscripts');
+      }, { timeout: 60_000 });
+      const result = await shell.executeLine(
+        'moment(_.first(_.map(db.test.find().toArray(), "d"))).format("X")');
+      expect(result).to.include('1617787494');
+    });
+
     it('expands explain output indefinitely', async() => {
       await shell.executeLine('explainOutput = db.test.find().explain()');
       await shell.executeLine('explainOutput.a = {b:{c:{d:{e:{f:{g:{h:{i:{j:{}}}}}}}}}}');