110 lines
		
	
	
	
		
			4.5 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			110 lines
		
	
	
	
		
			4.5 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| // Copyright 2017 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"
 | |
| 	"testing"
 | |
| 
 | |
| 	"code.gitea.io/gitea/models"
 | |
| 	"code.gitea.io/gitea/modules/auth"
 | |
| 	"code.gitea.io/gitea/modules/setting"
 | |
| 	api "code.gitea.io/gitea/modules/structs"
 | |
| 	issue_service "code.gitea.io/gitea/services/issue"
 | |
| 
 | |
| 	"github.com/stretchr/testify/assert"
 | |
| )
 | |
| 
 | |
| func TestAPIViewPulls(t *testing.T) {
 | |
| 	defer prepareTestEnv(t)()
 | |
| 	repo := models.AssertExistsAndLoadBean(t, &models.Repository{ID: 1}).(*models.Repository)
 | |
| 	owner := models.AssertExistsAndLoadBean(t, &models.User{ID: repo.OwnerID}).(*models.User)
 | |
| 
 | |
| 	session := loginUser(t, "user2")
 | |
| 	token := getTokenForLoggedInUser(t, session)
 | |
| 	req := NewRequestf(t, "GET", "/api/v1/repos/%s/%s/pulls?state=all&token="+token, owner.Name, repo.Name)
 | |
| 	resp := session.MakeRequest(t, req, http.StatusOK)
 | |
| 
 | |
| 	var pulls []*api.PullRequest
 | |
| 	DecodeJSON(t, resp, &pulls)
 | |
| 	expectedLen := models.GetCount(t, &models.Issue{RepoID: repo.ID}, models.Cond("is_pull = ?", true))
 | |
| 	assert.Len(t, pulls, expectedLen)
 | |
| }
 | |
| 
 | |
| // TestAPIMergePullWIP ensures that we can't merge a WIP pull request
 | |
| func TestAPIMergePullWIP(t *testing.T) {
 | |
| 	defer prepareTestEnv(t)()
 | |
| 	repo := models.AssertExistsAndLoadBean(t, &models.Repository{ID: 1}).(*models.Repository)
 | |
| 	owner := models.AssertExistsAndLoadBean(t, &models.User{ID: repo.OwnerID}).(*models.User)
 | |
| 	pr := models.AssertExistsAndLoadBean(t, &models.PullRequest{Status: models.PullRequestStatusMergeable}, models.Cond("has_merged = ?", false)).(*models.PullRequest)
 | |
| 	pr.LoadIssue()
 | |
| 	issue_service.ChangeTitle(pr.Issue, owner, setting.Repository.PullRequest.WorkInProgressPrefixes[0]+" "+pr.Issue.Title)
 | |
| 
 | |
| 	// force reload
 | |
| 	pr.LoadAttributes()
 | |
| 
 | |
| 	assert.Contains(t, pr.Issue.Title, setting.Repository.PullRequest.WorkInProgressPrefixes[0])
 | |
| 
 | |
| 	session := loginUser(t, owner.Name)
 | |
| 	token := getTokenForLoggedInUser(t, session)
 | |
| 	req := NewRequestWithJSON(t, http.MethodPost, fmt.Sprintf("/api/v1/repos/%s/%s/pulls/%d/merge?token=%s", owner.Name, repo.Name, pr.Index, token), &auth.MergePullRequestForm{
 | |
| 		MergeMessageField: pr.Issue.Title,
 | |
| 		Do:                string(models.MergeStyleMerge),
 | |
| 	})
 | |
| 
 | |
| 	session.MakeRequest(t, req, http.StatusMethodNotAllowed)
 | |
| }
 | |
| 
 | |
| func TestAPICreatePullSuccess(t *testing.T) {
 | |
| 	defer prepareTestEnv(t)()
 | |
| 	repo10 := models.AssertExistsAndLoadBean(t, &models.Repository{ID: 10}).(*models.Repository)
 | |
| 	// repo10 have code, pulls units.
 | |
| 	repo11 := models.AssertExistsAndLoadBean(t, &models.Repository{ID: 11}).(*models.Repository)
 | |
| 	// repo11 only have code unit but should still create pulls
 | |
| 	owner10 := models.AssertExistsAndLoadBean(t, &models.User{ID: repo10.OwnerID}).(*models.User)
 | |
| 	owner11 := models.AssertExistsAndLoadBean(t, &models.User{ID: repo11.OwnerID}).(*models.User)
 | |
| 
 | |
| 	session := loginUser(t, owner11.Name)
 | |
| 	token := getTokenForLoggedInUser(t, session)
 | |
| 	req := NewRequestWithJSON(t, http.MethodPost, fmt.Sprintf("/api/v1/repos/%s/%s/pulls?token=%s", owner10.Name, repo10.Name, token), &api.CreatePullRequestOption{
 | |
| 		Head:  fmt.Sprintf("%s:master", owner11.Name),
 | |
| 		Base:  "master",
 | |
| 		Title: "create a failure pr",
 | |
| 	})
 | |
| 
 | |
| 	session.MakeRequest(t, req, 201)
 | |
| }
 | |
| 
 | |
| func TestAPIEditPull(t *testing.T) {
 | |
| 	defer prepareTestEnv(t)()
 | |
| 	repo10 := models.AssertExistsAndLoadBean(t, &models.Repository{ID: 10}).(*models.Repository)
 | |
| 	owner10 := models.AssertExistsAndLoadBean(t, &models.User{ID: repo10.OwnerID}).(*models.User)
 | |
| 
 | |
| 	session := loginUser(t, owner10.Name)
 | |
| 	token := getTokenForLoggedInUser(t, session)
 | |
| 	req := NewRequestWithJSON(t, http.MethodPost, fmt.Sprintf("/api/v1/repos/%s/%s/pulls?token=%s", owner10.Name, repo10.Name, token), &api.CreatePullRequestOption{
 | |
| 		Head:  "develop",
 | |
| 		Base:  "master",
 | |
| 		Title: "create a success pr",
 | |
| 	})
 | |
| 	pull := new(api.PullRequest)
 | |
| 	resp := session.MakeRequest(t, req, 201)
 | |
| 	DecodeJSON(t, resp, pull)
 | |
| 	assert.EqualValues(t, "master", pull.Base.Name)
 | |
| 
 | |
| 	req = NewRequestWithJSON(t, http.MethodPatch, fmt.Sprintf("/api/v1/repos/%s/%s/pulls/%d?token=%s", owner10.Name, repo10.Name, pull.Index, token), &api.EditPullRequestOption{
 | |
| 		Base:  "feature/1",
 | |
| 		Title: "edit a this pr",
 | |
| 	})
 | |
| 	resp = session.MakeRequest(t, req, 201)
 | |
| 	DecodeJSON(t, resp, pull)
 | |
| 	assert.EqualValues(t, "feature/1", pull.Base.Name)
 | |
| 
 | |
| 	req = NewRequestWithJSON(t, http.MethodPatch, fmt.Sprintf("/api/v1/repos/%s/%s/pulls/%d?token=%s", owner10.Name, repo10.Name, pull.Index, token), &api.EditPullRequestOption{
 | |
| 		Base: "not-exist",
 | |
| 	})
 | |
| 	session.MakeRequest(t, req, 404)
 | |
| }
 | 
