Updated validation for message

glific · Jan 20, 2021 · 8a6d361 · 8a6d361
1 parent f6e8bbc
commit 8a6d361
Showing 1 changed file with 25 additions and 22 deletions.
diff --git a/src/containers/Template/Form/Template.tsx b/src/containers/Template/Form/Template.tsx
@@ -20,28 +20,6 @@ import { CREATE_MEDIA_MESSAGE } from '../../../graphql/mutations/Chat';
 import { Checkbox } from '../../../components/UI/Form/Checkbox/Checkbox';
 import { USER_LANGUAGES } from '../../../graphql/queries/Organization';
 
-const validation = {
-  language: Yup.object().nullable().required('Language is required.'),
-  label: Yup.string().required('Title is required.').max(50, 'Title length is too long.'),
-  body: Yup.string()
-    .transform((current, original) => {
-      return original.getCurrentContent().getPlainText();
-    })
-    .required('Message is required.'),
-  type: Yup.object()
-    .nullable()
-    .when('attachmentURL', {
-      is: (val) => val && val !== '',
-      then: Yup.object().required('Type is required.'),
-    }),
-  attachmentURL: Yup.string()
-    .nullable()
-    .when('type', {
-      is: (val) => val && val.id,
-      then: Yup.string().required('Attachment URL is required.'),
-    }),
-};
-
 const HSMValidation = {
   example: Yup.string()
     .transform((current, original) => {
@@ -436,6 +414,31 @@ const Template: React.SFC<TemplateProps> = (props) => {
     return data;
   };
 
+  const validation = {
+    language: Yup.object().nullable().required('Language is required.'),
+    label: Yup.string().required('Title is required.').max(50, 'Title length is too long.'),
+    body: Yup.string()
+      .transform((current, original) => {
+        return original.getCurrentContent().getPlainText();
+      })
+      .when('type', {
+        is: (val) => (!defaultAttribute.isHsm && !val) || defaultAttribute.isHsm,
+        then: Yup.string().required('Message is required.'),
+      }),
+    type: Yup.object()
+      .nullable()
+      .when('attachmentURL', {
+        is: (val) => val && val !== '',
+        then: Yup.object().required('Type is required.'),
+      }),
+    attachmentURL: Yup.string()
+      .nullable()
+      .when('type', {
+        is: (val) => val && val.id,
+        then: Yup.string().required('Attachment URL is required.'),
+      }),
+  };
+
   const validationObj = defaultAttribute.isHsm ? { ...validation, ...HSMValidation } : validation;
   const FormSchema = Yup.object().shape(validationObj, [['type', 'attachmentURL']]);