From bb478aa99fe69bd7c0ef8dd71e84f372b39f5c54 Mon Sep 17 00:00:00 2001
From: Thomas Neirynck <thomas@elastic.co>
Date: Fri, 18 Sep 2020 11:22:03 -0400
Subject: [PATCH] add unit tests

---
 .../update_source_editor.test.tsx.snap        | 181 ++++++++++++++++++
 .../resolution_editor.test.tsx                |  29 +++
 .../update_source_editor.test.tsx             |  45 +++++
 3 files changed, 255 insertions(+)
 create mode 100644 x-pack/plugins/maps/public/classes/sources/es_geo_grid_source/__snapshots__/update_source_editor.test.tsx.snap
 create mode 100644 x-pack/plugins/maps/public/classes/sources/es_geo_grid_source/resolution_editor.test.tsx
 create mode 100644 x-pack/plugins/maps/public/classes/sources/es_geo_grid_source/update_source_editor.test.tsx

diff --git a/x-pack/plugins/maps/public/classes/sources/es_geo_grid_source/__snapshots__/update_source_editor.test.tsx.snap b/x-pack/plugins/maps/public/classes/sources/es_geo_grid_source/__snapshots__/update_source_editor.test.tsx.snap
new file mode 100644
index 0000000000000..97d51e77fb87c
--- /dev/null
+++ b/x-pack/plugins/maps/public/classes/sources/es_geo_grid_source/__snapshots__/update_source_editor.test.tsx.snap
@@ -0,0 +1,181 @@
+// Jest Snapshot v1, https://goo.gl/fbAQLP
+
+exports[`source editor geo_grid_source default vector layer config should allow super-fine option 1`] = `
+<Fragment>
+  <EuiPanel>
+    <EuiTitle
+      size="xs"
+    >
+      <h6>
+        <FormattedMessage
+          defaultMessage="Metrics"
+          id="xpack.maps.source.esGrid.metricsLabel"
+          values={Object {}}
+        />
+      </h6>
+    </EuiTitle>
+    <EuiSpacer
+      size="m"
+    />
+    <MetricsEditor
+      allowMultipleMetrics={true}
+      fields={null}
+      metrics={Array []}
+      metricsFilter={null}
+      onChange={[Function]}
+    />
+  </EuiPanel>
+  <EuiSpacer
+    size="s"
+  />
+  <EuiPanel>
+    <EuiTitle
+      size="xs"
+    >
+      <h6>
+        <FormattedMessage
+          defaultMessage="Grid parameters"
+          id="xpack.maps.source.esGrid.geoTileGridLabel"
+          values={Object {}}
+        />
+      </h6>
+    </EuiTitle>
+    <EuiSpacer
+      size="m"
+    />
+    <ResolutionEditor
+      includeSuperFine={true}
+      onChange={[Function]}
+      resolution="COARSE"
+    />
+    <RenderAsSelect
+      isColumnCompressed={true}
+      onChange={[Function]}
+      renderAs="point"
+    />
+  </EuiPanel>
+  <EuiSpacer
+    size="s"
+  />
+</Fragment>
+`;
+
+exports[`source editor geo_grid_source should put limitations based on heatmap-rendering selection should disable multiple metrics 1`] = `
+<Fragment>
+  <EuiPanel>
+    <EuiTitle
+      size="xs"
+    >
+      <h6>
+        <FormattedMessage
+          defaultMessage="Metrics"
+          id="xpack.maps.source.esGrid.metricsLabel"
+          values={Object {}}
+        />
+      </h6>
+    </EuiTitle>
+    <EuiSpacer
+      size="m"
+    />
+    <MetricsEditor
+      allowMultipleMetrics={false}
+      fields={null}
+      metrics={Array []}
+      metricsFilter={[Function]}
+      onChange={[Function]}
+    />
+  </EuiPanel>
+  <EuiSpacer
+    size="s"
+  />
+  <EuiPanel>
+    <EuiTitle
+      size="xs"
+    >
+      <h6>
+        <FormattedMessage
+          defaultMessage="Grid parameters"
+          id="xpack.maps.source.esGrid.geoTileGridLabel"
+          values={Object {}}
+        />
+      </h6>
+    </EuiTitle>
+    <EuiSpacer
+      size="m"
+    />
+    <ResolutionEditor
+      includeSuperFine={true}
+      onChange={[Function]}
+      resolution="COARSE"
+    />
+    <RenderAsSelect
+      isColumnCompressed={true}
+      onChange={[Function]}
+      renderAs="heatmap"
+    />
+  </EuiPanel>
+  <EuiSpacer
+    size="s"
+  />
+</Fragment>
+`;
+
+exports[`source editor geo_grid_source should put limitations based on heatmap-rendering selection should not allow super-fine option for heatmaps 1`] = `
+<Fragment>
+  <EuiPanel>
+    <EuiTitle
+      size="xs"
+    >
+      <h6>
+        <FormattedMessage
+          defaultMessage="Metrics"
+          id="xpack.maps.source.esGrid.metricsLabel"
+          values={Object {}}
+        />
+      </h6>
+    </EuiTitle>
+    <EuiSpacer
+      size="m"
+    />
+    <MetricsEditor
+      allowMultipleMetrics={true}
+      fields={null}
+      metrics={Array []}
+      metricsFilter={null}
+      onChange={[Function]}
+    />
+  </EuiPanel>
+  <EuiSpacer
+    size="s"
+  />
+  <EuiPanel>
+    <EuiTitle
+      size="xs"
+    >
+      <h6>
+        <FormattedMessage
+          defaultMessage="Grid parameters"
+          id="xpack.maps.source.esGrid.geoTileGridLabel"
+          values={Object {}}
+        />
+      </h6>
+    </EuiTitle>
+    <EuiSpacer
+      size="m"
+    />
+    <ResolutionEditor
+      includeSuperFine={false}
+      onChange={[Function]}
+      resolution="COARSE"
+    />
+    <RenderAsSelect
+      isColumnCompressed={true}
+      onChange={[Function]}
+      renderAs="point"
+    />
+  </EuiPanel>
+  <EuiSpacer
+    size="s"
+  />
+</Fragment>
+`;
diff --git a/x-pack/plugins/maps/public/classes/sources/es_geo_grid_source/resolution_editor.test.tsx b/x-pack/plugins/maps/public/classes/sources/es_geo_grid_source/resolution_editor.test.tsx
new file mode 100644
index 0000000000000..909dd210593ff
--- /dev/null
+++ b/x-pack/plugins/maps/public/classes/sources/es_geo_grid_source/resolution_editor.test.tsx
@@ -0,0 +1,29 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License;
+ * you may not use this file except in compliance with the Elastic License.
+ */
+
+import React from 'react';
+import { shallow } from 'enzyme';
+
+// @ts-expect-error
+import { ResolutionEditor } from './resolution_editor';
+import { GRID_RESOLUTION } from '../../../../common/constants';
+
+const defaultProps = {
+  resolution: GRID_RESOLUTION.COARSE,
+  onChange: () => {},
+  includeSuperFine: false,
+};
+
+describe('resolution editor', () => {
+  test('should omit super-fine option', async () => {
+    const component = shallow(<ResolutionEditor />);
+    expect(component).toMatchSnapshot();
+  });
+  test('should add super-fine option', async () => {
+    const component = shallow(<ResolutionEditor includeSuperFine={true} />);
+    expect(component).toMatchSnapshot();
+  });
+});
diff --git a/x-pack/plugins/maps/public/classes/sources/es_geo_grid_source/update_source_editor.test.tsx b/x-pack/plugins/maps/public/classes/sources/es_geo_grid_source/update_source_editor.test.tsx
new file mode 100644
index 0000000000000..2d9dce6c82f88
--- /dev/null
+++ b/x-pack/plugins/maps/public/classes/sources/es_geo_grid_source/update_source_editor.test.tsx
@@ -0,0 +1,45 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License;
+ * you may not use this file except in compliance with the Elastic License.
+ */
+
+import React from 'react';
+import { shallow } from 'enzyme';
+
+// @ts-expect-error
+import { UpdateSourceEditor } from './update_source_editor';
+import { GRID_RESOLUTION, LAYER_TYPE, RENDER_AS } from '../../../../common/constants';
+
+const defaultProps = {
+  currentLayerType: LAYER_TYPE.VECTOR,
+  indexPatternId: 'foobar',
+  onChange: () => {},
+  metrics: [],
+  renderAs: RENDER_AS.POINT,
+  resolution: GRID_RESOLUTION.COARSE,
+};
+
+describe('source editor geo_grid_source', () => {
+  describe('default vector layer config', () => {
+    test('should allow super-fine option', async () => {
+      const component = shallow(<UpdateSourceEditor {...defaultProps} />);
+      expect(component).toMatchSnapshot();
+    });
+  });
+
+  describe('should put limitations based on heatmap-rendering selection', () => {
+    test('should not allow super-fine option for heatmaps', async () => {
+      const component = shallow(
+        <UpdateSourceEditor {...defaultProps} currentLayerType={LAYER_TYPE.HEATMAP} />
+      );
+      expect(component).toMatchSnapshot();
+    });
+    test('should disable multiple metrics', async () => {
+      const component = shallow(
+        <UpdateSourceEditor {...defaultProps} renderAs={RENDER_AS.HEATMAP} />
+      );
+      expect(component).toMatchSnapshot();
+    });
+  });
+});