76 lines
		
	
	
		
			2.0 KiB
		
	
	
	
		
			Go
		
	
	
	
			
		
		
	
	
			76 lines
		
	
	
		
			2.0 KiB
		
	
	
	
		
			Go
		
	
	
	
// Copyright 2017 The Gitea Authors. All rights reserved.
 | 
						|
// SPDX-License-Identifier: MIT
 | 
						|
 | 
						|
package integration
 | 
						|
 | 
						|
import (
 | 
						|
	"net/http"
 | 
						|
	"net/url"
 | 
						|
	"testing"
 | 
						|
 | 
						|
	"code.gitea.io/gitea/modules/translation"
 | 
						|
	"code.gitea.io/gitea/tests"
 | 
						|
 | 
						|
	"github.com/stretchr/testify/assert"
 | 
						|
)
 | 
						|
 | 
						|
func TestViewBranches(t *testing.T) {
 | 
						|
	defer tests.PrepareTestEnv(t)()
 | 
						|
 | 
						|
	req := NewRequest(t, "GET", "/user2/repo1/branches")
 | 
						|
	resp := MakeRequest(t, req, http.StatusOK)
 | 
						|
 | 
						|
	htmlDoc := NewHTMLParser(t, resp.Body)
 | 
						|
	_, exists := htmlDoc.doc.Find(".delete-branch-button").Attr("data-url")
 | 
						|
	assert.False(t, exists, "The template has changed")
 | 
						|
}
 | 
						|
 | 
						|
func TestDeleteBranch(t *testing.T) {
 | 
						|
	defer tests.PrepareTestEnv(t)()
 | 
						|
 | 
						|
	deleteBranch(t)
 | 
						|
}
 | 
						|
 | 
						|
func TestUndoDeleteBranch(t *testing.T) {
 | 
						|
	onGiteaRun(t, func(t *testing.T, u *url.URL) {
 | 
						|
		deleteBranch(t)
 | 
						|
		htmlDoc, name := branchAction(t, ".restore-branch-button")
 | 
						|
		assert.Contains(t,
 | 
						|
			htmlDoc.doc.Find(".ui.positive.message").Text(),
 | 
						|
			translation.NewLocale("en-US").Tr("repo.branch.restore_success", name),
 | 
						|
		)
 | 
						|
	})
 | 
						|
}
 | 
						|
 | 
						|
func deleteBranch(t *testing.T) {
 | 
						|
	htmlDoc, name := branchAction(t, ".delete-branch-button")
 | 
						|
	assert.Contains(t,
 | 
						|
		htmlDoc.doc.Find(".ui.positive.message").Text(),
 | 
						|
		translation.NewLocale("en-US").Tr("repo.branch.deletion_success", name),
 | 
						|
	)
 | 
						|
}
 | 
						|
 | 
						|
func branchAction(t *testing.T, button string) (*HTMLDoc, string) {
 | 
						|
	session := loginUser(t, "user2")
 | 
						|
	req := NewRequest(t, "GET", "/user2/repo1/branches")
 | 
						|
	resp := session.MakeRequest(t, req, http.StatusOK)
 | 
						|
 | 
						|
	htmlDoc := NewHTMLParser(t, resp.Body)
 | 
						|
	link, exists := htmlDoc.doc.Find(button).Attr("data-url")
 | 
						|
	if !assert.True(t, exists, "The template has changed") {
 | 
						|
		t.Skip()
 | 
						|
	}
 | 
						|
 | 
						|
	req = NewRequestWithValues(t, "POST", link, map[string]string{
 | 
						|
		"_csrf": htmlDoc.GetCSRF(),
 | 
						|
	})
 | 
						|
	session.MakeRequest(t, req, http.StatusOK)
 | 
						|
 | 
						|
	url, err := url.Parse(link)
 | 
						|
	assert.NoError(t, err)
 | 
						|
	req = NewRequest(t, "GET", "/user2/repo1/branches")
 | 
						|
	resp = session.MakeRequest(t, req, http.StatusOK)
 | 
						|
 | 
						|
	return NewHTMLParser(t, resp.Body), url.Query().Get("name")
 | 
						|
}
 |