🐛 Save as Draft not working

client/src/app/pages/assessment/components/assessment-wizard/assessment-wizard.tsx

@@ -1,5 +1,5 @@
 import * as yup from "yup";
-import React, { useEffect, useMemo, useState } from "react";
+import React, { useMemo, useState } from "react";
 import { useTranslation } from "react-i18next";
 import { useHistory } from "react-router-dom";
 import { FieldErrors, FormProvider, useForm } from "react-hook-form";
@@ -131,18 +131,6 @@ export const AssessmentWizard: React.FC<AssessmentWizardProps> = ({
     return questions;
   }, [assessment]);
 
-  useEffect(() => {
-    console.log("asssessment.stakeholders", assessment?.stakeholders);
-    methods.reset({
-      stakeholders: assessment?.stakeholders?.map((sh) => sh.name).sort() ?? [],
-      stakeholderGroups:
-        assessment?.stakeholderGroups?.map((sg) => sg.name).sort() ?? [],
-      // comments: initialComments,
-      questions: initialQuestions,
-      [SAVE_ACTION_KEY]: SAVE_ACTION_VALUE.SAVE_AS_DRAFT,
-    });
-  }, [initialQuestions, assessment]);
-
   const validationSchema = yup.object().shape({
     stakeholders: yup.array().of(yup.string()),
     stakeholderGroups: yup.array().of(yup.string()),
@@ -169,7 +157,6 @@ export const AssessmentWizard: React.FC<AssessmentWizardProps> = ({
   const isValid = methods.formState.isValid;
   const isSubmitting = methods.formState.isSubmitting;
   const isValidating = methods.formState.isValidating;
-
   const watchAllFields = methods.watch();
 
   const disableNavigation = !isValid || isSubmitting;
@@ -204,17 +191,32 @@ export const AssessmentWizard: React.FC<AssessmentWizardProps> = ({
   //   return value !== null && value !== undefined && value.length > 0;
   // };
 
-  const shouldNextBtnBeEnabled = (section: Section): boolean => {
-    const allQuestionsValid = section?.questions.every((question) =>
-      isQuestionValid(question)
+  const areAllQuestionsAnswered = (section: Section): boolean => {
+    return (
+      section?.questions.every((question) => {
+        return questionHasValue(question);
+      }) ?? false
     );
-    const allQuestionsAnswered = section?.questions.every((question) => {
+  };
+
+  const hasPartialAnswers = (section: Section): boolean => {
+    const someQuestionsAnswered = section?.questions.some((question) => {
       return questionHasValue(question);
     });
-    return (
-      allQuestionsAnswered && allQuestionsValid
-      //  && isCommentValid(category)
+
+    const allQuestionsAnswered = areAllQuestionsAnswered(section);
+
+    return someQuestionsAnswered && !allQuestionsAnswered;
+  };
+
+  const shouldNextBtnBeEnabled = (section: Section): boolean => {
+    const allQuestionsValid = section?.questions.every((question) =>
+      isQuestionValid(question)
     );
+
+    const allQuestionsAnswered = areAllQuestionsAnswered(section);
+
+    return allQuestionsAnswered && allQuestionsValid;
   };
 
   const maxCategoryWithData = [...sortedSections].reverse().find((section) => {
@@ -541,6 +543,7 @@ export const AssessmentWizard: React.FC<AssessmentWizardProps> = ({
         (currentStep === sortedSections.length &&
           !shouldNextBtnBeEnabled(sortedSections[currentStep - 1]))
       }
+      hasAnswers={hasPartialAnswers(sortedSections[currentStep - 1])}
       isFormInvalid={!isValid}
       onSave={(review) => {
         const saveActionValue = review

client/src/app/pages/assessment/components/custom-wizard-footer/custom-wizard-footer.tsx

@@ -13,6 +13,7 @@ export interface CustomWizardFooterProps {
   isDisabled: boolean;
   isFormInvalid: boolean;
   isArchetype?: boolean;
+  hasAnswers?: boolean;
   onSave: (review: boolean) => void;
   onSaveAsDraft: () => void;
 }
@@ -23,6 +24,7 @@ export const CustomWizardFooter: React.FC<CustomWizardFooterProps> = ({
   isDisabled,
   isFormInvalid,
   isArchetype,
+  hasAnswers,
   onSave,
   onSaveAsDraft,
 }) => {
@@ -84,7 +86,7 @@ export const CustomWizardFooter: React.FC<CustomWizardFooterProps> = ({
               <Button
                 variant="link"
                 onClick={onSaveAsDraft}
-                isDisabled={isFormInvalid}
+                isDisabled={isFormInvalid || isFirstStep || !hasAnswers}
                 cy-data="save-as-draft"
               >
                 {t("actions.saveAsDraft")}

client/src/app/pages/assessment/components/custom-wizard-footer/tests/custom-wizard-footer.test.tsx

@@ -111,6 +111,7 @@ describe("AppPlaceholder", () => {
         isLastStep={false}
         isDisabled={false}
         isFormInvalid={false}
+        hasAnswers={true}
         onSave={jest.fn()}
         onSaveAsDraft={onSaveAsDraftSpy}
       />

client/src/app/queries/assessments.ts

@@ -107,15 +107,15 @@
 };
 
 export const useFetchAssessmentById = (id?: number | string) => {
-  const { data, isLoading, error } = useQuery({
+  const { data, isLoading, error, isFetching } = useQuery({
     queryKey: [assessmentQueryKey, id],
     queryFn: () => (id ? getAssessmentById(id) : undefined),
     onError: (error: AxiosError) => console.log("error, ", error),
     enabled: !!id,
   });
   return {
     assessment: data,
-    isFetching: isLoading,
+    isFetching: isLoading || isFetching,
     fetchError: error,
   };
 };