chore(test): add restart service model e2e test (#3097)

Signed-off-by: Vladimir Lazar <[email protected]>

Author: cbr7

tests/playwright/src/ai-lab-extension.spec.ts

@@ -324,6 +324,24 @@ test.describe.serial(`AI Lab extension installation and verification`, () => {
         }).toPass({ timeout: 600_000, intervals: [5_000] });
       });
 
+      test(`Restart model service for ${modelName}`, async () => {
+        test.skip(modelName === 'ggerganov/whisper.cpp');
+        test.setTimeout(180_000);
+
+        await modelServiceDetailsPage.stopService();
+        await playExpect(modelServiceDetailsPage.startServiceButton).toBeEnabled({ timeout: 120_000 });
+        await playExpect
+          // eslint-disable-next-line sonarjs/no-nested-functions
+          .poll(async () => await modelServiceDetailsPage.getServiceState(), { timeout: 120_000 })
+          .toBe('');
+
+        await modelServiceDetailsPage.startService();
+        await playExpect
+          // eslint-disable-next-line sonarjs/no-nested-functions
+          .poll(async () => await modelServiceDetailsPage.getServiceState(), { timeout: 120_000 })
+          .toBe('RUNNING');
+      });
+
       test(`Delete model service for ${modelName}`, async () => {
         test.setTimeout(150_000);
         const modelServicePage = await modelServiceDetailsPage.deleteService();

tests/playwright/src/model/ai-lab-service-details-page.ts

@@ -29,6 +29,7 @@ export class AILabServiceDetailsPage extends AILabBasePage {
   readonly codeSnippet: Locator;
   readonly deleteServiceButton: Locator;
   readonly stopServiceButton: Locator;
+  readonly startServiceButton: Locator;
 
   constructor(page: Page, webview: Page) {
     super(page, webview, 'Service details');
@@ -38,6 +39,7 @@ export class AILabServiceDetailsPage extends AILabBasePage {
     this.codeSnippet = this.webview.getByLabel('Code Snippet', { exact: true });
     this.deleteServiceButton = this.webview.getByRole('button', { name: 'Delete service' });
     this.stopServiceButton = this.webview.getByRole('button', { name: 'Stop service' });
+    this.startServiceButton = this.webview.getByRole('button', { name: 'Start service' });
   }
 
   async waitForLoad(): Promise<void> {
@@ -51,6 +53,16 @@ export class AILabServiceDetailsPage extends AILabBasePage {
     return new AiModelServicePage(this.page, this.webview);
   }
 
+  async stopService(): Promise<void> {
+    await playExpect(this.stopServiceButton).toBeEnabled();
+    await this.stopServiceButton.click();
+  }
+
+  async startService(): Promise<void> {
+    await playExpect(this.startServiceButton).toBeEnabled();
+    await this.startServiceButton.click();
+  }
+
   async getInferenceServerPort(): Promise<string> {
     const split = (await this.endpointURL.textContent())?.split(':');
     const port = split ? split[split.length - 1].split('/')[0] : '';