feat: allow unhandle pull request

1024pix · Dec 23, 2024 · 2493c26 · 2493c26
1 parent f7759ac
commit 2493c26
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/build/services/merge-queue.js b/build/services/merge-queue.js
@@ -46,6 +46,14 @@ export class MergeQueue {
     });
     await this.handle({ repositoryName });
   }
+
+  async unhandlePullRequest({ repositoryName, number }) {
+    await this.#pullRequestRepository.remove({
+      repositoryName,
+      number,
+    });
+    await this.handle({ repositoryName });
+  }
 }
 
 export const mergeQueue = new MergeQueue({ pullRequestRepository, githubService });
diff --git a/test/unit/run/services/merge-queue_test.js b/test/unit/run/services/merge-queue_test.js
@@ -79,4 +79,26 @@ describe('Unit | Build | Services | merge-queue', function () {
       expect(mergeQueue.handle).to.have.been.calledOnceWithExactly({ repositoryName });
     });
   });
+
+  describe('#unhandlePullRequest', function () {
+    it('should remove pr and call handle method', async function () {
+      const repositoryName = Symbol('repository-name');
+      const pullRequestNumber = Symbol('pull-request-number');
+
+      const pullRequestRepository = {
+        remove: sinon.stub(),
+      };
+
+      const mergeQueue = new MergeQueue({ pullRequestRepository });
+      mergeQueue.handle = sinon.stub();
+
+      await mergeQueue.unhandlePullRequest({ repositoryName, number: pullRequestNumber });
+
+      expect(pullRequestRepository.remove).to.have.been.calledOnceWithExactly({
+        repositoryName,
+        number: pullRequestNumber,
+      });
+      expect(mergeQueue.handle).to.have.been.calledOnceWithExactly(repositoryName);
+    });
+  });
 });