Change types in test

elastic · Nov 20, 2020 · 26db7b8 · 26db7b8
1 parent c0a16e4
commit 26db7b8
Showing 1 changed file with 13 additions and 8 deletions.
diff --git a/x-pack/plugins/lens/public/indexpattern_datasource/indexpattern.test.ts b/x-pack/plugins/lens/public/indexpattern_datasource/indexpattern.test.ts
@@ -660,30 +660,35 @@ describe('IndexPattern Data Source', () => {
 
       it('should skip columns that are being referenced', () => {
         publicAPI = indexPatternDatasource.getPublicAPI({
-          state: ({
+          state: {
+            ...enrichBaseState(baseState),
             layers: {
               first: {
                 indexPatternId: '1',
                 columnOrder: ['col1', 'col2'],
                 columns: {
                   col1: {
-                    dataType: 'number',
-                    operationType: 'sum',
                     label: 'Sum',
+                    dataType: 'number',
                     isBucketed: false,
+
+                    operationType: 'sum',
                     sourceField: 'test',
-                  },
+                    params: {},
+                  } as IndexPatternColumn,
                   col2: {
-                    operationType: 'cumulative_sum',
+                    label: 'Cumulative sum',
                     dataType: 'number',
                     isBucketed: false,
-                    label: 'Cumulative sum',
+
+                    operationType: 'cumulative_sum',
                     references: ['col1'],
-                  },
+                    params: {},
+                  } as IndexPatternColumn,
                 },
               },
             },
-          } as unknown) as IndexPatternPrivateState,
+          },
           layerId: 'first',
         });