cover sever testcases (#119)

Author: RidaZubair

tests/e2e/pages/cli/anaconda-ai-cmds.ts

@@ -1,7 +1,7 @@
 import { type ShellResult, shellCommand, stripAnsiSgrAndTrim, verifyShellExitCode } from 'tests/utils/CliUtils';
 import * as cliCommands from './cliCommands';
 import { expect } from 'tests/test-setup/page-setup';
-import { INVALID_MODEL_ERROR_MESSAGE, ModelApi, ServerApi } from '@testdata/model-api';
+import { INVALID_MODEL_ERROR_MESSAGE, INVALID_SERVER_NAME, ModelApi, ServerApi } from '@testdata/model-api';
 
 // CLI helpers for Anaconda AI package CLI commands.
 export class AnacondaAiCli {
@@ -59,9 +59,10 @@ export class AnacondaAiCli {
     const server = servers.find(s => s.model === expectedModelFile);
 
     expect(server, `Expected server for model "${expectedModelFile}" to appear in servers list`).toBeDefined();
-    expect(server!.server_id, `Expected server_id to contain model name "${modelName}"`).toContain(modelName);
-    expect(server!.model, `Expected model to be "${expectedModelFile}"`).toBe(expectedModelFile);
-    expect(server!.status, `Expected server status to be "running"`).toBe('running');
+    const { server_id, model, status } = server as ServerApi;
+    expect(server_id, `Expected server_id to contain model name "${modelName}"`).toContain(modelName);
+    expect(model, `Expected model to be "${expectedModelFile}"`).toBe(expectedModelFile);
+    expect(status, `Expected server status to be "running"`).toBe('running');
   }
 
   // Executes `anaconda ai download <model>/<quant>` (positional; no --model)
@@ -163,6 +164,22 @@ export class AnacondaAiCli {
     ).toBeTruthy();
   }
 
+  // Executes `anaconda ai stop <serverName>` with a non-existent server name
+  public async runStopModelNotFoundCommand(): Promise<ShellResult> {
+    return await shellCommand(cliCommands.stopModelNotFoundCmd(INVALID_SERVER_NAME));
+  }
+
+  // Validates that stopping a non-existent server exits non-zero with a ServerNotFoundError
+  public verifyStopModelNotFoundCommand(result: ShellResult): void {
+    expect(result.exitCode, 'Expected non-zero exit code for unknown server').not.toBe(0);
+
+    const output = stripAnsiSgrAndTrim(result.output).toLowerCase();
+    expect(
+      output.includes('servernotfounderror') || output.includes('was not found'),
+      'Expected ServerNotFoundError for unknown server',
+    ).toBeTruthy();
+  }
+
   // Validates that launching with an invalid format exits non-zero with a ValueError
   public verifyLaunchModelInvalidFormatCommand(result: ShellResult): void {
     expect(result.exitCode, 'Expected non-zero exit code for invalid format').not.toBe(0);

tests/e2e/pages/cli/cliCommands.ts

@@ -51,3 +51,7 @@ export const stopModelCmd = (modelName: string, modelQuantization: string): stri
 // Stop and remove server — same as stop but with --rm to delete it
 export const stopAndRemoveModelCmd = (modelName: string, modelQuantization: string): string =>
   condaRun(`anaconda ai stop ${modelName}_${modelQuantization}.gguf --rm`);
+
+// Negative: stop a server that does not exist — triggers ServerNotFoundError
+export const stopModelNotFoundCmd = (serverName: string): string =>
+  condaRun(`anaconda ai stop ${serverName}`);

tests/e2e/specs/cli/anaconda-ai.spec.ts

@@ -101,4 +101,9 @@ test.describe('Anaconda AI CLI Commands @anaconda-ai', () => {
     const result = await anacondaAiCli.runLaunchModelCommand(INVALID_MODEL_NAME, DOWNLOAD_TEST_MODEL_QUANTIZATION);
     anacondaAiCli.verifyLaunchModelNotFoundCommand(result);
   });
+
+  test('anaconda ai stop - unknown server returns ServerNotFoundError', async ({ anacondaAiCli }) => {
+    const result = await anacondaAiCli.runStopModelNotFoundCommand();
+    anacondaAiCli.verifyStopModelNotFoundCommand(result);
+  });
 });

tests/e2e/testdata/model-api.ts

@@ -29,3 +29,6 @@ export const DOWNLOAD_TEST_MODEL_QUANTIZATION = 'q4_k_m';
 export const INVALID_MODEL_NAME = 'invalid-model';
 export const INVALID_MODEL_QUANTIZATION = 'invalid-quantization';
 export const INVALID_MODEL_ERROR_MESSAGE = 'you must include the quantization method in the model';
+
+// Invalid server name for negative stop tests
+export const INVALID_SERVER_NAME = 'invalid-server';