[ES|QL] separate WHERE autocomplete routine

packages/kbn-esql-validation-autocomplete/src/autocomplete/__tests__/autocomplete.command.where.test.ts

@@ -0,0 +1,245 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the "Elastic License
+ * 2.0", the "GNU Affero General Public License v3.0 only", and the "Server Side
+ * Public License v 1"; you may not use this file except in compliance with, at
+ * your election, the "Elastic License 2.0", the "GNU Affero General Public
+ * License v3.0 only", or the "Server Side Public License, v 1".
+ */
+
+import { ESQL_COMMON_NUMERIC_TYPES } from '../../shared/esql_types';
+import { getDateLiterals } from '../factories';
+import { log10ParameterTypes, powParameterTypes } from './constants';
+import {
+  attachTriggerCommand,
+  fields,
+  getFieldNamesByType,
+  getFunctionSignaturesByReturnType,
+  setup,
+} from './helpers';
+
+describe('WHERE expression', () => {
+  const allEvalFns = getFunctionSignaturesByReturnType('where', 'any', {
+    scalar: true,
+  });
+  test('beginning an expression', async () => {
+    const { assertSuggestions } = await setup();
+
+    await assertSuggestions('from a | where /', [
+      ...getFieldNamesByType('any')
+        .map((field) => `${field} `)
+        .map(attachTriggerCommand),
+      ...allEvalFns,
+    ]);
+    await assertSuggestions(
+      'from a | eval var0 = 1 | where /',
+      [
+        ...getFieldNamesByType('any')
+          .map((name) => `${name} `)
+          .map(attachTriggerCommand),
+        attachTriggerCommand('var0 '),
+        ...allEvalFns,
+      ],
+      {
+        callbacks: {
+          getColumnsFor: () => Promise.resolve([...fields, { name: 'var0', type: 'integer' }]),
+        },
+      }
+    );
+  });
+
+  describe('within the expression', () => {
+    test('after a field name', async () => {
+      const { assertSuggestions } = await setup();
+
+      await assertSuggestions('from a | where keywordField /', [
+        // all functions compatible with a keywordField type
+        ...getFunctionSignaturesByReturnType(
+          'where',
+          'boolean',
+          {
+            builtin: true,
+          },
+          undefined,
+          ['and', 'or', 'not']
+        ),
+      ]);
+    });
+
+    test('suggests dates after a comparison with a date', async () => {
+      const { assertSuggestions } = await setup();
+
+      const expectedComparisonWithDateSuggestions = [
+        ...getDateLiterals(),
+        ...getFieldNamesByType(['date']),
+        // all functions compatible with a keywordField type
+        ...getFunctionSignaturesByReturnType('where', ['date'], { scalar: true }),
+      ];
+      await assertSuggestions(
+        'from a | where dateField == /',
+        expectedComparisonWithDateSuggestions
+      );
+
+      await assertSuggestions(
+        'from a | where dateField < /',
+        expectedComparisonWithDateSuggestions
+      );
+
+      await assertSuggestions(
+        'from a | where dateField >= /',
+        expectedComparisonWithDateSuggestions
+      );
+    });
+
+    test('after a comparison with a string field', async () => {
+      const { assertSuggestions } = await setup();
+
+      const expectedComparisonWithTextFieldSuggestions = [
+        ...getFieldNamesByType(['text', 'keyword', 'ip', 'version']),
+        ...getFunctionSignaturesByReturnType('where', ['text', 'keyword', 'ip', 'version'], {
+          scalar: true,
+        }),
+      ];
+
+      await assertSuggestions(
+        'from a | where textField >= /',
+        expectedComparisonWithTextFieldSuggestions
+      );
+      await assertSuggestions(
+        'from a | where textField >= textField/',
+        expectedComparisonWithTextFieldSuggestions
+      );
+    });
+
+    test('after a logical operator', async () => {
+      const { assertSuggestions } = await setup();
+
+      for (const op of ['and', 'or']) {
+        await assertSuggestions(`from a | where keywordField >= keywordField ${op} /`, [
+          ...getFieldNamesByType('any'),
+          ...getFunctionSignaturesByReturnType('where', 'any', { scalar: true }),
+        ]);
+        await assertSuggestions(`from a | where keywordField >= keywordField ${op} doubleField /`, [
+          ...getFunctionSignaturesByReturnType('where', 'boolean', { builtin: true }, ['double']),
+        ]);
+        await assertSuggestions(
+          `from a | where keywordField >= keywordField ${op} doubleField == /`,
+          [
+            ...getFieldNamesByType(ESQL_COMMON_NUMERIC_TYPES),
+            ...getFunctionSignaturesByReturnType('where', ESQL_COMMON_NUMERIC_TYPES, {
+              scalar: true,
+            }),
+          ]
+        );
+      }
+    });
+
+    test('suggests operators after a field name', async () => {
+      const { assertSuggestions } = await setup();
+
+      await assertSuggestions('from a | stats a=avg(doubleField) | where a /', [
+        ...getFunctionSignaturesByReturnType('where', 'any', { builtin: true, skipAssign: true }, [
+          'double',
+        ]),
+      ]);
+    });
+
+    test('accounts for fields lost in previous commands', async () => {
+      const { assertSuggestions } = await setup();
+
+      // Mind this test: suggestion is aware of previous commands when checking for fields
+      // in this case the doubleField has been wiped by the STATS command and suggest cannot find it's type
+      await assertSuggestions('from a | stats a=avg(doubleField) | where doubleField /', [], {
+        callbacks: { getColumnsFor: () => Promise.resolve([{ name: 'a', type: 'double' }]) },
+      });
+    });
+
+    test('suggests function arguments', async () => {
+      const { assertSuggestions } = await setup();
+
+      // The editor automatically inject the final bracket, so it is not useful to test with just open bracket
+      await assertSuggestions(
+        'from a | where log10(/)',
+        [
+          ...getFieldNamesByType(log10ParameterTypes),
+          ...getFunctionSignaturesByReturnType(
+            'where',
+            log10ParameterTypes,
+            { scalar: true },
+            undefined,
+            ['log10']
+          ),
+        ],
+        { triggerCharacter: '(' }
+      );
+      await assertSuggestions(
+        'from a | WHERE pow(doubleField, /)',
+        [
+          ...getFieldNamesByType(powParameterTypes),
+          ...getFunctionSignaturesByReturnType(
+            'where',
+            powParameterTypes,
+            { scalar: true },
+            undefined,
+            ['pow']
+          ),
+        ],
+        { triggerCharacter: ',' }
+      );
+    });
+
+    test('suggests boolean and numeric operators after a numeric function result', async () => {
+      const { assertSuggestions } = await setup();
+
+      await assertSuggestions('from a | where log10(doubleField) /', [
+        ...getFunctionSignaturesByReturnType('where', 'double', { builtin: true }, ['double']),
+        ...getFunctionSignaturesByReturnType('where', 'boolean', { builtin: true }, ['double']),
+      ]);
+    });
+
+    test('suggestions after NOT', async () => {
+      const { assertSuggestions } = await setup();
+      await assertSuggestions('from index | WHERE keywordField not /', [
+        'LIKE $0',
+        'RLIKE $0',
+        'IN $0',
+      ]);
+      await assertSuggestions('from index | WHERE keywordField NOT /', [
+        'LIKE $0',
+        'RLIKE $0',
+        'IN $0',
+      ]);
+      await assertSuggestions('from index | WHERE not /', [
+        ...getFieldNamesByType('boolean'),
+        ...getFunctionSignaturesByReturnType('eval', 'boolean', { scalar: true }),
+      ]);
+    });
+
+    test('suggestions after IN', async () => {
+      const { assertSuggestions } = await setup();
+
+      await assertSuggestions('from index | WHERE doubleField in /', ['( $0 )']);
+      await assertSuggestions('from index | WHERE doubleField not in /', ['( $0 )']);
+      await assertSuggestions(
+        'from index | WHERE doubleField not in (/)',
+        [
+          ...getFieldNamesByType('double').filter((name) => name !== 'doubleField'),
+          ...getFunctionSignaturesByReturnType('where', 'double', { scalar: true }),
+        ],
+        { triggerCharacter: '(' }
+      );
+      await assertSuggestions('from index | WHERE doubleField in ( `any#Char$Field`, /)', [
+        ...getFieldNamesByType('double').filter(
+          (name) => name !== '`any#Char$Field`' && name !== 'doubleField'
+        ),
+        ...getFunctionSignaturesByReturnType('where', 'double', { scalar: true }),
+      ]);
+      await assertSuggestions('from index | WHERE doubleField not in ( `any#Char$Field`, /)', [
+        ...getFieldNamesByType('double').filter(
+          (name) => name !== '`any#Char$Field`' && name !== 'doubleField'
+        ),
+        ...getFunctionSignaturesByReturnType('where', 'double', { scalar: true }),
+      ]);
+    });
+  });
+});

packages/kbn-esql-validation-autocomplete/src/autocomplete/__tests__/hidden_functions_and_commands.test.ts

@@ -50,6 +50,7 @@ describe('hidden functions', () => {
     expect(suggestedFunctions).toContain('VISIBLE_FUNCTION($0)');
     expect(suggestedFunctions).not.toContain('HIDDEN_FUNCTION($0)');
   });
+
   it('does not suggest hidden agg functions', async () => {
     setTestFunctions([
       {