[Discover] Migrate session session initialization to state container (e…

…lastic#156957)

## Summary

Migrate `useSearchSession` to state container's `initializeAndSync` function, to remove another `hook`.

src/plugins/discover/public/application/main/discover_main_app.tsx

@@ -9,7 +9,6 @@ import React, { useCallback, useEffect } from 'react';
 import { RootDragDropProvider } from '@kbn/dom-drag-drop';
 import { useHistory } from 'react-router-dom';
 import { useUrlTracking } from './hooks/use_url_tracking';
-import { useSearchSession } from './hooks/use_search_session';
 import { DiscoverStateContainer } from './services/discover_state';
 import { DiscoverLayout } from './components/layout';
 import { setBreadcrumbsTitle } from '../../utils/breadcrumbs';
@@ -44,11 +43,6 @@ export function DiscoverMainApp(props: DiscoverMainProps) {
 
   useUrlTracking(stateContainer.savedSearchState);
 
-  /**
-   * Search session logic
-   */
-  useSearchSession({ services, stateContainer });
-
   /**
    * Adhoc data views functionality
    */

src/plugins/discover/public/application/main/services/discover_state.test.ts

@@ -34,7 +34,7 @@ const startSync = (appState: DiscoverAppStateContainer) => {
   return stop;
 };
 
-async function getState(url: string, savedSearch?: SavedSearch) {
+async function getState(url: string = '/', savedSearch?: SavedSearch) {
   const nextHistory = createBrowserHistory();
   nextHistory.push(url);
   const nextState = getDiscoverStateContainer({
@@ -237,6 +237,16 @@ describe('createSearchSessionRestorationDataProvider', () => {
   });
 });
 
+describe('searchSessionManager', () => {
+  test('getting the next session id', async () => {
+    const { state } = await getState();
+    const nextId = 'id';
+    discoverServiceMock.data.search.session.start = jest.fn(() => nextId);
+    state.actions.initializeAndSync();
+    expect(state.searchSessionManager.getNextSearchSessionId()).toBe(nextId);
+  });
+});
+
 describe('actions', () => {
   beforeEach(async () => {
     discoverServiceMock.data.query.timefilter.timefilter.getTime = jest.fn(() => {

src/plugins/discover/public/application/main/services/discover_state.ts

@@ -16,6 +16,7 @@ import {
 } from '@kbn/kibana-utils-plugin/public';
 import {
   DataPublicPluginStart,
+  noSearchSessionStorageCapabilityMessage,
   QueryState,
   SearchSessionInfoProvider,
 } from '@kbn/data-plugin/public';
@@ -382,6 +383,23 @@ export function getDiscoverStateContainer({
       fetchData();
     });
 
+    services.data.search.session.enableStorage(
+      createSearchSessionRestorationDataProvider({
+        appStateContainer,
+        data: services.data,
+        getSavedSearch: () => savedSearchContainer.getState(),
+      }),
+      {
+        isDisabled: () =>
+          services.capabilities.discover.storeSearchSession
+            ? { disabled: false }
+            : {
+                disabled: true,
+                reasonText: noSearchSessionStorageCapabilityMessage,
+              },
+      }
+    );
+
     return () => {
       unsubscribeData();
       appStateUnsubscribe();