From f29736c806531fa7b4d5968f570fa2536f710e59 Mon Sep 17 00:00:00 2001
From: Rob Anderson <robandpdx@github.com>
Date: Mon, 18 Dec 2023 19:55:07 -0500
Subject: [PATCH] support all merge methods

---
 app.js  | 10 +++++++-
 test.js | 71 ++++++++++++++++++++++++++++++++++++++++++++++++++++++---
 2 files changed, 77 insertions(+), 4 deletions(-)

diff --git a/app.js b/app.js
index 947f750..3b5d59c 100644
--- a/app.js
+++ b/app.js
@@ -69,11 +69,19 @@ module.exports = (app) => {
       // Merge PR, if configured to do so
       if (process.env.MERGE_PR == 'true') {
         console.log(`Merging PR`);
+        let mergeMethod;
+        if (context.payload.pull_request.base.repo.allow_merge_commit == true) {
+          mergeMethod = 'merge';
+        } else if (context.payload.pull_request.base.repo.allow_squash_merge == true) {
+          mergeMethod = 'squash';
+        } else {
+          mergeMethod = 'rebase';
+        }
         await context.octokit.rest.pulls.merge({
           owner: context.payload.repository.owner.login,
           repo: context.payload.repository.name,
           pull_number: context.payload.pull_request.number,
-          //merge_method: "squash"
+          merge_method: mergeMethod
         }).then(response => {
           console.log(`PR merged`);
         }).catch(error => {
diff --git a/test.js b/test.js
index eb9c4c5..e33f6b2 100644
--- a/test.js
+++ b/test.js
@@ -29,7 +29,14 @@ const payloadPrLabeled = {
       number: 1,
       url: "https://api.github.com/repos/robandpdx/superbigmono/pulls/1",
       html_url: "https://github.com/robandpdx/superbigmono/pull/1",
-      merged: false
+      merged: false,
+      base: {
+        repo: {
+          allow_merge_commit: true,
+          allow_squash_merge: true,
+          allow_rebase_merge: true
+        }
+      }
     },
     organization: {
       login: "robandpdx",
@@ -60,7 +67,14 @@ const payloadPrLabeledByBot = {
       number: 1,
       url: "https://api.github.com/repos/robandpdx/superbigmono/pulls/1",
       html_url: "https://github.com/robandpdx/superbigmono/pull/1",
-      merged: false
+      merged: false,
+      base: {
+        repo: {
+          allow_merge_commit: true,
+          allow_squash_merge: true,
+          allow_rebase_merge: true
+        }
+      }
     },
     organization: {
       login: "robandpdx",
@@ -285,6 +299,8 @@ test.after.each(() => {
   delete process.env.SLACK_MESSAGE_FILE;
   delete process.env.EMERGENCY_LABEL_PERMANENT;
   delete process.env.AUTHORIZED_TEAM;
+  payloadPrLabeled.payload.pull_request.base.repo.allow_merge_commit = true;
+  payloadPrLabeled.payload.pull_request.base.repo.allow_squash_merge = true;
 });
 
 // This test sends a payload that is not an emergency label
@@ -445,7 +461,56 @@ test("recieves pull_request.labeled event, merge the PR", async function () {
   
   // mock the request to add approval to the pr
   const mock = nock("https://api.github.com")
-    .put("/repos/robandpdx/superbigmono/pulls/1/merge").reply(200);
+    .put("/repos/robandpdx/superbigmono/pulls/1/merge",
+      (requestBody) => {
+        assert.equal(requestBody.merge_method, "merge");
+        return true;
+    }
+  ).reply(200);
+
+  await probot.receive(payloadPrLabeled);
+  assert.equal(mock.pendingMocks(), []);
+});
+
+// This test will only merge the PR, using squash merge
+test("recieves pull_request.labeled event, squash merge the PR", async function () {
+  process.env.APPROVE_PR = 'false';
+  process.env.CREATE_ISSUE = 'false';
+  process.env.MERGE_PR = 'true';
+  process.env.SLACK_NOTIFY = 'false';
+  payloadPrLabeled.payload.pull_request.base.repo.allow_merge_commit = false;
+  
+  // mock the request to add approval to the pr
+  const mock = nock("https://api.github.com")
+    .put("/repos/robandpdx/superbigmono/pulls/1/merge",
+      (requestBody) => {
+        assert.equal(requestBody.merge_method, "squash");
+        return true;
+      }
+    ).reply(200);
+    
+
+  await probot.receive(payloadPrLabeled);
+  assert.equal(mock.pendingMocks(), []);
+});
+
+// This test will only merge the PR, using rebase merge
+test("recieves pull_request.labeled event, rebase merge the PR", async function () {
+  process.env.APPROVE_PR = 'false';
+  process.env.CREATE_ISSUE = 'false';
+  process.env.MERGE_PR = 'true';
+  process.env.SLACK_NOTIFY = 'false';
+  payloadPrLabeled.payload.pull_request.base.repo.allow_merge_commit = false;
+  payloadPrLabeled.payload.pull_request.base.repo.allow_squash_merge = false;
+  
+  // mock the request to add approval to the pr
+  const mock = nock("https://api.github.com")
+    .put("/repos/robandpdx/superbigmono/pulls/1/merge",
+      (requestBody) => {
+        assert.equal(requestBody.merge_method, "rebase");
+        return true;
+      }
+    ).reply(200);
 
   await probot.receive(payloadPrLabeled);
   assert.equal(mock.pendingMocks(), []);