[ML] Explain Log Rate Spikes: Adds jest tests for query_utils/buildBa…

…seFilterCriteria. (#141213) (#141438)

Adds jest tests for query_utils/buildBaseFilterCriteria.

(cherry picked from commit dd2cfb7)

Co-authored-by: Walter Rafelsberger <walter.rafelsberger@elastic.co>

x-pack/plugins/aiops/public/application/utils/query_utils.test.ts

@@ -0,0 +1,254 @@
+/*
+ * 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; you may not use this file except in compliance with the Elastic License
+ * 2.0.
+ */
+
+import type { ChangePoint } from '@kbn/ml-agg-utils';
+
+import type { GroupTableItem } from '../../components/spike_analysis_table/spike_analysis_table_groups';
+
+import { buildBaseFilterCriteria } from './query_utils';
+
+const selectedChangePointMock: ChangePoint = {
+  doc_count: 53408,
+  bg_count: 1154,
+  fieldName: 'meta.cloud.instance_id.keyword',
+  fieldValue: '1234',
+  normalizedScore: 1,
+  pValue: 0.01,
+  score: 708.3964185322641,
+  total_bg_count: 179657,
+  total_doc_count: 114011,
+};
+
+const selectedGroupMock: GroupTableItem = {
+  id: '21289599',
+  docCount: 20468,
+  pValue: 2.2250738585072626e-308,
+  group: {
+    'error.message': 'rate limit exceeded',
+    message: 'too many requests',
+    'user_agent.original.keyword': 'Mozilla/5.0',
+  },
+  repeatedValues: {
+    'beat.hostname.keyword': 'ip-192-168-1-1',
+    'beat.name.keyword': 'i-1234',
+    'docker.container.id.keyword': 'asdf',
+  },
+  histogram: [],
+};
+
+describe('query_utils', () => {
+  describe('buildBaseFilterCriteria', () => {
+    it('returns range filter based on minimum supplied arguments', () => {
+      const baseFilterCriteria = buildBaseFilterCriteria('the-time-field-name', 1234, 5678);
+
+      expect(baseFilterCriteria).toEqual([
+        {
+          range: {
+            'the-time-field-name': {
+              format: 'epoch_millis',
+              gte: 1234,
+              lte: 5678,
+            },
+          },
+        },
+      ]);
+    });
+
+    it('returns filters including default query with supplied arguments provided via UI', () => {
+      const baseFilterCriteria = buildBaseFilterCriteria(
+        '@timestamp',
+        1640082000012,
+        1640103600906,
+        { match_all: {} }
+      );
+
+      expect(baseFilterCriteria).toEqual([
+        {
+          range: {
+            '@timestamp': {
+              format: 'epoch_millis',
+              gte: 1640082000012,
+              lte: 1640103600906,
+            },
+          },
+        },
+        { match_all: {} },
+      ]);
+    });
+
+    it('includes a term filter when including a selectedChangePoint', () => {
+      const baseFilterCriteria = buildBaseFilterCriteria(
+        '@timestamp',
+        1640082000012,
+        1640103600906,
+        { match_all: {} },
+        selectedChangePointMock
+      );
+
+      expect(baseFilterCriteria).toEqual([
+        {
+          range: {
+            '@timestamp': {
+              format: 'epoch_millis',
+              gte: 1640082000012,
+              lte: 1640103600906,
+            },
+          },
+        },
+        { match_all: {} },
+        { term: { 'meta.cloud.instance_id.keyword': '1234' } },
+      ]);
+    });
+
+    it('includes a term filter with must_not when excluding a selectedChangePoint', () => {
+      const baseFilterCriteria = buildBaseFilterCriteria(
+        '@timestamp',
+        1640082000012,
+        1640103600906,
+        { match_all: {} },
+        selectedChangePointMock,
+        false
+      );
+
+      expect(baseFilterCriteria).toEqual([
+        {
+          range: {
+            '@timestamp': {
+              format: 'epoch_millis',
+              gte: 1640082000012,
+              lte: 1640103600906,
+            },
+          },
+        },
+        { match_all: {} },
+        { bool: { must_not: [{ term: { 'meta.cloud.instance_id.keyword': '1234' } }] } },
+      ]);
+    });
+
+    it('includes multiple term filters when including a selectedGroupMock', () => {
+      const baseFilterCriteria = buildBaseFilterCriteria(
+        '@timestamp',
+        1640082000012,
+        1640103600906,
+        { match_all: {} },
+        undefined,
+        true,
+        selectedGroupMock
+      );
+
+      expect(baseFilterCriteria).toEqual([
+        {
+          range: {
+            '@timestamp': {
+              format: 'epoch_millis',
+              gte: 1640082000012,
+              lte: 1640103600906,
+            },
+          },
+        },
+        { match_all: {} },
+        {
+          term: {
+            'error.message': 'rate limit exceeded',
+          },
+        },
+        {
+          term: {
+            message: 'too many requests',
+          },
+        },
+        {
+          term: {
+            'user_agent.original.keyword': 'Mozilla/5.0',
+          },
+        },
+        {
+          term: {
+            'beat.hostname.keyword': 'ip-192-168-1-1',
+          },
+        },
+        {
+          term: {
+            'beat.name.keyword': 'i-1234',
+          },
+        },
+        {
+          term: {
+            'docker.container.id.keyword': 'asdf',
+          },
+        },
+      ]);
+    });
+
+    it('includes a must_not with nested term filters when excluding a selectedGroup', () => {
+      const baseFilterCriteria = buildBaseFilterCriteria(
+        '@timestamp',
+        1640082000012,
+        1640103600906,
+        { match_all: {} },
+        undefined,
+        false,
+        selectedGroupMock
+      );
+
+      expect(baseFilterCriteria).toEqual([
+        {
+          range: {
+            '@timestamp': {
+              format: 'epoch_millis',
+              gte: 1640082000012,
+              lte: 1640103600906,
+            },
+          },
+        },
+        { match_all: {} },
+        {
+          bool: {
+            must_not: [
+              {
+                bool: {
+                  filter: [
+                    {
+                      term: {
+                        'error.message': 'rate limit exceeded',
+                      },
+                    },
+                    {
+                      term: {
+                        message: 'too many requests',
+                      },
+                    },
+                    {
+                      term: {
+                        'user_agent.original.keyword': 'Mozilla/5.0',
+                      },
+                    },
+                    {
+                      term: {
+                        'beat.hostname.keyword': 'ip-192-168-1-1',
+                      },
+                    },
+                    {
+                      term: {
+                        'beat.name.keyword': 'i-1234',
+                      },
+                    },
+                    {
+                      term: {
+                        'docker.container.id.keyword': 'asdf',
+                      },
+                    },
+                  ],
+                },
+              },
+            ],
+          },
+        },
+      ]);
+    });
+  });
+});