[Logs Explorer] Add test suite for Dataset Selector

x-pack/plugins/discover_log_explorer/public/components/dataset_selector/dataset_selector.tsx

@@ -168,6 +168,7 @@ export function DatasetSelector({
         onPanelChange={changePanel}
         className="eui-yScroll"
         css={contextMenuStyles}
+        data-test-subj="datasetSelectorContextMenu"
         size="s"
       />
     </DatasetsPopover>

x-pack/plugins/discover_log_explorer/public/components/dataset_selector/sub_components/datasets_list.tsx

@@ -44,6 +44,7 @@ export const DatasetsList = ({
   if (hasError) {
     return (
       <EuiEmptyPrompt
+        data-test-subj="datasetErrorPrompt"
         iconType="warning"
         iconColor="danger"
         paddingSize="m"
@@ -70,6 +71,7 @@ export const DatasetsList = ({
   if (isEmpty) {
     return (
       <EuiEmptyPrompt
+        data-test-subj="emptyDatasetPrompt"
         iconType="search"
         paddingSize="m"
         title={<h2>{noDatasetsLabel}</h2>}

x-pack/plugins/discover_log_explorer/public/components/dataset_selector/sub_components/datasets_popover.tsx

@@ -45,15 +45,16 @@ export const DatasetsPopover = ({
   return (
     <EuiPopover
       id={POPOVER_ID}
-      data-test-subj={POPOVER_ID}
+      data-test-subj="datasetSelectorPopover"
+      anchorPosition={isMobile ? 'downCenter' : 'downLeft'}
       button={
         <EuiButton
           css={buttonStyles}
           iconType="arrowDown"
           iconSide="right"
           onClick={onClick}
           fullWidth={isMobile}
-          data-test-subj={`${POPOVER_ID}-button`}
+          data-test-subj="datasetSelectorPopoverButton"
         >
           {iconType ? (
             <EuiIcon type={iconType} />
@@ -74,7 +75,12 @@ export const DatasetsPopover = ({
       {...(isMobile && { display: 'block' })}
       {...props}
     >
-      <EuiPanel paddingSize="none" hasShadow={false} css={panelStyle}>
+      <EuiPanel
+        paddingSize="none"
+        hasShadow={false}
+        css={panelStyle}
+        data-test-subj="datasetSelectorContent"
+      >
         <Title size="xxs">
           <span>{selectDatasetLabel}</span>
         </Title>

x-pack/plugins/discover_log_explorer/public/components/dataset_selector/sub_components/datasets_skeleton.tsx

@@ -10,7 +10,7 @@ import { EuiPanel, EuiSkeletonText } from '@elastic/eui';
 import { uncategorizedLabel } from '../constants';
 
 export const DatasetSkeleton = () => (
-  <EuiPanel>
+  <EuiPanel data-test-subj="datasetSelectorSkeleton">
     <EuiSkeletonText lines={7} isLoading contentAriaLabel={uncategorizedLabel} />
   </EuiPanel>
 );

x-pack/plugins/discover_log_explorer/public/components/dataset_selector/sub_components/integrations_list_status.tsx

@@ -34,6 +34,7 @@ export const IntegrationsListStatus = ({
   if (hasError) {
     return (
       <EuiEmptyPrompt
+        data-test-subj="integrationsErrorPrompt"
         iconType="warning"
         iconColor="danger"
         paddingSize="m"

x-pack/plugins/discover_log_explorer/public/components/dataset_selector/sub_components/search_controls.tsx

@@ -33,7 +33,12 @@ export const SearchControls = ({ search, onSearch, onSort, isLoading }: SearchCo
   };
 
   return (
-    <EuiPanel paddingSize="s" hasShadow={false} css={{ width: DATA_VIEW_POPOVER_CONTENT_WIDTH }}>
+    <EuiPanel
+      paddingSize="s"
+      hasShadow={false}
+      css={{ width: DATA_VIEW_POPOVER_CONTENT_WIDTH }}
+      data-test-subj="datasetSelectorSearchControls"
+    >
       <EuiFlexGroup gutterSize="xs" responsive={false}>
         <EuiFlexItem>
           <EuiFieldSearch

x-pack/plugins/discover_log_explorer/public/components/dataset_selector/utils.tsx

@@ -61,6 +61,7 @@ export const buildIntegrationsTree = ({
       integrationsTree.items.push({
         name: title,
         icon: <PackageIcon packageName={name} version={version} size="m" icons={icons} tryApi />,
+        'data-test-subj': integration.id,
         panel: integration.id,
         ...(isLastIntegration && { buttonRef: spyRef }),
       });
@@ -85,6 +86,7 @@ export const createAllLogDatasetsItem = ({ onClick }: { onClick(): void }) => {
   const allLogDataset = Dataset.createAllLogsDataset();
   return {
     name: allLogDataset.title,
+    'data-test-subj': 'allLogDatasets',
     icon: allLogDataset.iconType && <EuiIcon type={allLogDataset.iconType} />,
     onClick,
   };
@@ -93,6 +95,7 @@ export const createAllLogDatasetsItem = ({ onClick }: { onClick(): void }) => {
 export const createUnmanagedDatasetsItem = ({ onClick }: { onClick: LoadDatasets }) => {
   return {
     name: uncategorizedLabel,
+    'data-test-subj': 'unmanagedDatasets',
     icon: <EuiIcon type="documents" />,
     onClick,
     panel: UNMANAGED_STREAMS_PANEL_ID,
@@ -103,5 +106,6 @@ export const createIntegrationStatusItem = (props: IntegrationsListStatusProps)
   return {
     disabled: true,
     name: <IntegrationsListStatus {...props} />,
+    'data-test-subj': 'integrationStatusItem',
   };
 };

x-pack/plugins/discover_log_explorer/public/customizations/custom_dataset_filters.tsx

@@ -12,6 +12,8 @@ import { DataPublicPluginStart } from '@kbn/data-plugin/public';
 import { useControlPanels } from '../hooks/use_control_panels';
 import { LogExplorerProfileStateService } from '../state_machines/log_explorer_profile';
 
+const DATASET_FILTERS_CUSTOMIZATION_ID = 'datasetFiltersCustomization';
+
 interface CustomDatasetFiltersProps {
   logExplorerProfileStateService: LogExplorerProfileStateService;
   data: DataPublicPluginStart;
@@ -27,7 +29,7 @@ const CustomDatasetFilters = ({
   );
 
   return (
-    <ControlGroupContainer>
+    <ControlGroupContainer data-test-subj={DATASET_FILTERS_CUSTOMIZATION_ID}>
       <ControlGroupRenderer
         ref={setControlGroupAPI}
         getCreationOptions={getInitialInput}

x-pack/test/functional/apps/discover_log_explorer/columns_selection.ts

@@ -0,0 +1,57 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License
+ * 2.0; you may not use this file except in compliance with the Elastic License
+ * 2.0.
+ */
+import expect from '@kbn/expect';
+import { FtrProviderContext } from '../../ftr_provider_context';
+
+const defaultLogColumns = ['@timestamp', 'message'];
+
+export default function ({ getService, getPageObjects }: FtrProviderContext) {
+  const esArchiver = getService('esArchiver');
+  const retry = getService('retry');
+  const PageObjects = getPageObjects(['common', 'discover']);
+
+  describe('Columns selection initialization and update', () => {
+    before(async () => {
+      await esArchiver.load(
+        'x-pack/test/functional/es_archives/discover_log_explorer/data_streams'
+      );
+    });
+
+    after(async () => {
+      await esArchiver.unload(
+        'x-pack/test/functional/es_archives/discover_log_explorer/data_streams'
+      );
+    });
+
+    describe('when the log explorer profile loads', () => {
+      it("should initialize the table columns to logs' default selection", async () => {
+        await PageObjects.common.navigateToApp('discover', { hash: '/p/log-explorer' });
+
+        await PageObjects.discover.expandTimeRangeAsSuggestedInNoResultsMessage();
+
+        await retry.try(async () => {
+          expect(await PageObjects.discover.getColumnHeaders()).to.eql(defaultLogColumns);
+        });
+      });
+
+      it('should restore the table columns from the URL state if exists', async () => {
+        await PageObjects.common.navigateToApp('discover', {
+          hash: '/p/log-explorer?_a=(columns:!(message,data_stream.namespace))',
+        });
+
+        await PageObjects.discover.expandTimeRangeAsSuggestedInNoResultsMessage();
+
+        await retry.try(async () => {
+          expect(await PageObjects.discover.getColumnHeaders()).to.eql([
+            ...defaultLogColumns,
+            'data_stream.namespace',
+          ]);
+        });
+      });
+    });
+  });
+}

x-pack/test/functional/apps/discover_log_explorer/customization.ts

@@ -25,14 +25,14 @@ export default function ({ getService, getPageObjects }: FtrProviderContext) {
       it('DatasetSelector should replace the DataViewPicker', async () => {
         // Assert does not render on discover app
         await PageObjects.common.navigateToApp('discover');
-        await testSubjects.missingOrFail('dataset-selector-popover');
+        await testSubjects.missingOrFail('datasetSelectorPopover');
 
         // Assert it renders on log-explorer profile
         await PageObjects.common.navigateToApp('discover', { hash: '/p/log-explorer' });
-        await testSubjects.existOrFail('dataset-selector-popover');
+        await testSubjects.existOrFail('datasetSelectorPopover');
       });
 
-      it('the TopNav bar should hide New, Open and Save options', async () => {
+      it('the TopNav bar should hide then New, Open and Save options', async () => {
         // Assert does not render on discover app
         await PageObjects.common.navigateToApp('discover');
         await testSubjects.existOrFail('discoverNewButton');
@@ -59,6 +59,16 @@ export default function ({ getService, getPageObjects }: FtrProviderContext) {
         const results = await PageObjects.navigationalSearch.getDisplayedResults();
         expect(results[0].label).to.eql('Discover / Logs Explorer');
       });
+
+      it('should render a filter controls section as part of the unified search bar', async () => {
+        // Assert does not render on discover app
+        await PageObjects.common.navigateToApp('discover');
+        await testSubjects.missingOrFail('datasetFiltersCustomization');
+
+        // Assert it renders on log-explorer profile
+        await PageObjects.common.navigateToApp('discover', { hash: '/p/log-explorer' });
+        await testSubjects.existOrFail('datasetFiltersCustomization', { allowHidden: true });
+      });
     });
   });
 }

x-pack/test/functional/apps/discover_log_explorer/dataset_selection_state.ts

@@ -23,7 +23,7 @@ export default function ({ getService, getPageObjects }: FtrProviderContext) {
       });
     });
 
-    describe('when the "index" query param exist', () => {
+    describe('when the "index" query param exists', () => {
       it('should decode and restore the selection from a valid encoded index', async () => {
         const azureActivitylogsIndex =
           'BQZwpgNmDGAuCWB7AdgLmAEwIay+W6yWAtmKgOQSIDmIAtFgF4CuATmAHRZzwBu8sAJ5VadAFTkANAlhRU3BPyEiQASklFS8lu2kC55AII6wAAgAyNEFN5hWIJGnIBGDgFYOAJgDM5deCgeFAAVQQAHMgdkaihVIA===';
@@ -37,7 +37,7 @@ export default function ({ getService, getPageObjects }: FtrProviderContext) {
         expect(datasetSelectionTitle).to.be('[Azure Logs] activitylogs');
       });
 
-      it('should fallback to "All log datasets" selection and notify the user for an invalid encoded index', async () => {
+      it('should fallback to the "All log datasets" selection and notify the user of an invalid encoded index', async () => {
         const invalidEncodedIndex = 'invalid-encoded-index';
         await PageObjects.common.navigateToApp('discover', {
           hash: `/p/log-explorer?_a=(index:${encodeURIComponent(invalidEncodedIndex)})`,