@ -5,19 +5,24 @@
package integrations
package integrations
import (
import (
"fmt"
"net/http"
"net/http"
"testing"
"testing"
"code.gitea.io/gitea/models"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/assert"
)
)
func testRepoFork ( t * testing . T , session * TestSession ) * TestResponse {
func testRepoFork ( t * testing . T , session * TestSession , ownerName , repoName , forkOwnerName , forkRepoName string ) * TestResponse {
forkOwner := models . AssertExistsAndLoadBean ( t , & models . User { Name : forkOwnerName } ) . ( * models . User )
// Step0: check the existence of the to-fork repo
// Step0: check the existence of the to-fork repo
req := NewRequest ( t , "GET" , "/user1/repo1" )
req := NewRequestf ( t , "GET" , "/%s/%s" , forkOwnerName , forkRepoName )
resp := session . MakeRequest ( t , req , http . StatusNotFound )
resp := session . MakeRequest ( t , req , http . StatusNotFound )
// Step1: go to the main page of repo
// Step1: go to the main page of repo
req = NewRequest ( t , "GET" , "/user2/repo1" )
req = NewRequestf ( t , "GET" , "/%s/%s" , ownerName , repoName )
resp = session . MakeRequest ( t , req , http . StatusOK )
resp = session . MakeRequest ( t , req , http . StatusOK )
// Step2: click the fork button
// Step2: click the fork button
@ -31,15 +36,17 @@ func testRepoFork(t *testing.T, session *TestSession) *TestResponse {
htmlDoc = NewHTMLParser ( t , resp . Body )
htmlDoc = NewHTMLParser ( t , resp . Body )
link , exists = htmlDoc . doc . Find ( "form.ui.form[action^=\"/repo/fork/\"]" ) . Attr ( "action" )
link , exists = htmlDoc . doc . Find ( "form.ui.form[action^=\"/repo/fork/\"]" ) . Attr ( "action" )
assert . True ( t , exists , "The template has changed" )
assert . True ( t , exists , "The template has changed" )
_ , exists = htmlDoc . doc . Find ( fmt . Sprintf ( ".owner.dropdown .item[data-value=\"%d\"]" , forkOwner . ID ) ) . Attr ( "data-value" )
assert . True ( t , exists , fmt . Sprintf ( "Fork owner '%s' is not present in select box" , forkOwnerName ) )
req = NewRequestWithValues ( t , "POST" , link , map [ string ] string {
req = NewRequestWithValues ( t , "POST" , link , map [ string ] string {
"_csrf" : htmlDoc . GetCSRF ( ) ,
"_csrf" : htmlDoc . GetCSRF ( ) ,
"uid" : "1" ,
"uid" : fmt . Sprintf ( "%d" , forkOwner . ID ) ,
"repo_name" : "repo1" ,
"repo_name" : forkRepoName ,
} )
} )
resp = session . MakeRequest ( t , req , http . StatusFound )
resp = session . MakeRequest ( t , req , http . StatusFound )
// Step4: check the existence of the forked repo
// Step4: check the existence of the forked repo
req = NewRequest ( t , "GET" , "/user1/repo1" )
req = NewRequestf ( t , "GET" , "/%s/%s" , forkOwnerName , forkRepoName )
resp = session . MakeRequest ( t , req , http . StatusOK )
resp = session . MakeRequest ( t , req , http . StatusOK )
return resp
return resp
@ -48,5 +55,19 @@ func testRepoFork(t *testing.T, session *TestSession) *TestResponse {
func TestRepoFork ( t * testing . T ) {
func TestRepoFork ( t * testing . T ) {
prepareTestEnv ( t )
prepareTestEnv ( t )
session := loginUser ( t , "user1" )
session := loginUser ( t , "user1" )
testRepoFork ( t , session )
testRepoFork ( t , session , "user2" , "repo1" , "user1" , "repo1" )
}
func TestRepoForkToOrg ( t * testing . T ) {
prepareTestEnv ( t )
session := loginUser ( t , "user2" )
testRepoFork ( t , session , "user2" , "repo1" , "user3" , "repo1" )
// Check that no more forking is allowed as user2 owns repository
// and user3 organization that owner user2 is also now has forked this repository
req := NewRequest ( t , "GET" , "/user2/repo1" )
resp := session . MakeRequest ( t , req , http . StatusOK )
htmlDoc := NewHTMLParser ( t , resp . Body )
_ , exists := htmlDoc . doc . Find ( "a.ui.button[href^=\"/repo/fork/\"]" ) . Attr ( "href" )
assert . False ( t , exists , "Forking should not be allowed anymore" )
}
}