feat: add test for handle save screen

ComPlat · Apr 1, 2024 · d2ffacd · d2ffacd
1 parent 299ee83
commit d2ffacd
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 2 deletions.
diff --git a/app/packs/src/utilities/ElementUtils.js b/app/packs/src/utilities/ElementUtils.js
@@ -480,7 +480,7 @@ const handleSaveDataset = (element, uiStoreState, callbackFunc, ...args) => {
   const { elementID, isSaving, elementType } = containerDataSet;
   if (!isSaving) return false;
 
-  const supporedTypes = ['sample', 'reaction', 'researchPlan', 'wellplate']
+  const supporedTypes = ['sample', 'reaction', 'researchPlan', 'wellplate', 'screen'];
   if (supporedTypes.includes(elementType)) {
     if (Array.isArray(element)) {
       if (elementType === 'sample') {

diff --git a/spec/javascripts/utils/ElementUtils.spec.js b/spec/javascripts/utils/ElementUtils.spec.js
@@ -319,7 +319,7 @@ describe('Handle container dataset saving', () => {
   });
 
   describe('save dataset for wellplate', () => {
-    it('when it is not the same wellplate plan id', () => {
+    it('when it is not the same wellplate id', () => {
       const wellplate = { id: 100 };
       const containerDataSet = { elementType: 'wellplate', isSaving: true, elementID: 101 };
       const uiState = { containerDataSet };
@@ -335,4 +335,22 @@ describe('Handle container dataset saving', () => {
       expect(triggered).toEqual(true);
     });
   });
+
+  describe('save dataset for screen', () => {
+    it('when it is not the same screen id', () => {
+      const screen = { id: 100 };
+      const containerDataSet = { elementType: 'screen', isSaving: true, elementID: 101 };
+      const uiState = { containerDataSet };
+      const triggered = handleSaveDataset(screen, uiState, handleSubmit);
+      expect(triggered).toEqual(false);
+    });
+
+    it('when it is the same screen id', () => {
+      const screen = { id: 100 };
+      const containerDataSet = { elementType: 'screen', isSaving: true, elementID: 100 };
+      const uiState = { containerDataSet };
+      const triggered = handleSaveDataset(screen, uiState, handleSubmit, true);
+      expect(triggered).toEqual(true);
+    });
+  });
 });