Fix mapColumn test

elastic · Jun 28, 2021 · bf3c1ab · bf3c1ab
1 parent 79b0bdf
commit bf3c1ab
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/plugins/expressions/common/expression_functions/specs/tests/map_column.test.ts b/src/plugins/expressions/common/expression_functions/specs/tests/map_column.test.ts
@@ -12,7 +12,8 @@ import { Datatable } from '../../../expression_types';
 import { mapColumn, MapColumnArguments } from '../map_column';
 import { emptyTable, functionWrapper, testTable, tableWithNulls } from './utils';
 
-const pricePlusTwo = (datatable: Datatable) => of(datatable.rows[0].price + 2);
+const pricePlusTwo = (datatable: Datatable) =>
+  of(typeof datatable.rows[0].price === 'number' ? datatable.rows[0].price + 2 : null);
 
 describe('mapColumn', () => {
   const fn = functionWrapper(mapColumn);
@@ -223,7 +224,7 @@ describe('mapColumn', () => {
     testScheduler.run(({ expectObservable }) => {
       expectObservable(
         runFn(
-          { ...emptyTable, rows: [...emptyTable.rows, { value: 5 }] },
+          { ...emptyTable, rows: [...emptyTable.rows, { price: 5 }] },
           { name: 'value', copyMetaFrom: 'time', expression: pricePlusTwo }
         )
       ).toBe('(0|)', [
@@ -236,6 +237,7 @@ describe('mapColumn', () => {
               meta: expect.objectContaining({ type: 'number' }),
             }),
           ],
+          rows: [{ price: 5, value: 7 }],
         }),
       ]);
     });
@@ -248,7 +250,6 @@ describe('mapColumn', () => {
           id: 'value',
           name: 'value',
           expression: pricePlusTwo,
-          onError: 'null',
         })
       ).toBe('(0|)', [
         expect.objectContaining({