[8.0] [ML] Fix flaky Trained Models tests (#125484) (#125623)

* [ML] Fix flaky Trained Models tests  (#125484)

* add extra assertion

* prevent extra call on init

* fix stats call

(cherry picked from commit b51c87f24c)

# Conflicts:
#	x-pack/test/functional/services/ml/common_ui.ts
#	x-pack/test/functional/services/ml/index.ts

* update refresh button check

x-pack/plugins/ml/public/application/services/ml_api_service/trained_models.ts

@@ -81,10 +81,7 @@ export function trainedModelsApiProvider(httpService: HttpService) {
      * @param params - Optional query params
      */
     getTrainedModelStats(modelId?: string | string[], params?: InferenceStatsQueryParams) {
-      let model = modelId ?? '_all';
-      if (Array.isArray(modelId)) {
-        model = modelId.join(',');
-      }
+      const model = (Array.isArray(modelId) ? modelId.join(',') : modelId) || '_all';
 
       return httpService.http<InferenceStatsResponse>({
         path: `${apiBasePath}/trained_models/${model}/_stats`,

x-pack/plugins/ml/public/application/trained_models/models_management/models_list.tsx

@@ -190,6 +190,7 @@ export const ModelsList: FC = () => {
 
   useEffect(
     function updateOnTimerRefresh() {
+      if (!refresh) return;
       fetchModelsData();
     },
     [refresh]

x-pack/test/functional/apps/ml/model_management/model_list.ts

@@ -14,8 +14,6 @@ export default function ({ getService }: FtrProviderContext) {
     before(async () => {
       await ml.trainedModels.createTestTrainedModels('classification', 15, true);
       await ml.trainedModels.createTestTrainedModels('regression', 15);
-      await ml.securityUI.loginAsMlPowerUser();
-      await ml.navigation.navigateToTrainedModels();
     });
 
     after(async () => {
@@ -46,6 +44,7 @@ export default function ({ getService }: FtrProviderContext) {
       before(async () => {
         await ml.securityUI.loginAsMlPowerUser();
         await ml.navigation.navigateToTrainedModels();
+        await ml.trainedModelsTable.waitForRefreshButtonLoaded();
       });
 
       after(async () => {
@@ -173,6 +172,7 @@ export default function ({ getService }: FtrProviderContext) {
       before(async () => {
         await ml.securityUI.loginAsMlViewer();
         await ml.navigation.navigateToTrainedModels();
+        await ml.trainedModelsTable.waitForRefreshButtonLoaded();
       });
 
       after(async () => {

x-pack/test/functional/services/ml/common_ui.ts

@@ -309,5 +309,9 @@ export function MachineLearningCommonUIProvider({
       await PageObjects.spaceSelector.goToSpecificSpace(spaceId);
       await PageObjects.spaceSelector.expectHomePage(spaceId);
     },
+
+    async waitForDatePickerIndicatorLoaded() {
+      await testSubjects.waitForEnabled('superDatePickerApplyTimeButton');
+    },
   };
 }

x-pack/test/functional/services/ml/index.ts

@@ -123,7 +123,7 @@ export function MachineLearningProvider(context: FtrProviderContext) {
   const alerting = MachineLearningAlertingProvider(context, commonUI);
   const swimLane = SwimLaneProvider(context);
   const trainedModels = TrainedModelsProvider(context, api, commonUI);
-  const trainedModelsTable = TrainedModelsTableProvider(context);
+  const trainedModelsTable = TrainedModelsTableProvider(context, commonUI);
 
   return {
     anomaliesTable,

x-pack/test/functional/services/ml/trained_models_table.ts

@@ -10,7 +10,8 @@ import { ProvidedType } from '@kbn/test';
 import { upperFirst } from 'lodash';
 
 import { WebElementWrapper } from 'test/functional/services/lib/web_element_wrapper';
-import { FtrProviderContext } from '../../ftr_provider_context';
+import type { FtrProviderContext } from '../../ftr_provider_context';
+import type { MlCommonUI } from './common_ui';
 
 export interface TrainedModelRowData {
   id: string;
@@ -20,7 +21,10 @@ export interface TrainedModelRowData {
 
 export type MlTrainedModelsTable = ProvidedType<typeof TrainedModelsTableProvider>;
 
-export function TrainedModelsTableProvider({ getService }: FtrProviderContext) {
+export function TrainedModelsTableProvider(
+  { getService }: FtrProviderContext,
+  mlCommonUI: MlCommonUI
+) {
   const testSubjects = getService('testSubjects');
   const retry = getService('retry');
 
@@ -218,6 +222,7 @@ export function TrainedModelsTableProvider({ getService }: FtrProviderContext) {
           await testSubjects.existOrFail('mlTrainedModelRowDetails', { timeout: 1000 });
         }
       });
+      await this.waitForRefreshButtonLoaded();
     }
 
     public async assertTabContent(