From e9e87e73baba5708f520941c0a78474b5c11dbc1 Mon Sep 17 00:00:00 2001
From: Yuta Nakamura <nakamurayuta0204@gmail.com>
Date: Sat, 2 Dec 2023 20:01:25 +0900
Subject: [PATCH] fix: add some more test

---
 __tests__/run.test.ts | 120 +++++++++++++++++++++++++++++++++---------
 src/run.ts            |   1 +
 2 files changed, 97 insertions(+), 24 deletions(-)

diff --git a/__tests__/run.test.ts b/__tests__/run.test.ts
index d5fe8ec..b2867ce 100644
--- a/__tests__/run.test.ts
+++ b/__tests__/run.test.ts
@@ -1,13 +1,17 @@
 import { setFailed, info } from '@actions/core'
 import { run } from '../src/run'
-import { fetchRemoteBranches } from '../src/git-util'
-import { fetchExistingPullRequest } from '../src/github-utils'
+import { fetchRemoteBranches, hasCommitsBetween } from '../src/git-util'
+import {
+  addLabelsToPullRequest,
+  createPullRequest,
+  fetchExistingPullRequest
+} from '../src/github-utils'
 import { vi, expect, describe, it, beforeEach } from 'vitest'
 
 vi.mock('@actions/core', () => ({
   getInput: vi.fn(),
-  setFailed: vi.fn(),
-  info: vi.fn()
+  setFailed: vi.fn((msg: string) => console.error(msg)),
+  info: vi.fn((msg: string) => console.log(msg))
 }))
 
 vi.mock('../src/git-util', () => ({
@@ -16,7 +20,9 @@ vi.mock('../src/git-util', () => ({
 }))
 
 vi.mock('../src/github-utils', () => ({
-  fetchExistingPullRequest: vi.fn()
+  fetchExistingPullRequest: vi.fn(),
+  createPullRequest: vi.fn(),
+  addLabelsToPullRequest: vi.fn()
 }))
 
 describe('main function tests', () => {
@@ -30,7 +36,8 @@ describe('main function tests', () => {
       targetBranch: 'target-branch',
       repoToken: 'dummy-token',
       repo: 'test-repo',
-      owner: 'test-owner'
+      owner: 'test-owner',
+      labels: []
     })
 
     expect(setFailed).toHaveBeenCalledWith(
@@ -46,7 +53,8 @@ describe('main function tests', () => {
       targetBranch: 'valid-branch',
       repoToken: 'dummy-token',
       repo: 'test-repo',
-      owner: 'test-owner'
+      owner: 'test-owner',
+      labels: []
     })
 
     expect(setFailed).toHaveBeenCalledWith(
@@ -54,26 +62,90 @@ describe('main function tests', () => {
     )
   })
 
-  it('should fail when pull request already exist', async () => {
-    const dummyPullRequest = 'https://dummy-pr.com'
-    ;(fetchRemoteBranches as any).mockResolvedValue([
-      'valid-branch',
-      'valid-branch-2'
-    ])
-    ;(fetchExistingPullRequest as any).mockResolvedValue({
-      html_url: dummyPullRequest
+  describe('when there is valid diffrence', () => {
+    beforeEach(() => {
+      ;(fetchRemoteBranches as any).mockResolvedValue([
+        'valid-branch',
+        'valid-branch-2'
+      ])
+      ;(hasCommitsBetween as any).mockResolvedValue(true)
     })
+    describe('pull request is not created when pull request already exist', async () => {
+      const dummyPullRequest = 'https://dummy-pr.com'
+      beforeEach(() => {
+        ;(fetchExistingPullRequest as any).mockResolvedValue({
+          html_url: dummyPullRequest
+        })
+      })
 
-    await run({
-      srcBranch: 'valid-branch',
-      targetBranch: 'valid-branch-2',
-      repoToken: 'dummy-token',
-      repo: 'test-repo',
-      owner: 'test-owner'
+      it('should not create pull request when there is existing pull request', async () => {
+        await run({
+          srcBranch: 'valid-branch',
+          targetBranch: 'valid-branch-2',
+          repoToken: 'dummy-token',
+          repo: 'test-repo',
+          owner: 'test-owner',
+          labels: []
+        })
+
+        expect(info).toHaveBeenCalledWith(
+          `Pull request already exists: ${dummyPullRequest}`
+        )
+        expect(createPullRequest).not.toHaveBeenCalled()
+
+        expect(addLabelsToPullRequest).not.toHaveBeenCalled()
+      })
+
+      it('should called addLabel request', async () => {
+        await run({
+          srcBranch: 'valid-branch',
+          targetBranch: 'valid-branch-2',
+          repoToken: 'dummy-token',
+          repo: 'test-repo',
+          owner: 'test-owner',
+          labels: ['test-label', 'test-label2']
+        })
+
+        expect(info).toHaveBeenCalledWith(
+          `Pull request already exists: ${dummyPullRequest}`
+        )
+        expect(createPullRequest).not.toHaveBeenCalled()
+
+        expect(addLabelsToPullRequest).toHaveBeenCalled()
+      })
     })
 
-    expect(info).toHaveBeenCalledWith(
-      `Pull request already exists: ${dummyPullRequest}`
-    )
+    it('should create pull request when there is no existing pull request', async () => {
+      ;(fetchExistingPullRequest as any).mockResolvedValue(undefined)
+
+      await run({
+        srcBranch: 'valid-branch',
+        targetBranch: 'valid-branch-2',
+        repoToken: 'dummy-token',
+        repo: 'test-repo',
+        owner: 'test-owner',
+        labels: []
+      })
+
+      expect(createPullRequest).toHaveBeenCalled()
+      expect(addLabelsToPullRequest).not.toHaveBeenCalled()
+    })
+
+    it('should create pull request and label when there is no existing pull request', async () => {
+      ;(fetchExistingPullRequest as any).mockResolvedValue(undefined)
+      ;(createPullRequest as any).mockResolvedValue({ pullNumber: 1 })
+
+      await run({
+        srcBranch: 'valid-branch',
+        targetBranch: 'valid-branch-2',
+        repoToken: 'dummy-token',
+        repo: 'test-repo',
+        owner: 'test-owner',
+        labels: ['test-label', 'test-label2']
+      })
+
+      expect(createPullRequest).toHaveBeenCalled()
+      expect(addLabelsToPullRequest).toHaveBeenCalled()
+    })
   })
 })
diff --git a/src/run.ts b/src/run.ts
index 7b14fdf..4d5ca02 100644
--- a/src/run.ts
+++ b/src/run.ts
@@ -68,6 +68,7 @@ export async function run(options: OptionParams) {
       `origin/${targetBranch}`,
       `origin/${srcBranch}`
     )
+
     if (!hasCommits) {
       info(`No commits between ${srcBranch} and ${targetBranch}`)
       return