|
|
|
@ -14,7 +14,7 @@ import ( |
|
|
|
|
"github.com/stretchr/testify/assert" |
|
|
|
|
) |
|
|
|
|
|
|
|
|
|
func testPullCreate(t *testing.T, session *TestSession, user, repo, branch string) *httptest.ResponseRecorder { |
|
|
|
|
func testPullCreate(t *testing.T, session *TestSession, user, repo, branch, title string) *httptest.ResponseRecorder { |
|
|
|
|
req := NewRequest(t, "GET", path.Join(user, repo)) |
|
|
|
|
resp := session.MakeRequest(t, req, http.StatusOK) |
|
|
|
|
|
|
|
|
@ -35,7 +35,7 @@ func testPullCreate(t *testing.T, session *TestSession, user, repo, branch strin |
|
|
|
|
assert.True(t, exists, "The template has changed") |
|
|
|
|
req = NewRequestWithValues(t, "POST", link, map[string]string{ |
|
|
|
|
"_csrf": htmlDoc.GetCSRF(), |
|
|
|
|
"title": "This is a pull title", |
|
|
|
|
"title": title, |
|
|
|
|
}) |
|
|
|
|
resp = session.MakeRequest(t, req, http.StatusFound) |
|
|
|
|
|
|
|
|
@ -47,7 +47,7 @@ func TestPullCreate(t *testing.T) { |
|
|
|
|
session := loginUser(t, "user1") |
|
|
|
|
testRepoFork(t, session, "user2", "repo1", "user1", "repo1") |
|
|
|
|
testEditFile(t, session, "user1", "repo1", "master", "README.md", "Hello, World (Edited)\n") |
|
|
|
|
resp := testPullCreate(t, session, "user1", "repo1", "master") |
|
|
|
|
resp := testPullCreate(t, session, "user1", "repo1", "master", "This is a pull title") |
|
|
|
|
|
|
|
|
|
// check the redirected URL
|
|
|
|
|
url := resp.HeaderMap.Get("Location") |
|
|
|
@ -68,3 +68,38 @@ func TestPullCreate(t *testing.T) { |
|
|
|
|
assert.Regexp(t, `Subject: \[PATCH\] Update 'README.md'`, resp.Body) |
|
|
|
|
assert.NotRegexp(t, "diff.*diff", resp.Body) // not two diffs, just one
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func TestPullCreate_TitleEscape(t *testing.T) { |
|
|
|
|
prepareTestEnv(t) |
|
|
|
|
session := loginUser(t, "user1") |
|
|
|
|
testRepoFork(t, session, "user2", "repo1", "user1", "repo1") |
|
|
|
|
testEditFile(t, session, "user1", "repo1", "master", "README.md", "Hello, World (Edited)\n") |
|
|
|
|
resp := testPullCreate(t, session, "user1", "repo1", "master", "<i>XSS PR</i>") |
|
|
|
|
|
|
|
|
|
// check the redirected URL
|
|
|
|
|
url := resp.HeaderMap.Get("Location") |
|
|
|
|
assert.Regexp(t, "^/user2/repo1/pulls/[0-9]*$", url) |
|
|
|
|
|
|
|
|
|
// Edit title
|
|
|
|
|
req := NewRequest(t, "GET", url) |
|
|
|
|
resp = session.MakeRequest(t, req, http.StatusOK) |
|
|
|
|
htmlDoc := NewHTMLParser(t, resp.Body) |
|
|
|
|
editTestTitleURL, exists := htmlDoc.doc.Find("#save-edit-title").First().Attr("data-update-url") |
|
|
|
|
assert.True(t, exists, "The template has changed") |
|
|
|
|
|
|
|
|
|
req = NewRequestWithValues(t, "POST", editTestTitleURL, map[string]string{ |
|
|
|
|
"_csrf": htmlDoc.GetCSRF(), |
|
|
|
|
"title": "<u>XSS PR</u>", |
|
|
|
|
}) |
|
|
|
|
session.MakeRequest(t, req, http.StatusOK) |
|
|
|
|
|
|
|
|
|
req = NewRequest(t, "GET", url) |
|
|
|
|
resp = session.MakeRequest(t, req, http.StatusOK) |
|
|
|
|
htmlDoc = NewHTMLParser(t, resp.Body) |
|
|
|
|
titleHTML, err := htmlDoc.doc.Find(".comments .event .text b").First().Html() |
|
|
|
|
assert.NoError(t, err) |
|
|
|
|
assert.Equal(t, "<i>XSS PR</i>", titleHTML) |
|
|
|
|
titleHTML, err = htmlDoc.doc.Find(".comments .event .text b").Next().Html() |
|
|
|
|
assert.NoError(t, err) |
|
|
|
|
assert.Equal(t, "<u>XSS PR</u>", titleHTML) |
|
|
|
|
} |
|
|
|
|