frontend: Create unit tests for trials table (kubeflow#1441)

* Create unit tests for trials-table component. Signed-off-by: Elena Zioga <[email protected]>
arrikto · Nov 29, 2022 · cbe3c4e · cbe3c4e
1 parent c30a12c
commit cbe3c4e
Showing 1 changed file with 155 additions and 2 deletions.
diff --git a/...ta1/frontend/src/app/pages/experiment-details/trials-table/trials-table.component.spec.ts b/...ta1/frontend/src/app/pages/experiment-details/trials-table/trials-table.component.spec.ts
@@ -11,6 +11,12 @@ import { MatButtonModule } from '@angular/material/button';
 import { TrialsTableComponent } from './trials-table.component';
 import { BrowserAnimationsModule } from '@angular/platform-browser/animations';
 
+import { SimpleChange } from '@angular/core';
+import { PropertyValue, StatusValue, ComponentValue } from 'kubeflow';
+import { parseStatus } from '../../experiments/utils';
+import lowerCase from 'lodash-es/lowerCase';
+import { KfpRunComponent } from './kfp-run/kfp-run.component';
+
 describe('TrialsTableComponent', () => {
   let component: TrialsTableComponent;
   let fixture: ComponentFixture<TrialsTableComponent>;
@@ -38,10 +44,157 @@ describe('TrialsTableComponent', () => {
   beforeEach(() => {
     fixture = TestBed.createComponent(TrialsTableComponent);
     component = fixture.componentInstance;
+    component.displayedColumns = [
+      'Status',
+      'Trial name',
+      'Validation loss',
+      'Lr',
+      'Batch size',
+      'Embed dim',
+      'Dropout',
+      'Sp dropout',
+      'Kfp run',
+    ];
+    component.data = [
+      [
+        'Succeeded',
+        'open-vaccine-0f37u-6cd03cbf',
+        '0.70573',
+        '0.0001',
+        '32',
+        '20',
+        '0.2',
+        '0.2',
+        '9af7c534-689a-48aa-996b-537d13989729',
+        '0',
+      ],
+      [
+        'Succeeded',
+        'open-vaccine-0f37u-8ec17b8f',
+        '0.76401',
+        '0.0001',
+        '96',
+        '20',
+        '0.2',
+        '0.2',
+        '19aed8e0-143c-49d8-8bd3-7ebb464181d8',
+        '1',
+      ],
+    ];
+    component.ngOnChanges({
+      displayedColumns: new SimpleChange(
+        null,
+        component.displayedColumns,
+        false,
+      ),
+    });
+
     fixture.detectChanges();
   });
 
-  it('should create', () => {
-    expect(component).toBeTruthy();
+  it('should create processedData', () => {
+    expect(component.processedData).toEqual([
+      {
+        'trial name': 'open-vaccine-0f37u-6cd03cbf',
+        status: 'Succeeded',
+        'validation loss': '0.70573',
+        lr: '0.0001',
+        'batch size': '32',
+        'embed dim': '20',
+        dropout: '0.2',
+        'sp dropout': '0.2',
+        'kfp run': '9af7c534-689a-48aa-996b-537d13989729',
+      },
+      {
+        'trial name': 'open-vaccine-0f37u-8ec17b8f',
+        status: 'Succeeded',
+        'validation loss': '0.76401',
+        lr: '0.0001',
+        'batch size': '96',
+        'embed dim': '20',
+        dropout: '0.2',
+        'sp dropout': '0.2',
+        'kfp run': '19aed8e0-143c-49d8-8bd3-7ebb464181d8',
+      },
+    ]);
+  });
+
+  it('should create config', () => {
+    expect(component.config).toEqual({
+      columns: [
+        {
+          matColumnDef: 'Status',
+          matHeaderCellDef: 'Status',
+          value: new StatusValue({
+            valueFn: parseStatus,
+          }),
+          sort: true,
+        },
+        {
+          matColumnDef: 'name',
+          matHeaderCellDef: 'Trial name',
+          value: new PropertyValue({
+            field: lowerCase(component.displayedColumns[1]),
+            isLink: true,
+          }),
+          sort: true,
+        },
+        {
+          matColumnDef: 'Validation loss',
+          matHeaderCellDef: 'Validation loss',
+          value: new PropertyValue({
+            field: lowerCase(component.displayedColumns[2]),
+          }),
+          sort: true,
+        },
+        {
+          matColumnDef: 'Lr',
+          matHeaderCellDef: 'Lr',
+          value: new PropertyValue({
+            field: lowerCase(component.displayedColumns[3]),
+          }),
+          sort: true,
+        },
+        {
+          matColumnDef: 'Batch size',
+          matHeaderCellDef: 'Batch size',
+          value: new PropertyValue({
+            field: lowerCase(component.displayedColumns[4]),
+          }),
+          sort: true,
+        },
+        {
+          matColumnDef: 'Embed dim',
+          matHeaderCellDef: 'Embed dim',
+          value: new PropertyValue({
+            field: lowerCase(component.displayedColumns[5]),
+          }),
+          sort: true,
+        },
+        {
+          matColumnDef: 'Dropout',
+          matHeaderCellDef: 'Dropout',
+          value: new PropertyValue({
+            field: lowerCase(component.displayedColumns[6]),
+          }),
+          sort: true,
+        },
+        {
+          matColumnDef: 'Sp dropout',
+          matHeaderCellDef: 'Sp dropout',
+          value: new PropertyValue({
+            field: lowerCase(component.displayedColumns[7]),
+          }),
+          sort: true,
+        },
+        {
+          matHeaderCellDef: '',
+          matColumnDef: 'actions',
+          value: new ComponentValue({
+            component: KfpRunComponent,
+          }),
+        },
+      ],
+    });
   });
 });