quosdolorem/packages/browser-runtime-core/src/autocompleter/shell-api-autocompleter.spec.ts

The MongoDB Shell

import { ShellApiAutocompleter } from './shell-api-autocompleter';
import { expect } from 'chai';
import { Topologies } from '@mongosh/shell-api';

const standalone440 = {
  topology: () => Topologies.Standalone,
  connectionInfo: () => ({
    is_atlas: false,
    is_data_lake: false,
    server_version: '4.4.0'
  }),
  getCollectionCompletionsForCurrentDb: () => ['bananas'],
  getDatabaseCompletions: () => ['databaseOne']
};

describe('Autocompleter', () => {
  describe('getCompletions', () => {
    let autocompleter: ShellApiAutocompleter;

    beforeEach(() => {
      autocompleter = new ShellApiAutocompleter(standalone440);
    });

    it('returns completions for text before cursor', async() => {
      const completions = await autocompleter.getCompletions('db.coll1.');

      expect(completions).to.deep.contain({
        completion: 'db.coll1.find'
      });
    });

    it('returns full completion value with text after dot', async() => {
      const completions = await autocompleter.getCompletions('db.coll1.f');

      expect(completions).to.deep.contain({
        completion: 'db.coll1.find'
      });
    });

    it('returns collection names value with text after dot', async() => {
      const completions = await autocompleter.getCompletions('db.b');

      expect(completions).to.deep.contain({
        completion: 'db.bananas'
      });
    });

    it('returns database names after use', async() => {
      const completions = await autocompleter.getCompletions('use da');

      expect(completions).to.deep.contain({
        completion: 'use databaseOne'
      });
    });
  });
});