Skip to content

Commit

Permalink
Registered model table follow-up and cypress testing
Browse files Browse the repository at this point in the history
  • Loading branch information
dpanshug committed Apr 17, 2024
1 parent 98bedee commit 3ef803f
Show file tree
Hide file tree
Showing 24 changed files with 562 additions and 58 deletions.
58 changes: 58 additions & 0 deletions frontend/src/__mocks__/mockModelRegistry.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,58 @@
import { ModelRegistryKind } from '~/k8sTypes';

type MockModelRegistryType = {
name?: string;
namespace?: string;
};

export const mockModelRegistry = ({
name = 'modelregistry-sample',
namespace = 'odh-model-registries',
}: MockModelRegistryType): ModelRegistryKind => ({
apiVersion: 'modelregistry.opendatahub.io/v1alpha1',
kind: 'ModelRegistry',
metadata: {
name,
creationTimestamp: '2024-03-14T08:01:42Z',
namespace,
},
spec: {
grpc: {
port: 9090,
},
postgres: {
database: 'model-registry',
host: 'model-registry-db',
passwordSecret: {
key: 'database-password',
name: 'model-registry-db',
},
port: 5432,
skipDBCreation: false,
sslMode: 'disable',
username: 'mlmduser',
},
rest: {
port: 8080,
serviceRoute: 'disabled',
},
},
status: {
conditions: [
{
lastTransitionTime: '2024-03-22T09:30:02Z',
message: 'Deployment for custom resource modelregistry-sample was successfully created',
reason: 'CreatedDeployment',
status: 'True',
type: 'Progressing',
},
{
lastTransitionTime: '2024-03-14T08:11:26Z',
message: 'Deployment for custom resource modelregistry-sample is available',
reason: 'DeploymentAvailable',
status: 'True',
type: 'Available',
},
],
},
});
20 changes: 20 additions & 0 deletions frontend/src/__mocks__/mockModelVersion.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,20 @@
import { ModelVersion, ModelVersionState } from '~/concepts/modelRegistry/types';

type MockModelVersionType = {
author?: string;
registeredModelID?: string;
};

export const mockModelVersion = ({
author = 'Test author',
registeredModelID = '1',
}: MockModelVersionType): ModelVersion => ({
author,
createTimeSinceEpoch: '1712234877179',
customProperties: {},
id: '26',
lastUpdateTimeSinceEpoch: '1712234877179',
name: 'fraud detection model version 1',
state: ModelVersionState.ARCHIVED,
registeredModelID,
});
9 changes: 9 additions & 0 deletions frontend/src/__mocks__/mockModelVersionList.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
import { ModelVersionList } from '~/concepts/modelRegistry/types';
import { mockModelVersion } from './mockModelVersion';

export const mockModelVersionList = (): ModelVersionList => ({
items: [mockModelVersion({ author: 'Author 1', registeredModelID: '1' })],
nextPageToken: '',
pageSize: 0,
size: 1,
});
23 changes: 18 additions & 5 deletions frontend/src/__mocks__/mockRegisteredModel.ts
Original file line number Diff line number Diff line change
@@ -1,17 +1,30 @@
import { RegisteredModel, RegisteredModelState } from '~/concepts/modelRegistry/types';
import {
ModelRegistryBase,
RegisteredModel,
RegisteredModelState,
} from '~/concepts/modelRegistry/types';

type MockRegisteredModelType = { name?: string; state?: RegisteredModelState };
type MockRegisteredModelType = {
id?: string;
name?: string;
state?: RegisteredModelState;
description?: string;
customProperties?: ModelRegistryBase['customProperties'];
};

export const mockRegisteredModel = ({
name = 'test',
state = RegisteredModelState.LIVE,
description = '',
customProperties = {},
id = '1',
}: MockRegisteredModelType): RegisteredModel => ({
createTimeSinceEpoch: '1710404288975',
description: 'test',
description,
externalID: '1234132asdfasdf',
id: '1',
id,
lastUpdateTimeSinceEpoch: '1710404288975',
name,
state,
customProperties: {},
customProperties,
});
108 changes: 107 additions & 1 deletion frontend/src/__mocks__/mockRegisteredModelsList.ts
Original file line number Diff line number Diff line change
@@ -1,8 +1,114 @@
/* eslint-disable camelcase */
import { RegisteredModelList } from '~/concepts/modelRegistry/types';
import { mockRegisteredModel } from './mockRegisteredModel';

export const mockRegisteredModelList = (): RegisteredModelList => ({
items: [mockRegisteredModel({ name: 'test-1' }), mockRegisteredModel({ name: 'test-2' })],
items: [
mockRegisteredModel({
name: 'Fraud detection model',
description:
'A machine learning model trained to detect fraudulent transactions in financial data',
customProperties: {
Financial: {
metadataType: 'MetadataStringValue',
string_value: 'non-empty',
},
'Financial data': {
metadataType: 'MetadataStringValue',
string_value: '',
},
'Fraud detection': {
metadataType: 'MetadataStringValue',
string_value: '',
},
'Test label': {
metadataType: 'MetadataStringValue',
string_value: '',
},
'Machine learning': {
metadataType: 'MetadataStringValue',
string_value: '',
},
'Next data to be overflow': {
metadataType: 'MetadataStringValue',
string_value: '',
},
},
}),
mockRegisteredModel({
name: 'Credit Scoring',
customProperties: {
'Credit Score Predictor': {
metadataType: 'MetadataStringValue',
string_value: '',
},
'Creditworthiness scoring system': {
metadataType: 'MetadataStringValue',
string_value: '',
},
'Default Risk Analyzer': {
metadataType: 'MetadataStringValue',
string_value: '',
},
'Portfolio Management': {
metadataType: 'MetadataStringValue',
string_value: '',
},
'Risk Assessment': {
metadataType: 'MetadataStringValue',
string_value: '',
},
},
}),
mockRegisteredModel({
name: 'Label modal',
description:
'A machine learning model trained to detect fraudulent transactions in financial data',
customProperties: {
'Testing label': {
metadataType: 'MetadataStringValue',
string_value: '',
},
Financial: {
metadataType: 'MetadataStringValue',
string_value: 'non-empty',
},
'Financial data': {
metadataType: 'MetadataStringValue',
string_value: '',
},
'Fraud detection': {
metadataType: 'MetadataStringValue',
string_value: '',
},
'Long label data to be truncated abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc abc':
{
metadataType: 'MetadataStringValue',
string_value: '',
},
'Machine learning': {
metadataType: 'MetadataStringValue',
string_value: '',
},
'Next data to be overflow': {
metadataType: 'MetadataStringValue',
string_value: '',
},
'Label x': {
metadataType: 'MetadataStringValue',
string_value: '',
},
'Label y': {
metadataType: 'MetadataStringValue',
string_value: '',
},
'Label z': {
metadataType: 'MetadataStringValue',
string_value: '',
},
},
}),
],
nextPageToken: '',
pageSize: 0,
size: 4,
Expand Down
57 changes: 57 additions & 0 deletions frontend/src/__mocks__/mockRouteK8sResource.ts
Original file line number Diff line number Diff line change
Expand Up @@ -140,3 +140,60 @@ export const mockRouteK8sResourceModelServing = ({
],
},
});

export const mockRouteK8sResourceModelRegistry = ({
name = 'modelregistry-sample',
namespace = 'shared',
}: MockResourceConfigType): RouteKind => ({
kind: 'Route',
apiVersion: 'route.openshift.io/v1',
metadata: {
name,
namespace,
uid: genUID('route'),
resourceVersion: '4789458',
creationTimestamp: '2023-02-14T21:44:13Z',
labels: {
app: name,
component: 'model-registry',
},
annotations: {
'openshift.io/host.generated': 'true',
},
managedFields: [],
},
spec: {
path: '',
host: `${name}-${namespace}.apps.user.com`,
to: {
kind: 'Service',
name,
weight: 100,
},
port: {
targetPort: 'oauth-proxy',
},
tls: {
termination: 'reencrypt',
insecureEdgeTerminationPolicy: 'Redirect',
},
wildcardPolicy: 'None',
},
status: {
ingress: [
{
host: `${name}-${namespace}.apps.user.com`,
routerName: 'default',
conditions: [
{
type: 'Admitted',
status: 'True',
lastTransitionTime: '2023-02-14T21:44:13Z',
},
],
wildcardPolicy: 'None',
routerCanonicalHostname: 'router-default.apps.user.com',
},
],
},
});
Loading

0 comments on commit 3ef803f

Please sign in to comment.