Платформа ЦРНП "Мирокод" для разработки проектов
https://git.mirocod.ru
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
52 lines
1.7 KiB
52 lines
1.7 KiB
// 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 ( |
|
"net/http" |
|
"strings" |
|
"testing" |
|
|
|
"code.gitea.io/gitea/models/unittest" |
|
user_model "code.gitea.io/gitea/models/user" |
|
"code.gitea.io/gitea/modules/setting" |
|
|
|
"github.com/stretchr/testify/assert" |
|
) |
|
|
|
func TestCsrfProtection(t *testing.T) { |
|
defer prepareTestEnv(t)() |
|
|
|
// test web form csrf via form |
|
user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2}).(*user_model.User) |
|
session := loginUser(t, user.Name) |
|
req := NewRequestWithValues(t, "POST", "/user/settings", map[string]string{ |
|
"_csrf": "fake_csrf", |
|
}) |
|
session.MakeRequest(t, req, http.StatusSeeOther) |
|
|
|
resp := session.MakeRequest(t, req, http.StatusSeeOther) |
|
loc := resp.Header().Get("Location") |
|
assert.Equal(t, setting.AppSubURL+"/", loc) |
|
resp = session.MakeRequest(t, NewRequest(t, "GET", loc), http.StatusOK) |
|
htmlDoc := NewHTMLParser(t, resp.Body) |
|
assert.Equal(t, "Bad Request: invalid CSRF token", |
|
strings.TrimSpace(htmlDoc.doc.Find(".ui.message").Text()), |
|
) |
|
|
|
// test web form csrf via header. TODO: should use an UI api to test |
|
req = NewRequest(t, "POST", "/user/settings") |
|
req.Header.Add("X-Csrf-Token", "fake_csrf") |
|
session.MakeRequest(t, req, http.StatusSeeOther) |
|
|
|
resp = session.MakeRequest(t, req, http.StatusSeeOther) |
|
loc = resp.Header().Get("Location") |
|
assert.Equal(t, setting.AppSubURL+"/", loc) |
|
resp = session.MakeRequest(t, NewRequest(t, "GET", loc), http.StatusOK) |
|
htmlDoc = NewHTMLParser(t, resp.Body) |
|
assert.Equal(t, "Bad Request: invalid CSRF token", |
|
strings.TrimSpace(htmlDoc.doc.Find(".ui.message").Text()), |
|
) |
|
}
|
|
|