squash_command_test.go

package main_test

import (
	"errors"
	"net/http"
	"net/http/httptest"

	"github.com/google/go-github/github"
	"github.com/salemove/github-review-helper/git"
	"github.com/salemove/github-review-helper/mocks"
	"github.com/stretchr/testify/mock"

	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"
)

var _ = TestWebhookHandler(func(context WebhookTestContext) {
	Describe("!squash comment", func() {
		var (
			handle      = context.Handle
			headers     = context.Headers
			requestJSON = context.RequestJSON

			responseRecorder *httptest.ResponseRecorder
			pullRequests     *mocks.PullRequests
		)
		BeforeEach(func() {
			responseRecorder = *context.ResponseRecorder
			pullRequests = *context.PullRequests
		})

		headers.Is(func() map[string]string {
			return map[string]string{
				"X-Github-Event": "issue_comment",
			}
		})
		requestJSON.Is(func() string {
			return IssueCommentEvent("!squash", arbitraryIssueAuthor)
		})

		ForCollaborator(context, repositoryOwner, repositoryName, arbitraryIssueAuthor, func() {
			Context("with GitHub request failing", func() {
				BeforeEach(func() {
					pullRequests.
						On("Get", anyContext, repositoryOwner, repositoryName, issueNumber).
						Return(emptyResult, emptyResponse, errors.New("an error"))
				})

				It("fails with a gateway error", func() {
					handle()
					Expect(responseRecorder.Code).To(Equal(http.StatusBadGateway))
				})
			})

			Context("with GitHub request succeeding", func() {
				pr := &github.PullRequest{
					Number: github.Int(issueNumber),
					Base: &github.PullRequestBranch{
						SHA:  github.String("1234"),
						Ref:  github.String("master"),
						Repo: repository,
					},
					Head: &github.PullRequestBranch{
						SHA:  github.String("1235"),
						Ref:  github.String("feature"),
						Repo: repository,
					},
				}

				BeforeEach(func() {
					pullRequests.
						On("Get", anyContext, repositoryOwner, repositoryName, issueNumber).
						Return(pr, emptyResponse, noError)
				})

				ItSquashesPR(context, pr)
			})
		})
	})
})

var ItSquashesPR = func(context WebhookTestContext, pr *github.PullRequest) {
	var (
		handle = context.Handle

		responseRecorder *httptest.ResponseRecorder
		repositories     *mocks.Repositories
		gitRepos         *mocks.Repos
		gitRepo          *mocks.Repo

		baseRef = *pr.Base.Ref
		headRef = *pr.Head.Ref
		headSHA = *pr.Head.SHA
	)

	BeforeEach(func() {
		responseRecorder = *context.ResponseRecorder
		repositories = *context.Repositories
		gitRepos = *context.GitRepos

		gitRepo = new(mocks.Repo)
		gitRepos.
			On("GetUpdatedRepo", sshURL, repositoryOwner, repositoryName).
			Return(gitRepo, noError)
	})

	AfterEach(func() {
		gitRepo.AssertExpectations(GinkgoT())
	})

	Context("with autosquash and push failing due to a squash conflict", func() {
		BeforeEach(func() {
			squashErr := &git.ErrSquashConflict{errors.New("merge conflict")}
			gitRepo.
				On("AutosquashAndPush", "origin/"+baseRef, headSHA, headRef).
				Return(squashErr)
		})

		It("reports the failure", func() {
			repositories.
				On("CreateStatus", anyContext, repositoryOwner, repositoryName, headSHA, mock.MatchedBy(func(status *github.RepoStatus) bool {
					return *status.State == "failure" && *status.Context == "review/squash"
				})).
				Return(emptyResult, emptyResponse, noError)

			handle()

			Expect(responseRecorder.Code).To(Equal(http.StatusOK))
		})
	})

	Context("with autosquash and push failing due to a reason other than a squash conflict", func() {
		BeforeEach(func() {
			gitRepo.
				On("AutosquashAndPush", "origin/"+baseRef, headSHA, headRef).
				Return(errors.New("other git error"))
		})

		It("responds with an internal server error", func() {
			handle()

			Expect(responseRecorder.Code).To(Equal(http.StatusInternalServerError))
		})
	})

	Context("with autosquash and push succeeding", func() {
		BeforeEach(func() {
			gitRepo.
				On("AutosquashAndPush", "origin/"+baseRef, headSHA, headRef).
				Return(noError)
		})

		It("returns 200 OK", func() {
			handle()

			Expect(responseRecorder.Code).To(Equal(http.StatusOK))
		})
	})
}