gitea/integrations/pull_status_test.go

// Copyright 2019 The Gitea Authors. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.
package integrations

import (
	"fmt"
	"net/http"
	"path"
	"testing"

	"code.gitea.io/gitea/models"
	api "code.gitea.io/sdk/gitea"

	"github.com/stretchr/testify/assert"
)

func TestPullCreate_CommitStatus(t *testing.T) {
	prepareTestEnv(t)
	session := loginUser(t, "user1")
	testRepoFork(t, session, "user2", "repo1", "user1", "repo1")
	testEditFileToNewBranch(t, session, "user1", "repo1", "master", "status1", "README.md", "status1")

	url := path.Join("user1", "repo1", "compare", "master...status1")
	req := NewRequestWithValues(t, "POST", url,
		map[string]string{
			"_csrf": GetCSRF(t, session, url),
			"title": "pull request from status1",
		},
	)
	session.MakeRequest(t, req, http.StatusFound)

	req = NewRequest(t, "GET", "/user1/repo1/pulls")
	resp := session.MakeRequest(t, req, http.StatusOK)
	doc := NewHTMLParser(t, resp.Body)

	// Request repository commits page
	req = NewRequest(t, "GET", "/user1/repo1/pulls/1/commits")
	resp = session.MakeRequest(t, req, http.StatusOK)
	doc = NewHTMLParser(t, resp.Body)

	// Get first commit URL
	commitURL, exists := doc.doc.Find("#commits-table tbody tr td.sha a").Last().Attr("href")
	assert.True(t, exists)
	assert.NotEmpty(t, commitURL)

	commitID := path.Base(commitURL)

	statusList := []models.CommitStatusState{
		models.CommitStatusPending,
		models.CommitStatusError,
		models.CommitStatusFailure,
		models.CommitStatusWarning,
		models.CommitStatusSuccess,
	}

	statesIcons := map[models.CommitStatusState]string{
		models.CommitStatusPending: "circle icon yellow",
		models.CommitStatusSuccess: "check icon green",
		models.CommitStatusError:   "warning icon red",
		models.CommitStatusFailure: "remove icon red",
		models.CommitStatusWarning: "warning sign icon yellow",
	}

	// Update commit status, and check if icon is updated as well
	for _, status := range statusList {

		// Call API to add status for commit
		token := getTokenForLoggedInUser(t, session)
		req = NewRequestWithJSON(t, "POST", fmt.Sprintf("/api/v1/repos/user1/repo1/statuses/%s?token=%s", commitID, token),
			api.CreateStatusOption{
				State:       api.StatusState(status),
				TargetURL:   "http://test.ci/",
				Description: "",
				Context:     "testci",
			},
		)
		session.MakeRequest(t, req, http.StatusCreated)

		req = NewRequestf(t, "GET", "/user1/repo1/pulls/1/commits")
		resp = session.MakeRequest(t, req, http.StatusOK)
		doc = NewHTMLParser(t, resp.Body)

		commitURL, exists = doc.doc.Find("#commits-table tbody tr td.sha a").Last().Attr("href")
		assert.True(t, exists)
		assert.NotEmpty(t, commitURL)
		assert.EqualValues(t, commitID, path.Base(commitURL))

		cls, ok := doc.doc.Find("#commits-table tbody tr td.message i.commit-status").Last().Attr("class")
		assert.True(t, ok)
		assert.EqualValues(t, "commit-status "+statesIcons[status], cls)
	}
}