gitea-sonarqube-bot/internal/api/main_test.go

package api

import (
	"bytes"
	"io/ioutil"
	"log"
	"net/http"
	"net/http/httptest"
	"os"
	"testing"

	giteaSdk "gitea-sonarqube-pr-bot/internal/clients/gitea"
	sqSdk "gitea-sonarqube-pr-bot/internal/clients/sonarqube"
	"gitea-sonarqube-pr-bot/internal/settings"

	"github.com/gin-gonic/gin"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/mock"
)

type SonarQubeHandlerMock struct {
	mock.Mock
}

func (h *SonarQubeHandlerMock) Handle(rw http.ResponseWriter, r *http.Request) {
	h.Called(rw, r)
}

type GiteaHandlerMock struct {
	mock.Mock
}

func (h *GiteaHandlerMock) HandleSynchronize(rw http.ResponseWriter, r *http.Request) {
	h.Called(rw, r)
}

func (h *GiteaHandlerMock) HandleComment(rw http.ResponseWriter, r *http.Request) {
	h.Called(rw, r)
}

type GiteaSdkMock struct {
	mock.Mock
}

func (h *GiteaSdkMock) PostComment(_ settings.GiteaRepository, _ int, _ string) error {
	return nil
}

func (h *GiteaSdkMock) DetermineHEAD(_ settings.GiteaRepository, _ int64) (string, error) {
	return "", nil
}

func (h *GiteaSdkMock) UpdateStatus(_ settings.GiteaRepository, _ string, _ giteaSdk.StatusDetails) error {
	return nil
}

type SQSdkMock struct {
	mock.Mock
}

func (h *SQSdkMock) GetMeasures(project string, branch string) (*sqSdk.MeasuresResponse, error) {
	return &sqSdk.MeasuresResponse{}, nil
}

func (h *SQSdkMock) GetPullRequestUrl(project string, index int64) string {
	return ""
}

func (h *SQSdkMock) GetPullRequest(project string, index int64) (*sqSdk.PullRequest, error) {
	return &sqSdk.PullRequest{
		Status: struct {
			QualityGateStatus string "json:\"qualityGateStatus\""
		}{
			QualityGateStatus: "OK",
		},
	}, nil
}

func (h *SQSdkMock) ComposeGiteaComment(data *sqSdk.CommentComposeData) (string, error) {
	return "", nil
}

// SETUP: mute logs
func TestMain(m *testing.M) {
	gin.SetMode(gin.TestMode)
	log.SetOutput(ioutil.Discard)
	os.Exit(m.Run())
}

func TestNonAPIRoutes(t *testing.T) {
	router := New(new(GiteaHandlerMock), new(SonarQubeHandlerMock))

	w := httptest.NewRecorder()
	req, _ := http.NewRequest("GET", "/favicon.ico", nil)
	router.Engine.ServeHTTP(w, req)
	assert.Equal(t, http.StatusNoContent, w.Code)

	w = httptest.NewRecorder()
	req, _ = http.NewRequest("GET", "/ping", nil)
	router.Engine.ServeHTTP(w, req)
	assert.Equal(t, http.StatusOK, w.Code)
}

func TestSonarQubeAPIRouteMissingProjectHeader(t *testing.T) {
	router := New(new(GiteaHandlerMock), new(SonarQubeHandlerMock))

	w := httptest.NewRecorder()
	req, _ := http.NewRequest("POST", "/hooks/sonarqube", bytes.NewBuffer([]byte(`{}`)))
	router.Engine.ServeHTTP(w, req)

	assert.Equal(t, http.StatusNotFound, w.Code)
}

func TestSonarQubeAPIRouteProcessing(t *testing.T) {
	sonarQubeHandlerMock := new(SonarQubeHandlerMock)
	sonarQubeHandlerMock.On("Handle", mock.Anything, mock.Anything).Return(nil)

	router := New(new(GiteaHandlerMock), sonarQubeHandlerMock)

	w := httptest.NewRecorder()
	req, _ := http.NewRequest("POST", "/hooks/sonarqube", bytes.NewBuffer([]byte(`{}`)))
	req.Header.Add("X-SonarQube-Project", "gitea-sonarqube-bot")
	router.Engine.ServeHTTP(w, req)

	assert.Equal(t, http.StatusOK, w.Code)
	sonarQubeHandlerMock.AssertNumberOfCalls(t, "Handle", 1)
}

func TestGiteaAPIRouteMissingEventHeader(t *testing.T) {
	router := New(new(GiteaHandlerMock), new(SonarQubeHandlerMock))

	w := httptest.NewRecorder()
	req, _ := http.NewRequest("POST", "/hooks/gitea", bytes.NewBuffer([]byte(`{}`)))
	router.Engine.ServeHTTP(w, req)

	assert.Equal(t, http.StatusNotFound, w.Code)
}

func TestGiteaAPIRouteSynchronizeProcessing(t *testing.T) {
	giteaHandlerMock := new(GiteaHandlerMock)
	giteaHandlerMock.On("HandleSynchronize", mock.Anything, mock.Anything).Return(nil)
	giteaHandlerMock.On("HandleComment", mock.Anything, mock.Anything).Return(nil)

	router := New(giteaHandlerMock, new(SonarQubeHandlerMock))

	w := httptest.NewRecorder()
	req, _ := http.NewRequest("POST", "/hooks/gitea", bytes.NewBuffer([]byte(`{}`)))
	req.Header.Add("X-Gitea-Event", "pull_request")
	router.Engine.ServeHTTP(w, req)

	assert.Equal(t, http.StatusOK, w.Code)
	giteaHandlerMock.AssertNumberOfCalls(t, "HandleSynchronize", 1)
	giteaHandlerMock.AssertNumberOfCalls(t, "HandleComment", 0)
}

func TestGiteaAPIRouteCommentProcessing(t *testing.T) {
	giteaHandlerMock := new(GiteaHandlerMock)
	giteaHandlerMock.On("HandleSynchronize", mock.Anything, mock.Anything).Return(nil)
	giteaHandlerMock.On("HandleComment", mock.Anything, mock.Anything).Return(nil)

	router := New(giteaHandlerMock, new(SonarQubeHandlerMock))

	w := httptest.NewRecorder()
	req, _ := http.NewRequest("POST", "/hooks/gitea", bytes.NewBuffer([]byte(`{}`)))
	req.Header.Add("X-Gitea-Event", "issue_comment")
	router.Engine.ServeHTTP(w, req)

	assert.Equal(t, http.StatusOK, w.Code)
	giteaHandlerMock.AssertNumberOfCalls(t, "HandleSynchronize", 0)
	giteaHandlerMock.AssertNumberOfCalls(t, "HandleComment", 1)
}

func TestGiteaAPIRouteUnknownEvent(t *testing.T) {
	giteaHandlerMock := new(GiteaHandlerMock)
	giteaHandlerMock.On("HandleSynchronize", mock.Anything, mock.Anything).Return(nil)
	giteaHandlerMock.On("HandleComment", mock.Anything, mock.Anything).Return(nil)

	router := New(giteaHandlerMock, new(SonarQubeHandlerMock))

	w := httptest.NewRecorder()
	req, _ := http.NewRequest("POST", "/hooks/gitea", bytes.NewBuffer([]byte(`{}`)))
	req.Header.Add("X-Gitea-Event", "unknown")
	router.Engine.ServeHTTP(w, req)

	assert.Equal(t, http.StatusOK, w.Code)
	giteaHandlerMock.AssertNumberOfCalls(t, "HandleSynchronize", 0)
	giteaHandlerMock.AssertNumberOfCalls(t, "HandleComment", 0)
}