From 098104d1bfd787526d58595999647391943bb672 Mon Sep 17 00:00:00 2001
From: jasquat <jasquat@users.noreply.github.com>
Date: Tue, 27 Sep 2022 14:39:42 -0400
Subject: [PATCH] allow viewing and editing script unit tests from the script
 editor w/ burnettk

---
 src/routes/ProcessInstanceShow.tsx     |   5 +-
 src/routes/ProcessModelEditDiagram.tsx | 175 +++++++++++++++++++++++--
 2 files changed, 169 insertions(+), 11 deletions(-)

diff --git a/src/routes/ProcessInstanceShow.tsx b/src/routes/ProcessInstanceShow.tsx
index 5548037..ff2791c 100644
--- a/src/routes/ProcessInstanceShow.tsx
+++ b/src/routes/ProcessInstanceShow.tsx
@@ -142,15 +142,16 @@ export default function ProcessInstanceShow() {
   };
 
   const taskDataDisplayArea = () => {
+    const taskToUse: any = taskToDisplay;
     if (taskToDisplay) {
       return (
         <Modal show={taskToDisplay} onHide={handleTaskDataDisplayClose}>
           <Modal.Header closeButton>
             <Modal.Title>
-              {(taskToDisplay as any).name}: {(taskToDisplay as any).state}
+              {taskToUse.name} ({taskToUse.type}): {taskToUse.state}
             </Modal.Title>
           </Modal.Header>
-          <pre>{JSON.stringify((taskToDisplay as any).data, null, 2)}</pre>
+          <pre>{JSON.stringify(taskToUse.data, null, 2)}</pre>
         </Modal>
       );
     }
diff --git a/src/routes/ProcessModelEditDiagram.tsx b/src/routes/ProcessModelEditDiagram.tsx
index f328e87..ead2cc4 100644
--- a/src/routes/ProcessModelEditDiagram.tsx
+++ b/src/routes/ProcessModelEditDiagram.tsx
@@ -1,6 +1,6 @@
 import { useContext, useEffect, useState } from 'react';
 import { useNavigate, useParams, useSearchParams } from 'react-router-dom';
-import { Button, Modal } from 'react-bootstrap';
+import { Button, Modal, Stack } from 'react-bootstrap';
 import Editor from '@monaco-editor/react';
 
 import ReactDiagramEditor from '../components/ReactDiagramEditor';
@@ -18,6 +18,17 @@ export default function ProcessModelEditDiagram() {
   const [showScriptEditor, setShowScriptEditor] = useState(false);
   const handleShowScriptEditor = () => setShowScriptEditor(true);
 
+  interface ScriptUnitTest {
+    id: string;
+    inputJson: any;
+    expectedOutputJson: any;
+  }
+
+  const [currentScriptUnitTest, setCurrentScriptUnitTest] =
+    useState<ScriptUnitTest | null>(null);
+  const [currentScriptUnitTestIndex, setCurrentScriptUnitTestIndex] =
+    useState<number>(-1);
+
   const params = useParams();
   const navigate = useNavigate();
   const [searchParams] = useSearchParams();
@@ -162,22 +173,164 @@ export default function ProcessModelEditDiagram() {
       successCallback: makeApiHandler(event),
     });
   };
+
+  const getScriptUnitTestElements = (element: any) => {
+    const { extensionElements } = element.businessObject;
+    if (extensionElements && extensionElements.values.length > 0) {
+      const unitTestModdleElements = extensionElements
+        .get('values')
+        .filter(function getInstanceOfType(e: any) {
+          return e.$instanceOf('spiffworkflow:unitTests');
+        })[0];
+      if (unitTestModdleElements) {
+        return unitTestModdleElements.unitTests;
+      }
+    }
+    return [];
+  };
+  const setScriptUnitTestElementWithIndex = (
+    scriptIndex: number,
+    element: any = scriptElement
+  ) => {
+    const unitTestsModdleElements = getScriptUnitTestElements(element);
+    if (unitTestsModdleElements.length > 0) {
+      setCurrentScriptUnitTest(unitTestsModdleElements[scriptIndex]);
+      setCurrentScriptUnitTestIndex(scriptIndex);
+    }
+  };
   const onLaunchScriptEditor = (element: any, modeling: any) => {
     setScriptText(element.businessObject.script || '');
     setScriptModeling(modeling);
     setScriptElement(element);
+    setScriptUnitTestElementWithIndex(0, element);
     handleShowScriptEditor();
   };
   const handleScriptEditorClose = () => {
     setShowScriptEditor(false);
   };
-  const handleEditorChange = (value: any) => {
+  const handleEditorScriptChange = (value: any) => {
     setScriptText(value);
     (scriptModeling as any).updateProperties(scriptElement, {
       scriptFormat: 'python',
       script: value,
     });
   };
+  const handleEditorScriptTestUnitInputChange = (value: any) => {
+    if (currentScriptUnitTest) {
+      currentScriptUnitTest.inputJson.value = value;
+      (scriptModeling as any).updateProperties(scriptElement, {});
+    }
+  };
+  const handleEditorScriptTestUnitOutputChange = (value: any) => {
+    if (currentScriptUnitTest) {
+      currentScriptUnitTest.expectedOutputJson.value = value;
+      (scriptModeling as any).updateProperties(scriptElement, {});
+    }
+  };
+
+  const generalEditorOptions = () => {
+    return {
+      glyphMargin: false,
+      folding: false,
+      lineNumbersMinChars: 0,
+    };
+  };
+  const setPreviousScriptUnitTest = () => {
+    const newScriptIndex = currentScriptUnitTestIndex - 1;
+    if (newScriptIndex >= 0) {
+      setScriptUnitTestElementWithIndex(newScriptIndex);
+    }
+  };
+  const setNextScriptUnitTest = () => {
+    const newScriptIndex = currentScriptUnitTestIndex + 1;
+    const unitTestsModdleElements = getScriptUnitTestElements(scriptElement);
+    if (newScriptIndex < unitTestsModdleElements.length) {
+      setScriptUnitTestElementWithIndex(newScriptIndex);
+    }
+  };
+  const runCurrentUnitTest = () => {};
+  const scriptUnitTestEditorElement = () => {
+    if (currentScriptUnitTest) {
+      let previousButtonDisable = true;
+      if (currentScriptUnitTestIndex > 0) {
+        previousButtonDisable = false;
+      }
+      let nextButtonDisable = true;
+      const unitTestsModdleElements = getScriptUnitTestElements(scriptElement);
+      if (currentScriptUnitTestIndex < unitTestsModdleElements.length - 1) {
+        nextButtonDisable = false;
+      }
+
+      // unset current unit test if all tests were deleted
+      if (unitTestsModdleElements.length < 1) {
+        setCurrentScriptUnitTest(null);
+        setCurrentScriptUnitTestIndex(-1);
+      }
+      return (
+        <main>
+          <Stack direction="horizontal" gap={3}>
+            <Button
+              data-qa="unit-test-previous-button"
+              style={{ fontSize: '1.5em' }}
+              onClick={setPreviousScriptUnitTest}
+              variant="link"
+              disabled={previousButtonDisable}
+            >
+              &laquo;
+            </Button>
+            <h3>Unit Test: {currentScriptUnitTest.id}</h3>
+            <Button
+              data-qa="unit-test-run"
+              style={{ fontSize: '1.5em' }}
+              onClick={runCurrentUnitTest}
+            >
+              Run
+            </Button>
+            <Button
+              data-qa="unit-test-next-button"
+              style={{ fontSize: '1.5em' }}
+              onClick={setNextScriptUnitTest}
+              variant="link"
+              disabled={nextButtonDisable}
+            >
+              &raquo;
+            </Button>
+          </Stack>
+          <Stack direction="horizontal" gap={3}>
+            <Stack>
+              <div>Input Json:</div>
+              <div>
+                <Editor
+                  height={200}
+                  defaultLanguage="json"
+                  options={Object.assign(generalEditorOptions(), {
+                    minimap: { enabled: false },
+                  })}
+                  value={currentScriptUnitTest.inputJson.value}
+                  onChange={handleEditorScriptTestUnitInputChange}
+                />
+              </div>
+            </Stack>
+            <Stack>
+              <div>Expected Output Json:</div>
+              <div>
+                <Editor
+                  height={200}
+                  defaultLanguage="json"
+                  options={Object.assign(generalEditorOptions(), {
+                    minimap: { enabled: false },
+                  })}
+                  value={currentScriptUnitTest.expectedOutputJson.value}
+                  onChange={handleEditorScriptTestUnitOutputChange}
+                />
+              </div>
+            </Stack>
+          </Stack>
+        </main>
+      );
+    }
+    return null;
+  };
   const scriptEditor = () => {
     let scriptName = '';
     if (scriptElement) {
@@ -188,13 +341,17 @@ export default function ProcessModelEditDiagram() {
         <Modal.Header closeButton>
           <Modal.Title>Editing Script: {scriptName}</Modal.Title>
         </Modal.Header>
-        <Editor
-          height={600}
-          width="auto"
-          defaultLanguage="python"
-          defaultValue={scriptText}
-          onChange={handleEditorChange}
-        />
+        <Modal.Body>
+          <Editor
+            height={500}
+            width="auto"
+            options={generalEditorOptions()}
+            defaultLanguage="python"
+            defaultValue={scriptText}
+            onChange={handleEditorScriptChange}
+          />
+          {scriptUnitTestEditorElement()}
+        </Modal.Body>
         <Modal.Footer>
           <Button variant="secondary" onClick={handleScriptEditorClose}>
             Close