test: pause/unpause test cases

determined-ai · Jul 8, 2024 · df69351 · df69351
1 parent 05957fc
commit df69351
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/webui/react/src/utils/flatRun.test.ts b/webui/react/src/utils/flatRun.test.ts
@@ -3,6 +3,7 @@ import dayjs from 'dayjs';
 import {
   deletableRunStates,
   killableRunStates,
+  pausableRunStates,
   runStateList,
   terminalRunStates,
 } from 'constants/states';
@@ -214,5 +215,29 @@ describe('Flat Run Utilities', () => {
         expect(canActionFlatRun(FlatRunAction.Move, bothArchivedFlatRun)).toBeFalsy();
       });
     });
+
+    describe('Pause Action', () => {
+      it.each(Object.values(RunState).filter((v) => pausableRunStates.has(v)))(
+        'should be pausable (%s)',
+        (runState) => {
+          const flatRun: FlatRun = {
+            ...BASE_FLAT_RUN,
+            state: runState,
+          };
+          expect(canActionFlatRun(FlatRunAction.Pause, flatRun)).toBeTruthy();
+        },
+      );
+
+      it.each(Object.values(RunState).filter((v) => !pausableRunStates.has(v)))(
+        'should not be pausable (%s)',
+        (runState) => {
+          const flatRun: FlatRun = {
+            ...BASE_FLAT_RUN,
+            state: runState,
+          };
+          expect(canActionFlatRun(FlatRunAction.Pause, flatRun)).toBeFalsy();
+        },
+      );
+    });
   });
 });