hacking

src/autoupdater.js

@@ -1,5 +1,55 @@
 const core = require("@actions/core");
 
+async function getStatusChecks(pr, repo, octokit) {
+  core.info(`Getting require status checks for ${pr.head.ref}`);
+
+  const response = await octokit.repos.getStatusChecksProtection({
+    ...repo,
+    branch: pr.head.ref,
+  });
+  const statusChecks = response.data;
+
+  core.info(
+    `Required Status checks for ${pr.head.ref}: ${statusChecks.contexts.join(
+      ", "
+    )}`
+  );
+
+  return statusChecks;
+}
+
+async function waitForStatusChecks(pr, repo, octokit) {
+  const requiredChecks = getStatusChecks(pr, repo, octokit);
+
+  let completed;
+  let requiredCheckRuns;
+  let count = 0;
+
+  do {
+    if (count > 0) {
+      core.info("Waiting for checks to pass...");
+      await new Promise((resolve) => setTimeout(resolve, 5000));
+    }
+
+    const response = await octokit.checks.listForRef({
+      ...repo,
+      ref: pr.head.ref,
+    });
+
+    const checks = response.data;
+
+    requiredCheckRuns = checks.check_runs.filter((run) =>
+      requiredChecks.contexts.includes(run.name)
+    );
+
+    completed = requiredCheckRuns.every(
+      (run) => run.status == "completed" && run.conclusion == "success"
+    );
+
+    count++;
+  } while (!completed);
+}
+
 async function performUpdate({ pullRequest, repo, octokit }) {
   core.startGroup(`Updating PR: Updating ${pullRequest.head.ref}`);
 
@@ -9,14 +59,20 @@ async function performUpdate({ pullRequest, repo, octokit }) {
     context: `autoupdate from ${pullRequest.base.ref}`,
   };
 
-  core.debug("Sending pending status");
-
   await octokit.repos.createCommitStatus({
     ...commitStatusParams,
     state: "pending",
     description: "Updating",
   });
 
+  const timeout = setTimeout(() => {
+    throw Error("Timed out waiting for status checks.");
+  }, 30000);
+
+  await waitForStatusChecks(pullRequest, repo, octokit);
+
+  clearTimeout(timeout);
+
   let error;
 
   try {

test/autoupdater.test.js

@@ -15,8 +15,27 @@ async function expectError(callback, expected) {
 }
 
 function createOctokitMock({ prs = [], repos, pulls }) {
+  const checksMock = {
+    listForRef: jest.fn().mockReturnValue(
+      new Promise((resolve) =>
+        resolve({
+          data: {
+            check_runs: [],
+          },
+        })
+      )
+    ),
+  };
+
   const reposMock = {
     createCommitStatus: jest.fn(),
+    getStatusChecksProtection: jest.fn().mockReturnValue(
+      new Promise((resolve) =>
+        resolve({
+          data: { contexts: [] },
+        })
+      )
+    ),
     ...repos,
   };
 
@@ -31,6 +50,7 @@ function createOctokitMock({ prs = [], repos, pulls }) {
   return {
     repos: reposMock,
     pulls: pullsMock,
+    checks: checksMock,
   };
 }