From 9d871530ce508e65861db98b83a120726d20e14f Mon Sep 17 00:00:00 2001
From: Timothy Sullivan <tsullivan@elastic.co>
Date: Mon, 15 Mar 2021 14:37:55 -0700
Subject: [PATCH] test the logic to sort the fields

---
 .../__snapshots__/generate_csv.test.ts.snap   |  6 ++
 .../generate_csv/generate_csv.test.ts         | 64 +++++++++++++++++++
 2 files changed, 70 insertions(+)

diff --git a/x-pack/plugins/reporting/server/export_types/csv_searchsource/generate_csv/__snapshots__/generate_csv.test.ts.snap b/x-pack/plugins/reporting/server/export_types/csv_searchsource/generate_csv/__snapshots__/generate_csv.test.ts.snap
index 6ec54a1c9c112..62c9ecff830ff 100644
--- a/x-pack/plugins/reporting/server/export_types/csv_searchsource/generate_csv/__snapshots__/generate_csv.test.ts.snap
+++ b/x-pack/plugins/reporting/server/export_types/csv_searchsource/generate_csv/__snapshots__/generate_csv.test.ts.snap
@@ -12,6 +12,12 @@ exports[`fields provides top-level underscored fields as columns 1`] = `
 "
 `;
 
+exports[`fields sorts the fields when they are to be used as table column names 1`] = `
+"\\"_id\\",\\"_index\\",\\"_score\\",\\"_type\\",date,\\"message_t\\",\\"message_u\\",\\"message_v\\",\\"message_w\\",\\"message_x\\",\\"message_y\\",\\"message_z\\"
+\\"my-cool-id\\",\\"my-cool-index\\",\\"'-\\",\\"'-\\",\\"2020-12-31T00:14:28.000Z\\",\\"test field T\\",\\"test field U\\",\\"test field V\\",\\"test field W\\",\\"test field X\\",\\"test field Y\\",\\"test field Z\\"
+"
+`;
+
 exports[`formats a search result to CSV content 1`] = `
 "date,ip,message
 \\"2020-12-31T00:14:28.000Z\\",\\"110.135.176.89\\",\\"This is a great message!\\"
diff --git a/x-pack/plugins/reporting/server/export_types/csv_searchsource/generate_csv/generate_csv.test.ts b/x-pack/plugins/reporting/server/export_types/csv_searchsource/generate_csv/generate_csv.test.ts
index 0e651c98ce06b..0193eaaff2c8d 100644
--- a/x-pack/plugins/reporting/server/export_types/csv_searchsource/generate_csv/generate_csv.test.ts
+++ b/x-pack/plugins/reporting/server/export_types/csv_searchsource/generate_csv/generate_csv.test.ts
@@ -431,6 +431,70 @@ describe('fields', () => {
     expect(csvResult.content).toMatchSnapshot();
     expect(csvResult.csv_contains_formulas).toBe(false);
   });
+
+  it('sorts the fields when they are to be used as table column names', async () => {
+    searchSourceMock.getField = jest.fn().mockImplementation((key: string) => {
+      if (key === 'fields') {
+        return ['*'];
+      }
+      return mockSearchSourceGetFieldDefault(key);
+    });
+    mockDataClient.search = jest.fn().mockImplementation(() =>
+      Rx.of({
+        rawResponse: {
+          hits: {
+            hits: [
+              {
+                _id: 'my-cool-id',
+                _index: 'my-cool-index',
+                _version: 4,
+                fields: {
+                  date: ['2020-12-31T00:14:28.000Z'],
+                  message_z: [`test field Z`],
+                  message_y: [`test field Y`],
+                  message_x: [`test field X`],
+                  message_w: [`test field W`],
+                  message_v: [`test field V`],
+                  message_u: [`test field U`],
+                  message_t: [`test field T`],
+                },
+              },
+            ],
+            total: 1,
+          },
+        },
+      })
+    );
+
+    const generateCsv = new CsvGenerator(
+      createMockJob({
+        searchSource: {
+          query: { query: '', language: 'kuery' },
+          sort: [{ '@date': 'desc' }],
+          index: '93f4bc50-6662-11eb-98bc-f550e2308366',
+          fields: ['*'],
+          filter: [],
+        },
+      }),
+      mockConfig,
+      {
+        es: mockEsClient,
+        data: mockDataClient,
+        uiSettings: uiSettingsClient,
+      },
+      {
+        searchSourceStart: mockSearchSourceService,
+        fieldFormatsRegistry: mockFieldFormatsRegistry,
+      },
+      new CancellationToken(),
+      logger
+    );
+
+    const csvResult = await generateCsv.generateData();
+
+    expect(csvResult.content).toMatchSnapshot();
+    expect(csvResult.csv_contains_formulas).toBe(false);
+  });
 });
 
 describe('formulas', () => {