[ML] Add jest tests.

Author: walterra

x-pack/plugins/ml/public/application/components/scatterplot_matrix/scatterplot_matrix_vega_lite_spec.test.ts

@@ -5,6 +5,8 @@
  * 2.0.
  */
 
+import 'jest-canvas-mock';
+
 // @ts-ignore
 import { compile } from 'vega-lite/build/vega-lite';
 
@@ -156,4 +158,25 @@ describe('getScatterplotMatrixVegaLiteSpec()', () => {
       { field: 'y', type: 'quantitative' },
     ]);
   });
+
+  it('should escape special characters', () => {
+    const data = [{ ['x.a']: 1, ['y[a]']: 1 }];
+
+    const vegaLiteSpec = getScatterplotMatrixVegaLiteSpec(
+      data,
+      ['x.a', 'y[a]'],
+      euiThemeLight,
+      undefined,
+      'the-color-field',
+      LEGEND_TYPES.NOMINAL
+    );
+
+    // column values should be escaped
+    expect(vegaLiteSpec.repeat).toEqual({
+      column: ['x\\.a', 'y\\[a\\]'],
+      row: ['y\\[a\\]', 'x\\.a'],
+    });
+    // raw data should not be escaped
+    expect(vegaLiteSpec.spec.data.values).toEqual(data);
+  });
 });