🐛 Add missing function

elastic · Aug 17, 2023 · 9fc470d · 9fc470d
1 parent c011c98
commit 9fc470d
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/x-pack/plugins/lens/public/datasources/form_based/mocks.ts b/x-pack/plugins/lens/public/datasources/form_based/mocks.ts
@@ -232,3 +232,15 @@ export function createMockedDragDropContext(): jest.Mocked<DragContextState> {
     registerDropTarget: jest.fn(),
   };
 }
+
+export const createMockedIndexPatternWithAdditionalFields = (
+  newFields: IndexPatternField[]
+): IndexPattern => {
+  const { fields, ...otherIndexPatternProps } = createMockedIndexPattern();
+  const completeFields = fields.concat(newFields);
+  return {
+    ...otherIndexPatternProps,
+    fields: completeFields,
+    getFieldByName: getFieldByNameFactory(completeFields),
+  };
+};