From 8809d7c700cd1aeec4b95e6d9145f7bb43630d66 Mon Sep 17 00:00:00 2001
From: Zacqary Xeper <zacqary.xeper@elastic.co>
Date: Tue, 23 Jun 2020 11:56:51 -0500
Subject: [PATCH] Add Expression enzyme test for prefill

---
 .../components/expression.test.tsx            | 36 +++++++++++++------
 .../containers/with_kuery_autocompletion.tsx  |  2 +-
 2 files changed, 27 insertions(+), 11 deletions(-)

diff --git a/x-pack/plugins/infra/public/alerting/metric_threshold/components/expression.test.tsx b/x-pack/plugins/infra/public/alerting/metric_threshold/components/expression.test.tsx
index c1addc84dd656..f0602ccf15d44 100644
--- a/x-pack/plugins/infra/public/alerting/metric_threshold/components/expression.test.tsx
+++ b/x-pack/plugins/infra/public/alerting/metric_threshold/components/expression.test.tsx
@@ -9,15 +9,20 @@ import { actionTypeRegistryMock } from '../../../../../triggers_actions_ui/publi
 import { alertTypeRegistryMock } from '../../../../../triggers_actions_ui/public/application/alert_type_registry.mock';
 import { coreMock } from '../../../../../../../src/core/public/mocks';
 import { AlertsContextValue } from '../../../../../triggers_actions_ui/public/application/context/alerts_context';
-import { AlertContextMeta, MetricExpression } from '../types';
+import { AlertContextMeta } from '../types';
 import { MetricsExplorerMetric } from '../../../../common/http_api/metrics_explorer';
 import React from 'react';
 import { Expressions, defaultExpression } from './expression';
 import { act } from 'react-dom/test-utils';
 // eslint-disable-next-line @kbn/eslint/no-restricted-paths
 import { Comparator } from '../../../../server/lib/alerting/metric_threshold/types';
-import { MetricsExplorerResponse } from '../../../../common/http_api';
-import { validateMetricThreshold } from './validation';
+
+jest.mock('../../../containers/source/use_source_via_http', () => ({
+  useSourceViaHttp: (any) => ({
+    source: { id: 'default' },
+    createDerivedIndexPattern: () => ({ fields: [], title: 'metricbeat-*' }),
+  }),
+}));
 
 describe('Expression', () => {
   async function setup(currentOptions: {
@@ -26,7 +31,9 @@ describe('Expression', () => {
     groupBy?: string;
   }) {
     const alertParams = {
-      criteria: [defaultExpression],
+      criteria: [],
+      groupBy: undefined,
+      filterQueryText: '',
     };
 
     const mocks = coreMock.createSetup();
@@ -55,14 +62,12 @@ describe('Expression', () => {
       },
     };
 
-    const validationResult = validateMetricThreshold([alertParams.criteria]);
-
     const wrapper = mountWithIntl(
       <Expressions
         alertsContext={context}
         alertInterval="1m"
         alertParams={alertParams}
-        errors={validationResult.errors}
+        errors={[]}
         setAlertParams={(key, value) => Reflect.set(alertParams, key, value)}
         setAlertProperty={() => {}}
       />
@@ -83,7 +88,10 @@ describe('Expression', () => {
     const currentOptions = {
       groupBy: 'host.hostname',
       filterQuery: 'foo',
-      metrics: [{ aggregation: 'avg', field: 'system.load.1' }],
+      metrics: [
+        { aggregation: 'avg', field: 'system.load.1' },
+        { aggregation: 'cardinality', field: 'system.cpu.user.pct' },
+      ] as MetricsExplorerMetric[],
     };
     const { alertParams } = await setup(currentOptions);
     expect(alertParams.groupBy).toBe('host.hostname');
@@ -93,10 +101,18 @@ describe('Expression', () => {
         metric: 'system.load.1',
         comparator: Comparator.GT,
         threshold: [],
-        timeSize,
-        timeUnit,
+        timeSize: 1,
+        timeUnit: 'm',
         aggType: 'avg',
       },
+      {
+        metric: 'system.cpu.user.pct',
+        comparator: Comparator.GT,
+        threshold: [],
+        timeSize: 1,
+        timeUnit: 'm',
+        aggType: 'cardinality',
+      },
     ]);
   });
 });
diff --git a/x-pack/plugins/infra/public/containers/with_kuery_autocompletion.tsx b/x-pack/plugins/infra/public/containers/with_kuery_autocompletion.tsx
index a04897d9c738d..2c76b3bb925ee 100644
--- a/x-pack/plugins/infra/public/containers/with_kuery_autocompletion.tsx
+++ b/x-pack/plugins/infra/public/containers/with_kuery_autocompletion.tsx
@@ -59,7 +59,7 @@ class WithKueryAutocompletionComponent extends React.Component<
   ) => {
     const { indexPattern } = this.props;
     const language = 'kuery';
-    const hasQuerySuggestions = this.props.kibana.services.data.autocomplete.hasQuerySuggestions(
+    const hasQuerySuggestions = this.props.kibana.services.data?.autocomplete.hasQuerySuggestions(
       language
     );