add an integration test

Author: lerouxb

packages/cli-repl/src/mongosh-repl.ts

@@ -444,8 +444,7 @@ class MongoshNodeRepl implements EvaluationListener {
       line: string
     ) => Promise<[string[], string, 'exclusive'] | [string[], string]>;
     if (process.env.USE_NEW_AUTOCOMPLETE) {
-      const autocompletionContext =
-        instanceState.getAutocompletionContext(instanceState);
+      const autocompletionContext = instanceState.getAutocompletionContext();
       newMongoshCompleter = new MongoDBAutocompleter({
         context: autocompletionContext,
       });

packages/shell-api/src/integration.spec.ts

@@ -2772,6 +2772,44 @@ describe('Shell API (integration)', function () {
       });
     });
 
+    describe('getAutocompletionContext', function () {
+      beforeEach(async function () {
+        // Make sure the collection is present so it is included in autocompletion.
+        await collection.insertOne({});
+        // Make sure 'database' is the current db in the eyes of the instance state object.
+        instanceState.setDbFunc(database);
+      });
+
+      it('returns information for autocomplete', async function () {
+        const context = instanceState.getAutocompletionContext();
+        const { connectionId, databaseName } =
+          context.currentDatabaseAndConnection();
+        const databaseNames = await context.databasesForConnection(
+          connectionId
+        );
+        expect(databaseNames).to.include(database.getName());
+        const collectionNames = await context.collectionsForDatabase(
+          connectionId,
+          databaseName
+        );
+        expect(collectionNames).to.include(collection.getName());
+        const schema = await context.schemaInformationForCollection(
+          connectionId,
+          database.getName(),
+          collection.getName()
+        );
+        expect(schema).to.deep.equal({
+          bsonType: 'object',
+          properties: {
+            _id: {
+              bsonType: 'objectId',
+            },
+          },
+          required: ['_id'],
+        });
+      });
+    });
+
     describe('getAutocompleteParameters', function () {
       let connectionString: string;
       beforeEach(async function () {

packages/shell-api/src/shell-instance-state.ts

@@ -415,20 +415,18 @@ export class ShellInstanceState {
     throw new Error(`mongo with connection id ${connectionId} not found`);
   }
 
-  public getAutocompletionContext(
-    instanceState: ShellInstanceState
-  ): AutocompletionContext {
+  public getAutocompletionContext(): AutocompletionContext {
     return {
       currentDatabaseAndConnection: () => {
         return {
-          connectionId: instanceState.currentDb.getMongo()._getConnectionId(),
-          databaseName: instanceState.currentDb.getName(),
+          connectionId: this.currentDb.getMongo()._getConnectionId(),
+          databaseName: this.currentDb.getName(),
         };
       },
       databasesForConnection: async (
         connectionId: string
       ): Promise<string[]> => {
-        const mongo = instanceState.getMongoByConnectionId(connectionId);
+        const mongo = this.getMongoByConnectionId(connectionId);
         try {
           const dbNames = await mongo._getDatabaseNamesForCompletion();
           return dbNames.filter(
@@ -448,7 +446,7 @@ export class ShellInstanceState {
         connectionId: string,
         databaseName: string
       ): Promise<string[]> => {
-        const mongo = instanceState.getMongoByConnectionId(connectionId);
+        const mongo = this.getMongoByConnectionId(connectionId);
         try {
           const collectionNames = await mongo
             ._getDb(databaseName)
@@ -471,7 +469,7 @@ export class ShellInstanceState {
         databaseName: string,
         collectionName: string
       ): Promise<JSONSchema> => {
-        const mongo = instanceState.getMongoByConnectionId(connectionId);
+        const mongo = this.getMongoByConnectionId(connectionId);
         const docs = await mongo
           ._getDb(databaseName)
           .getCollection(collectionName)