split dispatch and find workflow

testdriverai · Feb 27, 2024 · a6aff03 · a6aff03
1 parent 63ca1b3
commit a6aff03
Show file tree

Hide file tree

Showing 2 changed files with 74 additions and 4 deletions.
diff --git a/dist/index.js b/dist/index.js
@@ -38848,8 +38848,10 @@ const waitFor = (ms) => new Promise((r) => setTimeout(r, ms));
 
   console.log("inputs", { repo, branch, prompt });
 
+  console.log('TestDriver: "Dispatching testdriver..."'.green);
+
   const {
-    data: { workflowId },
+    data: { dispatchId },
   } = await axios.post(
     `${baseUrl}/testdriver-dispatch`,
     {
@@ -38863,7 +38865,40 @@ const waitFor = (ms) => new Promise((r) => setTimeout(r, ms));
     }
   );
 
-  console.log("workflow id:", workflowId);
+  console.log("dispatch id:", dispatchId);
+
+  console.log('TestDriver: "Finding the dispatched workflow..."'.green);
+
+  const checkWorkflow = async () => {
+    const {
+      data: { workflowId },
+    } = await axios.post(
+      `${baseUrl}/testdriver-workflow`,
+      {
+        dispatchId,
+      },
+      {
+        Accept: "application/json",
+        "Content-Type": "application/json",
+      }
+    );
+
+    return workflowId;
+  };
+
+  const waitUntilWorkflowAvailable = async () => {
+    let workflowId;
+    while (!workflowId) {
+      await waitFor(1000 * 60);
+      workflowId = await checkWorkflow();
+    }
+
+    return workflowId;
+  };
+
+  const workflowId = await waitUntilWorkflowAvailable();
+
+  console.log("Found workflow id:", workflowId);
 
   const checkStatus = async () => {
     const {

diff --git a/src/index.js b/src/index.js
@@ -36,8 +36,10 @@ const waitFor = (ms) => new Promise((r) => setTimeout(r, ms));
 
   console.log("inputs", { repo, branch, prompt });
 
+  console.log('TestDriver: "Dispatching testdriver..."'.green);
+
   const {
-    data: { workflowId },
+    data: { dispatchId },
   } = await axios.post(
     `${baseUrl}/testdriver-dispatch`,
     {
@@ -51,7 +53,40 @@ const waitFor = (ms) => new Promise((r) => setTimeout(r, ms));
     }
   );
 
-  console.log("workflow id:", workflowId);
+  console.log("dispatch id:", dispatchId);
+
+  console.log('TestDriver: "Finding the dispatched workflow..."'.green);
+
+  const checkWorkflow = async () => {
+    const {
+      data: { workflowId },
+    } = await axios.post(
+      `${baseUrl}/testdriver-workflow`,
+      {
+        dispatchId,
+      },
+      {
+        Accept: "application/json",
+        "Content-Type": "application/json",
+      }
+    );
+
+    return workflowId;
+  };
+
+  const waitUntilWorkflowAvailable = async () => {
+    let workflowId;
+    while (!workflowId) {
+      await waitFor(1000 * 60);
+      workflowId = await checkWorkflow();
+    }
+
+    return workflowId;
+  };
+
+  const workflowId = await waitUntilWorkflowAvailable();
+
+  console.log("Found workflow id:", workflowId);
 
   const checkStatus = async () => {
     const {