addressed comments

Signed-off-by: rsun19 <robertssun1234@gmail.com>

Author: rsun19

clients/ui/frontend/src/app/hooks/__tests__/useModelTransferJobs.spec.ts

@@ -1,4 +1,4 @@
-import { waitFor } from '@testing-library/react';
+import { act, waitFor } from '@testing-library/react';
 import { useFetchState, POLL_INTERVAL } from 'mod-arch-core';
 import useModelTransferJobs from '~/app/hooks/useModelTransferJobs';
 import { useModelRegistryAPI } from '~/app/hooks/useModelRegistryAPI';
@@ -129,10 +129,9 @@ describe('useModelTransferJobs', () => {
     // toggle hasActiveJobs to true and cause a re-render with refreshRate = POLL_INTERVAL.
     const capturedCallback = getCapturedCallback();
     expect(capturedCallback).toBeDefined();
-    await capturedCallback?.({});
-
-    // Wait for the hook to re-render after state update
-    await renderResult.waitForNextUpdate();
+    await act(async () => {
+      await capturedCallback?.({});
+    });
 
     // The latest call to useFetchState should have refreshRate set to POLL_INTERVAL
     const lastCallOptions = optionsCalls[optionsCalls.length - 1];
@@ -184,7 +183,9 @@ describe('useModelTransferJobs', () => {
     // so refreshRate should remain undefined (no need to wait for another update).
     const capturedCallback = getCapturedCallback();
     expect(capturedCallback).toBeDefined();
-    await capturedCallback?.({});
+    await act(async () => {
+      await capturedCallback?.({});
+    });
 
     const lastCallOptions = optionsCalls[optionsCalls.length - 1];
     expect(lastCallOptions.refreshRate).toBeUndefined();

clients/ui/frontend/src/app/pages/modelRegistry/screens/ModelTransferJobs/__tests__/getStatusLabel.spec.ts

@@ -2,33 +2,19 @@ import { ModelTransferJobStatus } from '~/app/types';
 import { getStatusLabel } from '~/app/pages/modelRegistry/screens/ModelTransferJobs/ModelTransferJobTableRow';
 
 describe('getStatusLabel', () => {
-  it('returns green "Complete" for COMPLETED status', () => {
-    const result = getStatusLabel(ModelTransferJobStatus.COMPLETED);
-    expect(result.label).toBe('Complete');
-    expect(result.color).toBe('green');
-  });
-
-  it('returns blue "Running" for RUNNING status', () => {
-    const result = getStatusLabel(ModelTransferJobStatus.RUNNING);
-    expect(result.label).toBe('Running');
-    expect(result.color).toBe('blue');
-  });
-
-  it('returns purple "Pending" for PENDING status', () => {
-    const result = getStatusLabel(ModelTransferJobStatus.PENDING);
-    expect(result.label).toBe('Pending');
-    expect(result.color).toBe('purple');
-  });
-
-  it('returns red "Failed" for FAILED status', () => {
-    const result = getStatusLabel(ModelTransferJobStatus.FAILED);
-    expect(result.label).toBe('Failed');
-    expect(result.color).toBe('red');
-  });
-
-  it('returns grey "Canceled" for CANCELLED status', () => {
-    const result = getStatusLabel(ModelTransferJobStatus.CANCELLED);
-    expect(result.label).toBe('Canceled');
-    expect(result.color).toBe('grey');
-  });
+  it.each([
+    [ModelTransferJobStatus.COMPLETED, 'Complete', 'green'],
+    [ModelTransferJobStatus.RUNNING, 'Running', 'blue'],
+    [ModelTransferJobStatus.PENDING, 'Pending', 'purple'],
+    [ModelTransferJobStatus.FAILED, 'Failed', 'red'],
+    [ModelTransferJobStatus.CANCELLED, 'Canceled', 'grey'],
+  ])(
+    'returns correct label, color, and icon for %s status',
+    (status, expectedLabel, expectedColor) => {
+      const result = getStatusLabel(status);
+      expect(result.label).toBe(expectedLabel);
+      expect(result.color).toBe(expectedColor);
+      expect(result.icon).toBeDefined();
+    },
+  );
 });

clients/ui/frontend/src/app/pages/settings/__tests__/ModelRegistryTableRowStatus.spec.tsx

@@ -169,7 +169,7 @@ describe('ModelRegistryTableRowStatus', () => {
     );
     expect(screen.getByText('Ready')).toBeVisible();
   });
-  it('renders "Progressing" status', async () => {
+  it('renders "Unavailable" status', async () => {
     const user = userEvent.setup();
 
     render(