From fff036df37c55946dab4a0577c977e8dc2b8a5a6 Mon Sep 17 00:00:00 2001
From: Stratoula Kalafateli <efstratia.kalafateli@elastic.co>
Date: Mon, 6 May 2024 11:23:20 +0200
Subject: [PATCH] Correct existing wrong Discover tests

---
 .../data_table_columns.test.tsx.snap          | 48 +++++++++++++++++--
 .../components/data_table_columns.test.tsx    |  5 ++
 2 files changed, 48 insertions(+), 5 deletions(-)

diff --git a/packages/kbn-unified-data-table/src/components/__snapshots__/data_table_columns.test.tsx.snap b/packages/kbn-unified-data-table/src/components/__snapshots__/data_table_columns.test.tsx.snap
index 3759462c21b7d..09b51e02f4db9 100644
--- a/packages/kbn-unified-data-table/src/components/__snapshots__/data_table_columns.test.tsx.snap
+++ b/packages/kbn-unified-data-table/src/components/__snapshots__/data_table_columns.test.tsx.snap
@@ -2626,6 +2626,8 @@ Array [
     },
     "cellActions": Array [
       [Function],
+      [Function],
+      [Function],
     ],
     "display": <Memo(DataTableTimeColumnHeader)
       dataView={
@@ -2783,7 +2785,9 @@ Array [
           "customDescription": undefined,
           "customLabel": undefined,
           "defaultFormatter": undefined,
-          "esTypes": undefined,
+          "esTypes": Array [
+            "dateTime",
+          ],
           "lang": undefined,
           "name": "timestamp",
           "readFromDocValues": false,
@@ -2791,7 +2795,7 @@ Array [
           "scripted": false,
           "searchable": true,
           "subType": undefined,
-          "type": "unknown",
+          "type": "date",
         }
       }
       headerRowHeight={5}
@@ -2800,7 +2804,7 @@ Array [
     "id": "timestamp",
     "initialWidth": 212,
     "isSortable": true,
-    "schema": "kibana-json",
+    "schema": "datetime",
     "visibleCellActions": undefined,
   },
   Object {
@@ -2844,10 +2848,27 @@ Array [
     },
     "cellActions": Array [
       [Function],
+      [Function],
+      [Function],
     ],
     "display": <Memo(DataTableColumnHeader)
       columnDisplayName="extension"
       columnName="extension"
+      columnsMeta={
+        Object {
+          "extension": Object {
+            "type": "number",
+          },
+          "message": Object {
+            "esType": "keyword",
+            "type": "string",
+          },
+          "timestamp": Object {
+            "esType": "dateTime",
+            "type": "date",
+          },
+        }
+      }
       dataView={
         Object {
           "docvalueFields": Array [],
@@ -3000,7 +3021,7 @@ Array [
     "displayAsText": "extension",
     "id": "extension",
     "isSortable": true,
-    "schema": "kibana-json",
+    "schema": "numeric",
     "visibleCellActions": undefined,
   },
   Object {
@@ -3044,10 +3065,27 @@ Array [
     },
     "cellActions": Array [
       [Function],
+      [Function],
+      [Function],
     ],
     "display": <Memo(DataTableColumnHeader)
       columnDisplayName="message"
       columnName="message"
+      columnsMeta={
+        Object {
+          "extension": Object {
+            "type": "number",
+          },
+          "message": Object {
+            "esType": "keyword",
+            "type": "string",
+          },
+          "timestamp": Object {
+            "esType": "dateTime",
+            "type": "date",
+          },
+        }
+      }
       dataView={
         Object {
           "docvalueFields": Array [],
@@ -3200,7 +3238,7 @@ Array [
     "displayAsText": "message",
     "id": "message",
     "isSortable": true,
-    "schema": "kibana-json",
+    "schema": "string",
     "visibleCellActions": undefined,
   },
 ]
diff --git a/packages/kbn-unified-data-table/src/components/data_table_columns.test.tsx b/packages/kbn-unified-data-table/src/components/data_table_columns.test.tsx
index 72482436aff1f..070eb2db92576 100644
--- a/packages/kbn-unified-data-table/src/components/data_table_columns.test.tsx
+++ b/packages/kbn-unified-data-table/src/components/data_table_columns.test.tsx
@@ -93,6 +93,11 @@ describe('Data table columns', function () {
         hasEditDataViewPermission: () =>
           servicesMock.dataViewFieldEditor.userPermissions.editIndexPattern(),
         onFilter: () => {},
+        columnsMeta: {
+          extension: { type: 'number' },
+          message: { type: 'string', esType: 'keyword' },
+          timestamp: { type: 'date', esType: 'dateTime' },
+        },
       });
       expect(actual).toMatchSnapshot();
     });