Task/APPENG-2738: NIM Cypress E2E tests

frontend/src/__mocks__/mockDashboardConfig.ts

@@ -1,7 +1,7 @@
 import { DashboardConfigKind, KnownLabels } from '~/k8sTypes';
 import { NotebookSize } from '~/types';
 
-type MockDashboardConfigType = {
+export type MockDashboardConfigType = {
   disableInfo?: boolean;
   disableSupport?: boolean;
   disableClusterManager?: boolean;

frontend/src/__mocks__/mockNimResource.ts

@@ -0,0 +1,132 @@
+import {
+  ConfigMapKind,
+  InferenceServiceKind,
+  PersistentVolumeClaimKind,
+  SecretKind,
+  ServingRuntimeKind,
+  TemplateKind,
+} from '~/k8sTypes';
+import { ServingRuntimeAPIProtocol, ServingRuntimePlatform } from '~/types';
+import { mockConfigMap } from './mockConfigMap';
+import { mockServingRuntimeK8sResource } from './mockServingRuntimeK8sResource';
+import { mockInferenceServiceK8sResource } from './mockInferenceServiceK8sResource';
+import { mockServingRuntimeTemplateK8sResource } from './mockServingRuntimeTemplateK8sResource';
+import { mockSecretK8sResource } from './mockSecretK8sResource';
+import { mockPVCK8sResource } from './mockPVCK8sResource';
+
+export const mockNimImages = (): ConfigMapKind =>
+  mockConfigMap({
+    name: 'nvidia-nim-images-data',
+    namespace: 'opendatahub',
+    data: {
+      alphafold2:
+        '{' +
+        '   "name": "alphafold2",' +
+        '   "displayName": "AlphaFold2",' +
+        '   "shortDescription": "A widely used model for predicting the 3D structures of proteins from their amino acid sequences.",' +
+        '   "namespace": "nim/deepmind",' +
+        '   "tags": [' +
+        '   "1.0.0"' +
+        '   ],' +
+        '   "latestTag": "1.0.0",' +
+        '   "updatedDate": "2024-08-27T01:51:55.642Z"' +
+        '  }',
+      'arctic-embed-l':
+        '{' +
+        '   "name": "arctic-embed-l",' +
+        '   "displayName": "Snowflake Arctic Embed Large Embedding",' +
+        '   "shortDescription": "NVIDIA NIM for GPU accelerated Snowflake Arctic Embed Large Embedding inference",' +
+        '   "namespace": "nim/snowflake",' +
+        '   "tags": [' +
+        '   "1.0.1",' +
+        '   "1.0.0"' +
+        '   ],' +
+        '   "latestTag": "1.0.1",' +
+        '   "updatedDate": "2024-07-27T00:38:40.927Z"' +
+        '  }',
+    },
+  });
+
+export const mockNimInferenceService = (): InferenceServiceKind => {
+  const inferenceService = mockInferenceServiceK8sResource({
+    name: 'test-name',
+    modelName: 'test-name',
+    displayName: 'Test Name',
+    kserveInternalLabel: true,
+    resources: {
+      limits: { cpu: '2', memory: '8Gi' },
+      requests: { cpu: '1', memory: '4Gi' },
+    },
+  });
+  delete inferenceService.metadata.labels?.name;
+  delete inferenceService.metadata.creationTimestamp;
+  delete inferenceService.metadata.generation;
+  delete inferenceService.metadata.resourceVersion;
+  delete inferenceService.metadata.uid;
+  if (inferenceService.spec.predictor.model?.modelFormat) {
+    inferenceService.spec.predictor.model.modelFormat.name = 'arctic-embed-l';
+  }
+  delete inferenceService.spec.predictor.model?.modelFormat?.version;
+  delete inferenceService.spec.predictor.model?.storage;
+  delete inferenceService.status;
+
+  return inferenceService;
+};
+
+export const mockNimServingRuntime = (): ServingRuntimeKind => {
+  const servingRuntime = mockServingRuntimeK8sResource({
+    name: 'test-name',
+    displayName: 'Test Name',
+  });
+  if (servingRuntime.metadata.annotations) {
+    servingRuntime.metadata.annotations['opendatahub.io/template-display-name'] = 'NVIDIA NIM';
+    servingRuntime.metadata.annotations['opendatahub.io/template-name'] = 'nvidia-nim-runtime';
+  }
+
+  return servingRuntime;
+};
+
+export const mockNimServingRuntimeTemplate = (): TemplateKind => {
+  const templateMock = mockServingRuntimeTemplateK8sResource({
+    name: 'nvidia-nim-serving-template',
+    displayName: 'NVIDIA NIM',
+    platforms: [ServingRuntimePlatform.SINGLE],
+    apiProtocol: ServingRuntimeAPIProtocol.REST,
+    namespace: 'opendatahub',
+  });
+  if (templateMock.metadata.annotations != null) {
+    templateMock.metadata.annotations['opendatahub.io/dashboard'] = 'true';
+  }
+
+  return templateMock;
+};
+
+export const mockNvidiaNimAccessSecret = (): SecretKind => {
+  const secret = mockSecretK8sResource({
+    name: 'nvidia-nim-access',
+  });
+  delete secret.data;
+  secret.data = {};
+  secret.data.api_key = 'api-key'; // eslint-disable-line camelcase
+  secret.data.configMapName = 'bnZpZGlhLW5pbS12YWxpZGF0aW9uLXJlc3VsdA==';
+
+  return secret;
+};
+
+export const mockNvidiaNimImagePullSecret = (): SecretKind => {
+  const secret = mockSecretK8sResource({
+    name: 'nvidia-nim-image-pull',
+  });
+  delete secret.data;
+  secret.data = {};
+  secret.data['.dockerconfigjson'] = 'ZG9ja2VyY29uZmlnCg==';
+
+  return secret;
+};
+
+export const mockNimModelPVC = (): PersistentVolumeClaimKind => {
+  const pvc = mockPVCK8sResource({
+    name: 'nim-pvc',
+  });
+  return pvc;
+};

frontend/src/__tests__/cypress/cypress.config.ts

@@ -16,6 +16,7 @@ import { env, cypressEnv, BASE_URL } from '~/__tests__/cypress/cypress/utils/tes
 const resultsDir = `${env.CY_RESULTS_DIR || 'results'}/${env.CY_MOCK ? 'mocked' : 'e2e'}`;
 
 export default defineConfig({
+  experimentalStudio: false,
   experimentalMemoryManagement: true,
   // Use relative path as a workaround to https://github.com/cypress-io/cypress/issues/6406
   reporter: '../../../node_modules/cypress-multi-reporters',

frontend/src/__tests__/cypress/cypress/pages/nimModelDialog.ts

@@ -0,0 +1,45 @@
+import { Modal } from '~/__tests__/cypress/cypress/pages/components/Modal';
+
+class NIMDeployModal extends Modal {
+  constructor(private edit = false) {
+    super(`${edit ? 'Edit' : 'Deploy'} model with NVIDIA NIM`);
+  }
+
+  findSubmitButton() {
+    return this.findFooter().findByTestId('modal-submit-button');
+  }
+
+  findModelNameInput() {
+    return this.find().findByTestId('model-deployment-name-section');
+  }
+
+  findNIMToDeploy() {
+    return this.find().findByTestId('nim-model-list-selection');
+  }
+
+  findNimStorageSizeInput() {
+    return cy.get('[data-testid="pvc-size"] input');
+  }
+
+  findStorageSizeMinusButton() {
+    return this.find().findByTestId('pvc-size').findByRole('button', { name: 'Minus' });
+  }
+
+  findStorageSizePlusButton() {
+    return this.find().findByTestId('pvc-size').findByRole('button', { name: 'Plus' });
+  }
+
+  findNimModelReplicas() {
+    return cy.get('[id="model-server-replicas"]');
+  }
+
+  findNimModelReplicasMinusButton() {
+    return this.find().find('button[aria-label="Minus"]').eq(1);
+  }
+
+  findNimModelReplicasPlusButton() {
+    return this.find().find('button[aria-label="Plus"]').eq(1);
+  }
+}
+
+export const nimDeployModal = new NIMDeployModal();

frontend/src/__tests__/cypress/cypress/tests/mocked/modelServing/modelServingNim.cy.ts

@@ -0,0 +1,42 @@
+import {
+  initInterceptsToEnableNim,
+  validateNimInmferenceModelsTable,
+} from '~/__tests__/cypress/cypress/utils/nimUtils';
+import { mockNimInferenceService, mockNimServingRuntime } from '~/__mocks__/mockNimResource';
+import {
+  InferenceServiceModel,
+  ServingRuntimeModel,
+} from '~/__tests__/cypress/cypress/utils/models';
+import { mockK8sResourceList } from '~/__mocks__';
+import { modelServingGlobal } from '~/__tests__/cypress/cypress/pages/modelServing';
+
+describe('NIM Models Deployments', () => {
+  it('should be listed in the global models list', () => {
+    initInterceptsToEnableNim({ hasAllModels: false });
+    cy.interceptK8sList(InferenceServiceModel, mockK8sResourceList([mockNimInferenceService()]));
+    cy.interceptK8sList(ServingRuntimeModel, mockK8sResourceList([mockNimServingRuntime()]));
+
+    modelServingGlobal.visit('test-project');
+
+    validateNimInmferenceModelsTable();
+  });
+
+  it('should only be allowed to be deleted, no edit', () => {
+    initInterceptsToEnableNim({});
+    cy.interceptK8sList(InferenceServiceModel, mockK8sResourceList([mockNimInferenceService()]));
+    cy.interceptK8sList(ServingRuntimeModel, mockK8sResourceList([mockNimServingRuntime()]));
+
+    modelServingGlobal.visit('test-project');
+    modelServingGlobal.getModelRow('Test Name').get('button[aria-label="Kebab toggle"]').click();
+
+    modelServingGlobal
+      .getModelRow('Test Name')
+      .get('button[role="menuitem"]')
+      .should('have.length', 1);
+    modelServingGlobal
+      .getModelRow('Test Name')
+      .get('button[role="menuitem"]')
+      .contains('Delete')
+      .should('exist');
+  });
+});