✅ Add tests for it

elastic · Jan 4, 2022 · 5ba59ff · 5ba59ff
1 parent 5336fd2
commit 5ba59ff
Show file tree

Hide file tree

Showing 3 changed files with 749 additions and 0 deletions.
diff --git a/src/plugins/data/common/exports/export_csv.test.ts b/src/plugins/data/common/exports/export_csv.test.ts
@@ -84,4 +84,28 @@ describe('CSV exporter', () => {
       })
     ).toMatch('columnOne\r\n"\'=1"\r\n');
   });
+
+  test('should filter out columns when there are meta information', () => {
+    const datatable = getDataTable({ multipleColumns: true });
+    // Make only the first column visible
+    datatable.columns[0].meta.dimensionName = 'First column';
+    expect(
+      datatableToCSV(datatable, {
+        ...getDefaultOptions(),
+        escapeFormulaValues: true,
+        formatFactory: () => ({ convert: (v: unknown) => v } as FieldFormat),
+      })
+    ).toMatch('columnOne\r\nvalue\r\n');
+  });
+
+  test('should not filter out columns if no meta information are found', () => {
+    const datatable = getDataTable({ multipleColumns: true });
+    expect(
+      datatableToCSV(datatable, {
+        ...getDefaultOptions(),
+        escapeFormulaValues: true,
+        formatFactory: () => ({ convert: (v: unknown) => v } as FieldFormat),
+      })
+    ).toMatch('columnOne,columnTwo\r\nvalue,5\r\n');
+  });
 });