From 8ede0934be31311bed213baed527b1253edfc1fa Mon Sep 17 00:00:00 2001
From: Leo Unoki <leo.unoki@trili.tech>
Date: Tue, 24 Oct 2023 10:44:00 +0100
Subject: [PATCH 1/2] Hide preview button for batch

---
 src/components/SendFlow/Tez/FormPage.tsx | 3 ++-
 src/components/SendFlow/utils.tsx        | 6 ++++--
 src/views/batch/BatchPage.tsx            | 2 +-
 3 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/src/components/SendFlow/Tez/FormPage.tsx b/src/components/SendFlow/Tez/FormPage.tsx
index fb7c385d8a..ebfe0203fd 100644
--- a/src/components/SendFlow/Tez/FormPage.tsx
+++ b/src/components/SendFlow/Tez/FormPage.tsx
@@ -43,7 +43,7 @@ const toOperation = (formValues: FormValues): TezTransfer => ({
   recipient: parsePkh(formValues.recipient),
 });
 
-const FormPage: React.FC<FormPageProps<FormValues>> = props => {
+const FormPage: React.FC<FormPageProps<FormValues> & {showPreview? :boolean}> = ({showPreview = true, ...props}) => {
   const openSignPage = useOpenSignPageFormAction({
     SignPage,
     signPageExtraData: undefined,
@@ -126,6 +126,7 @@ const FormPage: React.FC<FormPageProps<FormValues>> = props => {
               isValid={isValid}
               onSingleSubmit={handleSubmit(onSingleSubmit)}
               onAddToBatch={handleSubmit(onBatchSubmit)}
+              showPreview={showPreview}
             />
           </ModalFooter>
         </form>
diff --git a/src/components/SendFlow/utils.tsx b/src/components/SendFlow/utils.tsx
index 7b31034519..7ed426c0f5 100644
--- a/src/components/SendFlow/utils.tsx
+++ b/src/components/SendFlow/utils.tsx
@@ -51,16 +51,18 @@ export const FormSubmitButtons = ({
   isValid,
   onSingleSubmit,
   onAddToBatch,
+  showPreview = true,
 }: {
   isLoading: boolean;
   isValid: boolean;
   onSingleSubmit: () => Promise<void>;
   onAddToBatch: () => Promise<void>;
+  showPreview? :boolean,
 }) => {
   return (
     <>
       <Box width="100%">
-        <Button
+        {showPreview && <Button
           onClick={onSingleSubmit}
           width="100%"
           size="lg"
@@ -70,7 +72,7 @@ export const FormSubmitButtons = ({
           mb="16px"
         >
           Preview
-        </Button>
+        </Button>}
         <Button
           onClick={onAddToBatch}
           width="100%"
diff --git a/src/views/batch/BatchPage.tsx b/src/views/batch/BatchPage.tsx
index 468008ccca..7ed1b4b14c 100644
--- a/src/views/batch/BatchPage.tsx
+++ b/src/views/batch/BatchPage.tsx
@@ -65,7 +65,7 @@ const EmptyBatch = () => {
         </Text>
         <Flex justifyContent="space-around" mt="30px">
           <Box>
-            <Button onClick={() => openWith(<SendTezForm />)}>Start a Batch</Button>
+            <Button onClick={() => openWith(<SendTezForm showPreview={false} />)}>Start a Batch</Button>
             <Button ml="15px" variant="tertiary" onClick={() => openWith(<CSVFileUploadForm />)}>
               Load CSV file
             </Button>

From 73df666fc0339216fcbd933771cdf20ff7b1730f Mon Sep 17 00:00:00 2001
From: Leo Unoki <leo.unoki@trili.tech>
Date: Tue, 24 Oct 2023 10:44:09 +0100
Subject: [PATCH 2/2] Add test for FormSubmitButtons

format
---
 src/components/SendFlow/Tez/FormPage.tsx |  5 ++++-
 src/components/SendFlow/utils.test.tsx   | 27 ++++++++++++++++++++++++
 src/components/SendFlow/utils.tsx        | 26 ++++++++++++-----------
 src/views/batch/BatchPage.tsx            |  4 +++-
 4 files changed, 48 insertions(+), 14 deletions(-)

diff --git a/src/components/SendFlow/Tez/FormPage.tsx b/src/components/SendFlow/Tez/FormPage.tsx
index ebfe0203fd..0be17e8ea6 100644
--- a/src/components/SendFlow/Tez/FormPage.tsx
+++ b/src/components/SendFlow/Tez/FormPage.tsx
@@ -43,7 +43,10 @@ const toOperation = (formValues: FormValues): TezTransfer => ({
   recipient: parsePkh(formValues.recipient),
 });
 
-const FormPage: React.FC<FormPageProps<FormValues> & {showPreview? :boolean}> = ({showPreview = true, ...props}) => {
+const FormPage: React.FC<FormPageProps<FormValues> & { showPreview?: boolean }> = ({
+  showPreview = true,
+  ...props
+}) => {
   const openSignPage = useOpenSignPageFormAction({
     SignPage,
     signPageExtraData: undefined,
diff --git a/src/components/SendFlow/utils.test.tsx b/src/components/SendFlow/utils.test.tsx
index 1996a4d8db..ebe5ad602e 100644
--- a/src/components/SendFlow/utils.test.tsx
+++ b/src/components/SendFlow/utils.test.tsx
@@ -54,6 +54,33 @@ describe("SendFlow utils", () => {
       fireEvent.click(screen.getByText("Insert Into Batch"));
       expect(mockSingle).toHaveBeenCalled();
     });
+
+    it("shows preview button by default", () => {
+      render(
+        <FormSubmitButtons
+          isLoading={false}
+          isValid={true}
+          onSingleSubmit={async () => {}}
+          onAddToBatch={async () => {}}
+        />
+      );
+
+      expect(screen.getByText("Preview")).toBeInTheDocument();
+    });
+
+    it("hides preview button", () => {
+      render(
+        <FormSubmitButtons
+          showPreview={false}
+          isLoading={false}
+          isValid={true}
+          onSingleSubmit={async () => {}}
+          onAddToBatch={async () => {}}
+        />
+      );
+
+      expect(screen.queryByText("Preview")).not.toBeInTheDocument();
+    });
   });
 
   describe("smallestUnit", () => {
diff --git a/src/components/SendFlow/utils.tsx b/src/components/SendFlow/utils.tsx
index 7ed426c0f5..522ca7e771 100644
--- a/src/components/SendFlow/utils.tsx
+++ b/src/components/SendFlow/utils.tsx
@@ -57,22 +57,24 @@ export const FormSubmitButtons = ({
   isValid: boolean;
   onSingleSubmit: () => Promise<void>;
   onAddToBatch: () => Promise<void>;
-  showPreview? :boolean,
+  showPreview?: boolean;
 }) => {
   return (
     <>
       <Box width="100%">
-        {showPreview && <Button
-          onClick={onSingleSubmit}
-          width="100%"
-          size="lg"
-          isLoading={isLoading}
-          type="submit"
-          isDisabled={!isValid}
-          mb="16px"
-        >
-          Preview
-        </Button>}
+        {showPreview && (
+          <Button
+            onClick={onSingleSubmit}
+            width="100%"
+            size="lg"
+            isLoading={isLoading}
+            type="submit"
+            isDisabled={!isValid}
+            mb="16px"
+          >
+            Preview
+          </Button>
+        )}
         <Button
           onClick={onAddToBatch}
           width="100%"
diff --git a/src/views/batch/BatchPage.tsx b/src/views/batch/BatchPage.tsx
index 7ed1b4b14c..78aab349da 100644
--- a/src/views/batch/BatchPage.tsx
+++ b/src/views/batch/BatchPage.tsx
@@ -65,7 +65,9 @@ const EmptyBatch = () => {
         </Text>
         <Flex justifyContent="space-around" mt="30px">
           <Box>
-            <Button onClick={() => openWith(<SendTezForm showPreview={false} />)}>Start a Batch</Button>
+            <Button onClick={() => openWith(<SendTezForm showPreview={false} />)}>
+              Start a Batch
+            </Button>
             <Button ml="15px" variant="tertiary" onClick={() => openWith(<CSVFileUploadForm />)}>
               Load CSV file
             </Button>