From 5900a4d2d2c65d931e2cfc2d990c13ee403fac21 Mon Sep 17 00:00:00 2001
From: Renu <renu@Renus-MacBook-Pro.local>
Date: Fri, 24 Mar 2023 22:29:36 +0530
Subject: [PATCH] Fix issue 113

---
 admin-ui/pages/elections/[id]/open-test.tsx | 231 +++++++++++++-------
 lib/Election.js                             |  10 +
 2 files changed, 165 insertions(+), 76 deletions(-)

diff --git a/admin-ui/pages/elections/[id]/open-test.tsx b/admin-ui/pages/elections/[id]/open-test.tsx
index 0488e7e..2da12c1 100644
--- a/admin-ui/pages/elections/[id]/open-test.tsx
+++ b/admin-ui/pages/elections/[id]/open-test.tsx
@@ -1,4 +1,4 @@
-import { Button, Grid, Slider, SliderThumb, Typography } from "@mui/material";
+import { Alert, Button, Grid, Slider, SliderThumb, Typography } from "@mui/material";
 import LoggedInLayout from "layout/LoggedInLayout";
 import type { NextPage } from "next";
 import { useRouter } from "next/router";
@@ -15,8 +15,9 @@ import {
 import GC from "component/GC";
 import GI from "component/GI";
 import Loading from "component/Loading";
-import ElectionCard from "component/ElectionCard";
 import { useNavigate } from "react-router-dom";
+import useCurrentElection from "hooks/useCurrentElection";
+import LoadingButton from "component/LoadingButton";
 
 interface ThumbProps {
   children: ReactNode;
@@ -40,13 +41,18 @@ const TestElection: NextPage = () => {
   const { id } = query;
 
   const electionId = Array.isArray(id) ? id[0] : id;
-
+  const [currentElection, reloadCurrentElection, loadingCurrentElection] = useCurrentElection();
   const [election, setElection] = useState<Maybe<Election>>(null);
+  
+  const shouldShowSetCurrentBtn = !loadingCurrentElection && currentElection && currentElection?.electionId != electionId;
+  const [alertText, setAlertText] = useState<string>("");
+  const [maliciousRequest, setMaliciousRequest] = useState<boolean>(false);
 
   const loadElection = async () => {
     if (electionId) {
       const resp = await getElection(electionId);
       setElection(resp);
+      
     }
   };
 
@@ -56,13 +62,38 @@ const TestElection: NextPage = () => {
     }
   }, [electionId]);
 
+  useEffect(() => {
+    const modesNotAllowedForTest = [ElectionStatus.archived, ElectionStatus.closed, ElectionStatus.open];
+    if (election && modesNotAllowedForTest.includes(election.electionStatus)) {
+      // setAlertText(`${resp?.electionName} is ${resp.electionStatus} and so cannot be set to test mode.`)
+      setMaliciousRequest(true);
+    }
+  }, [election]);
+
   const testElection = async () => {
     if (electionId) {
-      await setCurrentElection(electionId);
-      await openElectionTest(electionId);
-      router.push("/dashboard");
+      try {
+        await openElectionTest(electionId);
+        router.push("/dashboard");
+      } catch (e: any) {
+        console.error(e);
+        setAlertText(e?.data?.error_description);
+      }
+      // await setCurrentElection(electionId);
+    }
+    // loadElection();
+  };
+
+  const runSetCurrentElection = async () => {
+    if (electionId) {
+      try {
+        await setCurrentElection(electionId);
+        reloadCurrentElection();
+      } catch (e: any) {
+        console.error(e);
+        setAlertText(`${currentElection?.electionJurisdictionName} ${currentElection?.electionName} ${e?.data?.error_description}`);
+      }
     }
-    loadElection();
   };
 
   const navigate = useNavigate();
@@ -71,83 +102,131 @@ const TestElection: NextPage = () => {
     navigate(-1);
   };
 
-  return (
-    <LoggedInLayout title="Test Election">
-      {!election && <Loading />}
-      {election &&
-        election.latMode ==
-          1 /*election?.electionStatus === ElectionStatus.test */ && (
-          <GC direction="column" spacing={2}>
+  const InTestMode = () => {
+    return (
+      <GC direction="column" spacing={2}>
+        <GI>
+          <Typography variant="h2">You are now in Testing Mode!</Typography>
+        </GI>
+        <Typography sx={{ fontSize: "3em", margin: "2em 0" }}>
+          You are now in testing mode! Please test your election with your
+          team, then come back to finish editing and launch your election!
+        </Typography>
+        <GI>
+          <GC justifyContent="space-between">
             <GI>
-              <Typography variant="h2">You are now in Testing Mode!</Typography>
+              <Button
+                onClick={() =>
+                  router.push(`/elections/${election.electionId}/test`)
+                }
+              >
+                View Test Status
+              </Button>
             </GI>
-            <Typography sx={{ fontSize: "3em", margin: "2em 0" }}>
-              You are now in testing mode! Please test your election with your
-              team, then come back to finish editing and launch your election!
-            </Typography>
             <GI>
-              <GC justifyContent="space-between">
-                <GI>
-                  <Button
-                    onClick={() =>
-                      router.push(`/elections/${election.electionId}/test`)
-                    }
-                  >
-                    View Test Status
-                  </Button>
-                </GI>
-                <GI>
-                  <Button onClick={() => router.push("/dashboard")}>
-                    Go to Dashboard
-                  </Button>
-                </GI>
-              </GC>
+              <Button onClick={() => router.push("/dashboard")}>
+                Go to Dashboard
+              </Button>
             </GI>
           </GC>
-        )}
-      {election && !election?.latMode && (
-        // This should be latMode 0; but some elections don't have that?
-        //==   0 /*election?.electionStatus !== ElectionStatus.test */ && (
-        <>
-          <Typography variant="h2">Please confirm to continue.</Typography>
-          <Typography sx={{ fontSize: "3em", margin: "2em 0" }}>
-            Please confirm that you would like to enter testing mode for{" "}
-            {election?.electionName}.
-          </Typography>
-          <Grid container spacing={4}>
-            <Grid item xs={3}>
+        </GI>
+      </GC>
+    )
+  }
+
+  const NeedCurrentMode = () => {
+    return (
+      <GC direction="column" spacing={2}>
+        <GI>
+          <Typography variant="h2">Prerequisites not met for test mode!</Typography>
+        </GI>
+        <Typography sx={{ fontSize: "3em", margin: "2em 0" }}>
+          Election for {election?.electionJurisdictionName} {election?.electionName} is not the current election and so cannot be set to test mode.
+        </Typography>
+        <GI>
+          <GC justifyContent="space-between">
+            <GI item xs={3}>
               <Button onClick={goBack}>Back</Button>
-            </Grid>
-            <Grid item xs={2}>
-              &nbsp;
-            </Grid>
-            <Grid item xs={6}>
-              <Slider
-                onChangeCommitted={(_event, newValue) => {
-                  if (newValue === 100) {
-                    testElection();
-                  }
-                }}
-                components={{
-                  Thumb: ThumbComponent,
-                }}
-                step={null}
-                marks={[
-                  {
-                    value: 0,
-                    label: "",
-                  },
-                  {
-                    value: 100,
-                    label: "",
-                  },
-                ]}
-                defaultValue={0}
-              />
-            </Grid>
+            </GI>
+            {!alertText ?
+            <GI item xs={3}>
+              <LoadingButton onClick={runSetCurrentElection}>
+                Set Current
+              </LoadingButton>
+            </GI>
+            : false
+            }
+          </GC>
+        </GI>
+      </GC>
+    )
+  }
+
+  const SetTestMode = () => {
+    return (
+      <>
+        <Typography variant="h2">Please confirm to continue.</Typography>
+        <Typography sx={{ fontSize: "3em", margin: "2em 0" }}>
+          Please confirm that you would like to enter testing mode for{" "}
+          {election?.electionName}.
+        </Typography>
+        <Grid container spacing={4}>
+          <Grid item xs={3}>
+            <Button onClick={goBack}>Back</Button>
+          </Grid>
+          <Grid item xs={2}>
+            &nbsp;
           </Grid>
+          <Grid item xs={6}>
+            <Slider
+              onChangeCommitted={(_event, newValue) => {
+                if (newValue === 100) {
+                  testElection();
+                }
+              }}
+              components={{
+                Thumb: ThumbComponent,
+              }}
+              step={null}
+              marks={[
+                {
+                  value: 0,
+                  label: "",
+                },
+                {
+                  value: 100,
+                  label: "",
+                },
+              ]}
+              defaultValue={0}
+            />
+          </Grid>
+        </Grid>
+      </>
+    )
+  }
+
+  return (
+    <LoggedInLayout title="Test Election">
+      {(!election || loadingCurrentElection) && <Loading />}
+      {maliciousRequest ? 
+        <Alert severity="error">{election?.electionName} is {election?.electionStatus} and so cannot be set to test mode.</Alert>
+      : 
+        <>
+        {!loadingCurrentElection && election &&
+          election.latMode == 1 /*election?.electionStatus === ElectionStatus.test */ && 
+          <InTestMode />
+        }
+        {!loadingCurrentElection && election && !election?.latMode && (
+          // This should be latMode 0; but some elections don't have that?
+          //==   0 /*election?.electionStatus !== ElectionStatus.test */ && (
+          shouldShowSetCurrentBtn ? <NeedCurrentMode /> : <SetTestMode />
+        )}
+        <Grid xs={12}>
+          {alertText && <Alert severity="error">{alertText}</Alert>}
+        </Grid>
         </>
-      )}
+      }
     </LoggedInLayout>
   );
 };
diff --git a/lib/Election.js b/lib/Election.js
index 5edbb2c..42da46c 100644
--- a/lib/Election.js
+++ b/lib/Election.js
@@ -1433,6 +1433,16 @@ class Election {
       case apiEndpoint.setCurrentElection:
         return Election.testConditions(endpoint, election, [
           [liveCurrentElection, "Current election is in a live state"],
+          [
+            electionAttributes?.["electionStatus"] ==
+              Election.electionStatus.archived,
+            "Election has been archived.",
+          ],
+          [
+            electionAttributes?.["electionStatus"] ==
+              Election.electionStatus.closed,
+            "Election has been closed",
+          ]
         ]);
 
       case apiEndpoint.openElectionTest: