// Copyright 2017 The Gitea Authors. All rights reserved. // Copyright 2024 The Forgejo Authors. All rights reserved. // SPDX-License-Identifier: MIT package integration import ( "fmt" "net/http" "strconv" "strings" "testing" "time" auth_model "code.gitea.io/gitea/models/auth" issues_model "code.gitea.io/gitea/models/issues" repo_model "code.gitea.io/gitea/models/repo" unit_model "code.gitea.io/gitea/models/unit" "code.gitea.io/gitea/models/unittest" user_model "code.gitea.io/gitea/models/user" "code.gitea.io/gitea/modules/setting" api "code.gitea.io/gitea/modules/structs" "code.gitea.io/gitea/modules/test" "code.gitea.io/gitea/modules/translation" gitea_context "code.gitea.io/gitea/services/context" "code.gitea.io/gitea/services/mailer" "code.gitea.io/gitea/tests" "github.com/pquerna/otp/totp" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" ) func TestViewUser(t *testing.T) { defer tests.PrepareTestEnv(t)() req := NewRequest(t, "GET", "/user2") MakeRequest(t, req, http.StatusOK) } func TestRenameUsername(t *testing.T) { defer tests.PrepareTestEnv(t)() session := loginUser(t, "user2") req := NewRequestWithValues(t, "POST", "/user/settings", map[string]string{ "_csrf": GetCSRF(t, session, "/user/settings"), "name": "newUsername", "email": "user2@example.com", "language": "en-US", }) session.MakeRequest(t, req, http.StatusSeeOther) unittest.AssertExistsAndLoadBean(t, &user_model.User{Name: "newUsername"}) unittest.AssertNotExistsBean(t, &user_model.User{Name: "user2"}) } func TestRenameInvalidUsername(t *testing.T) { defer tests.PrepareTestEnv(t)() invalidUsernames := []string{ "%2f*", "%2f.", "%2f..", "%00", "thisHas ASpace", "ptho>lo session := loginUser(t, "user1") token := getTokenForLoggedInUser(t, session, auth_model.AccessTokenScopeWriteUser) testCreateGPGKey(t, session.MakeRequest, token, http.StatusCreated, `-----BEGIN PGP PUBLIC KEY BLOCK----- mQENBFyy/VUBCADJ7zbM20Z1RWmFoVgp5WkQfI2rU1Vj9cQHes9i42wVLLtcbPeo QzubgzvMPITDy7nfWxgSf83E23DoHQ1ACFbQh/6eFSRrjsusp3YQ/08NSfPPbcu8 0M5G+VGwSfzS5uEcwBVQmHyKdcOZIERTNMtYZx1C3bjLD1XVJHvWz9D72Uq4qeO3 8SR+lzp5n6ppUakcmRnxt3nGRBj1+hEGkdgzyPo93iy+WioegY2lwCA9xMEo5dah BmYxWx51zyiXYlReTaxlyb3/nuSUt8IcW3Q8zjdtJj4Nu8U1SpV8EdaA1I9IPbHW 510OSLmD3XhqHH5m6mIxL1YoWxk3V7gpDROtABEBAAG0GVVzZXIxIDx1c2VyMUBl eGFtcGxlLmNvbT6JAU4EEwEIADgWIQTQEbrYxmXsp1z3j7z9+v0I6RSEHwUCXLL9 VQIbAwULCQgHAgYVCgkICwIEFgIDAQIeAQIXgAAKCRD9+v0I6RSEH22YCACFqL5+ 6M0m18AMC/pumcpnnmvAS1GrrKTF8nOROA1augZwp1WCNuKw2R6uOJIHANrYECSn u7+j6GBP2gbIW8mSAzS6HWCs7GGiPpVtT4wcu8wljUI6BxjpyZtoEkriyBjt6HfK rkegbkuySoJvjq4IcO5D1LB1JWgsUjMYQJj/ZpBIzVtjG9QtFSOiT1Hct4PoZHdC nsdSgyCkwRZXG+u3kT/wP9F663ba4o16vYlz3dCGo66lF2tyoG3qcyZ1OUzUrnuv 96ytAzT6XIhrE0nVoBprMxFF5zExotJD3bHjcGBFNLf944bhjKee3U6t9+OsfJVC l7N5xxIawCuTQdbfuQENBFyy/VUBCADe61yGEoTwKfsOKIhxLaNoRmD883O0tiWt soO/HPj9dPQLTOiwXgSgSCd8C+LNxGKct87wgFozpah4tDLC6c0nALuHJ0SLbkfz 55aRhLeOOcrAydatDp72GroXzqpZ0xZBk5wjIWdgEol2GmVRM8QGbeuakU/HVz5y lPzxUUocgdbSi3GE3zbzijQzVJdyL/kw/KP7pKT/PPKKJ2C5NQDLy0XGKEHddXGR EWKkVlRalxq/TjfaMR0bi3MpezBsQmp99ATPO/d7trayZUxQHRtXzGFiOXfDHATr qN730sODjqvU+mpc/SHCRwh9qWDjZRHSuKU5YDBjb5jIQJivZsQ/ABEBAAGJATYE GAEIACAWIQTQEbrYxmXsp1z3j7z9+v0I6RSEHwUCXLL9VQIbDAAKCRD9+v0I6RSE H7WoB/4tXl+97rQ6owPCGSVp1Xbwt2521V7COgsOFRVTRTryEWxRW8mm0S7wQvax C0TLXKur6NVYQMn01iyL+FZzRpEWNuYF3f9QeeLJ/+l2DafESNhNTy17+RPmacK6 21dccpqchByVw/UMDeHSyjQLiG2lxzt8Gfx2gHmSbrq3aWovTGyz6JTffZvfy/n2 0Hm437OBPazO0gZyXhdV2PE5RSUfvAgm44235tcV5EV0d32TJDfv61+Vr2GUbah6 7XhJ1v6JYuh8kaYaEz8OpZDeh7f6Ho6PzJrsy/TKTKhGgZNINj1iaPFyOkQgKR5M GrE0MHOxUbc9tbtyk0F1SuzREUBH =DDXw -----END PGP PUBLIC KEY BLOCK-----`) // Export new key testExportUserGPGKeys(t, "user1", `-----BEGIN PGP PUBLIC KEY BLOCK----- xsBNBFyy/VUBCADJ7zbM20Z1RWmFoVgp5WkQfI2rU1Vj9cQHes9i42wVLLtcbPeo QzubgzvMPITDy7nfWxgSf83E23DoHQ1ACFbQh/6eFSRrjsusp3YQ/08NSfPPbcu8 0M5G+VGwSfzS5uEcwBVQmHyKdcOZIERTNMtYZx1C3bjLD1XVJHvWz9D72Uq4qeO3 8SR+lzp5n6ppUakcmRnxt3nGRBj1+hEGkdgzyPo93iy+WioegY2lwCA9xMEo5dah BmYxWx51zyiXYlReTaxlyb3/nuSUt8IcW3Q8zjdtJj4Nu8U1SpV8EdaA1I9IPbHW 510OSLmD3XhqHH5m6mIxL1YoWxk3V7gpDROtABEBAAHNGVVzZXIxIDx1c2VyMUBl eGFtcGxlLmNvbT7CwI4EEwEIADgWIQTQEbrYxmXsp1z3j7z9+v0I6RSEHwUCXLL9 VQIbAwULCQgHAgYVCgkICwIEFgIDAQIeAQIXgAAKCRD9+v0I6RSEH22YCACFqL5+ 6M0m18AMC/pumcpnnmvAS1GrrKTF8nOROA1augZwp1WCNuKw2R6uOJIHANrYECSn u7+j6GBP2gbIW8mSAzS6HWCs7GGiPpVtT4wcu8wljUI6BxjpyZtoEkriyBjt6HfK rkegbkuySoJvjq4IcO5D1LB1JWgsUjMYQJj/ZpBIzVtjG9QtFSOiT1Hct4PoZHdC nsdSgyCkwRZXG+u3kT/wP9F663ba4o16vYlz3dCGo66lF2tyoG3qcyZ1OUzUrnuv 96ytAzT6XIhrE0nVoBprMxFF5zExotJD3bHjcGBFNLf944bhjKee3U6t9+OsfJVC l7N5xxIawCuTQdbfzsBNBFyy/VUBCADe61yGEoTwKfsOKIhxLaNoRmD883O0tiWt soO/HPj9dPQLTOiwXgSgSCd8C+LNxGKct87wgFozpah4tDLC6c0nALuHJ0SLbkfz 55aRhLeOOcrAydatDp72GroXzqpZ0xZBk5wjIWdgEol2GmVRM8QGbeuakU/HVz5y lPzxUUocgdbSi3GE3zbzijQzVJdyL/kw/KP7pKT/PPKKJ2C5NQDLy0XGKEHddXGR EWKkVlRalxq/TjfaMR0bi3MpezBsQmp99ATPO/d7trayZUxQHRtXzGFiOXfDHATr qN730sODjqvU+mpc/SHCRwh9qWDjZRHSuKU5YDBjb5jIQJivZsQ/ABEBAAHCwHYE GAEIACAWIQTQEbrYxmXsp1z3j7z9+v0I6RSEHwUCXLL9VQIbDAAKCRD9+v0I6RSE H7WoB/4tXl+97rQ6owPCGSVp1Xbwt2521V7COgsOFRVTRTryEWxRW8mm0S7wQvax C0TLXKur6NVYQMn01iyL+FZzRpEWNuYF3f9QeeLJ/+l2DafESNhNTy17+RPmacK6 21dccpqchByVw/UMDeHSyjQLiG2lxzt8Gfx2gHmSbrq3aWovTGyz6JTffZvfy/n2 0Hm437OBPazO0gZyXhdV2PE5RSUfvAgm44235tcV5EV0d32TJDfv61+Vr2GUbah6 7XhJ1v6JYuh8kaYaEz8OpZDeh7f6Ho6PzJrsy/TKTKhGgZNINj1iaPFyOkQgKR5M GrE0MHOxUbc9tbtyk0F1SuzREUBH =WFf5 -----END PGP PUBLIC KEY BLOCK-----`) } func testExportUserGPGKeys(t *testing.T, user, expected string) { session := loginUser(t, user) t.Logf("Testing username %s export gpg keys", user) req := NewRequest(t, "GET", "/"+user+".gpg") resp := session.MakeRequest(t, req, http.StatusOK) // t.Log(resp.Body.String()) assert.Equal(t, expected, resp.Body.String()) } func TestGetUserRss(t *testing.T) { defer tests.PrepareTestEnv(t)() t.Run("Normal", func(t *testing.T) { user34 := "the_34-user.with.all.allowedChars" req := NewRequestf(t, "GET", "/%s.rss", user34) resp := MakeRequest(t, req, http.StatusOK) if assert.EqualValues(t, "application/rss+xml;charset=utf-8", resp.Header().Get("Content-Type")) { rssDoc := NewHTMLParser(t, resp.Body).Find("channel") title, _ := rssDoc.ChildrenFiltered("title").Html() assert.EqualValues(t, "Feed of "the_1-user.with.all.allowedChars"", title) description, _ := rssDoc.ChildrenFiltered("description").Html() assert.EqualValues(t, "<p dir="auto">some <a href="https://commonmark.org/" rel="nofollow">commonmark</a>!</p>\n", description) } }) t.Run("Non-existent user", func(t *testing.T) { session := loginUser(t, "user2") req := NewRequestf(t, "GET", "/non-existent-user.rss") session.MakeRequest(t, req, http.StatusNotFound) }) } func TestListStopWatches(t *testing.T) { defer tests.PrepareTestEnv(t)() repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 1}) owner := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: repo.OwnerID}) session := loginUser(t, owner.Name) req := NewRequest(t, "GET", "/user/stopwatches") resp := session.MakeRequest(t, req, http.StatusOK) var apiWatches []*api.StopWatch DecodeJSON(t, resp, &apiWatches) stopwatch := unittest.AssertExistsAndLoadBean(t, &issues_model.Stopwatch{UserID: owner.ID}) issue := unittest.AssertExistsAndLoadBean(t, &issues_model.Issue{ID: stopwatch.IssueID}) if assert.Len(t, apiWatches, 1) { assert.EqualValues(t, stopwatch.CreatedUnix.AsTime().Unix(), apiWatches[0].Created.Unix()) assert.EqualValues(t, issue.Index, apiWatches[0].IssueIndex) assert.EqualValues(t, issue.Title, apiWatches[0].IssueTitle) assert.EqualValues(t, repo.Name, apiWatches[0].RepoName) assert.EqualValues(t, repo.OwnerName, apiWatches[0].RepoOwnerName) assert.Positive(t, apiWatches[0].Seconds) } } func TestUserLocationMapLink(t *testing.T) { setting.Service.UserLocationMapURL = "https://example/foo/" defer tests.PrepareTestEnv(t)() session := loginUser(t, "user2") req := NewRequestWithValues(t, "POST", "/user/settings", map[string]string{ "_csrf": GetCSRF(t, session, "/user/settings"), "name": "user2", "email": "user@example.com", "language": "en-US", "location": "A/b", }) session.MakeRequest(t, req, http.StatusSeeOther) req = NewRequest(t, "GET", "/user2/") resp := session.MakeRequest(t, req, http.StatusOK) htmlDoc := NewHTMLParser(t, resp.Body) htmlDoc.AssertElement(t, `a[href="https://example/foo/A%2Fb"]`, true) } func TestUserHints(t *testing.T) { defer tests.PrepareTestEnv(t)() user := unittest.AssertExistsAndLoadBean(t, &user_model.User{Name: "user2"}) session := loginUser(t, user.Name) token := getTokenForLoggedInUser(t, session, auth_model.AccessTokenScopeWriteUser) // Create a known-good repo, with only one unit enabled repo, _, f := tests.CreateDeclarativeRepo(t, user, "", []unit_model.Type{ unit_model.TypeCode, }, []unit_model.Type{ unit_model.TypePullRequests, unit_model.TypeProjects, unit_model.TypePackages, unit_model.TypeActions, unit_model.TypeIssues, unit_model.TypeWiki, }, nil) defer f() ensureRepoUnitHints := func(t *testing.T, hints bool) { t.Helper() req := NewRequestWithJSON(t, "PATCH", "/api/v1/user/settings", &api.UserSettingsOptions{ EnableRepoUnitHints: &hints, }).AddTokenAuth(token) resp := MakeRequest(t, req, http.StatusOK) var userSettings api.UserSettings DecodeJSON(t, resp, &userSettings) assert.Equal(t, hints, userSettings.EnableRepoUnitHints) } t.Run("API", func(t *testing.T) { t.Run("setting hints on and off", func(t *testing.T) { defer tests.PrintCurrentTest(t)() ensureRepoUnitHints(t, true) ensureRepoUnitHints(t, false) }) t.Run("retrieving settings", func(t *testing.T) { defer tests.PrintCurrentTest(t)() for _, v := range []bool{true, false} { ensureRepoUnitHints(t, v) req := NewRequest(t, "GET", "/api/v1/user/settings").AddTokenAuth(token) resp := MakeRequest(t, req, http.StatusOK) var userSettings api.UserSettings DecodeJSON(t, resp, &userSettings) assert.Equal(t, v, userSettings.EnableRepoUnitHints) } }) }) t.Run("user settings", func(t *testing.T) { defer tests.PrintCurrentTest(t)() // Set a known-good state, that isn't the default ensureRepoUnitHints(t, false) assertHintState := func(t *testing.T, enabled bool) { t.Helper() req := NewRequest(t, "GET", "/user/settings/appearance") resp := session.MakeRequest(t, req, http.StatusOK) htmlDoc := NewHTMLParser(t, resp.Body) _, hintChecked := htmlDoc.Find(`input[name="enable_repo_unit_hints"]`).Attr("checked") assert.Equal(t, enabled, hintChecked) link, _ := htmlDoc.Find("form[action='/user/settings/appearance/language'] a").Attr("href") assert.EqualValues(t, "https://forgejo.org/docs/latest/developer/localization/", link) } t.Run("view", func(t *testing.T) { defer tests.PrintCurrentTest(t)() assertHintState(t, false) }) t.Run("change", func(t *testing.T) { defer tests.PrintCurrentTest(t)() req := NewRequestWithValues(t, "POST", "/user/settings/appearance/hints", map[string]string{ "_csrf": GetCSRF(t, session, "/user/settings/appearance"), "enable_repo_unit_hints": "true", }) session.MakeRequest(t, req, http.StatusSeeOther) assertHintState(t, true) }) }) t.Run("repo view", func(t *testing.T) { assertAddMore := func(t *testing.T, present bool) { t.Helper() req := NewRequest(t, "GET", repo.Link()) resp := session.MakeRequest(t, req, http.StatusOK) htmlDoc := NewHTMLParser(t, resp.Body) htmlDoc.AssertElement(t, fmt.Sprintf("a[href='%s/settings/units']", repo.Link()), present) } t.Run("hints enabled", func(t *testing.T) { defer tests.PrintCurrentTest(t)() ensureRepoUnitHints(t, true) assertAddMore(t, true) }) t.Run("hints disabled", func(t *testing.T) { defer tests.PrintCurrentTest(t)() ensureRepoUnitHints(t, false) assertAddMore(t, false) }) }) } func TestUserPronouns(t *testing.T) { defer tests.PrepareTestEnv(t)() session := loginUser(t, "user2") token := getTokenForLoggedInUser(t, session, auth_model.AccessTokenScopeWriteUser) adminUser := unittest.AssertExistsAndLoadBean(t, &user_model.User{IsAdmin: true}) adminSession := loginUser(t, adminUser.Name) adminToken := getTokenForLoggedInUser(t, adminSession, auth_model.AccessTokenScopeWriteAdmin) t.Run("API", func(t *testing.T) { t.Run("user", func(t *testing.T) { defer tests.PrintCurrentTest(t)() req := NewRequest(t, "GET", "/api/v1/user").AddTokenAuth(token) resp := MakeRequest(t, req, http.StatusOK) // We check the raw JSON, because we want to test the response, not // what it decodes into. Contents doesn't matter, we're testing the // presence only. assert.Contains(t, resp.Body.String(), `"pronouns":`) }) t.Run("users/{username}", func(t *testing.T) { defer tests.PrintCurrentTest(t)() req := NewRequest(t, "GET", "/api/v1/users/user2") resp := MakeRequest(t, req, http.StatusOK) // We check the raw JSON, because we want to test the response, not // what it decodes into. Contents doesn't matter, we're testing the // presence only. assert.Contains(t, resp.Body.String(), `"pronouns":`) }) t.Run("user/settings", func(t *testing.T) { defer tests.PrintCurrentTest(t)() // Set pronouns first pronouns := "they/them" req := NewRequestWithJSON(t, "PATCH", "/api/v1/user/settings", &api.UserSettingsOptions{ Pronouns: &pronouns, }).AddTokenAuth(token) resp := MakeRequest(t, req, http.StatusOK) // Verify the response var user *api.UserSettings DecodeJSON(t, resp, &user) assert.Equal(t, pronouns, user.Pronouns) // Verify retrieving the settings again req = NewRequest(t, "GET", "/api/v1/user/settings").AddTokenAuth(token) resp = MakeRequest(t, req, http.StatusOK) DecodeJSON(t, resp, &user) assert.Equal(t, pronouns, user.Pronouns) }) t.Run("admin/users/{username}", func(t *testing.T) { defer tests.PrintCurrentTest(t)() // Set the pronouns for user2 pronouns := "she/her" req := NewRequestWithJSON(t, "PATCH", "/api/v1/admin/users/user2", &api.EditUserOption{ Pronouns: &pronouns, }).AddTokenAuth(adminToken) resp := MakeRequest(t, req, http.StatusOK) // Verify the API response var user *api.User DecodeJSON(t, resp, &user) assert.Equal(t, pronouns, user.Pronouns) // Verify via user2 too req = NewRequest(t, "GET", "/api/v1/user").AddTokenAuth(token) resp = MakeRequest(t, req, http.StatusOK) DecodeJSON(t, resp, &user) assert.Equal(t, pronouns, user.Pronouns) }) }) t.Run("UI", func(t *testing.T) { defer tests.PrintCurrentTest(t)() // Set the pronouns to a known state via the API pronouns := "she/her" req := NewRequestWithJSON(t, "PATCH", "/api/v1/user/settings", &api.UserSettingsOptions{ Pronouns: &pronouns, }).AddTokenAuth(token) MakeRequest(t, req, http.StatusOK) t.Run("profile view", func(t *testing.T) { defer tests.PrintCurrentTest(t)() req := NewRequest(t, "GET", "/user2") resp := MakeRequest(t, req, http.StatusOK) htmlDoc := NewHTMLParser(t, resp.Body) userNameAndPronouns := strings.TrimSpace(htmlDoc.Find(".profile-avatar-name .username").Text()) assert.Contains(t, userNameAndPronouns, pronouns) }) t.Run("settings", func(t *testing.T) { defer tests.PrintCurrentTest(t)() req := NewRequest(t, "GET", "/user/settings") resp := session.MakeRequest(t, req, http.StatusOK) htmlDoc := NewHTMLParser(t, resp.Body) // Check that the field is present pronounField, has := htmlDoc.Find(`input[name="pronouns"]`).Attr("value") assert.True(t, has) assert.Equal(t, pronouns, pronounField) // Check that updating the field works newPronouns := "they/them" req = NewRequestWithValues(t, "POST", "/user/settings", map[string]string{ "_csrf": GetCSRF(t, session, "/user/settings"), "pronouns": newPronouns, }) session.MakeRequest(t, req, http.StatusSeeOther) user2 := unittest.AssertExistsAndLoadBean(t, &user_model.User{Name: "user2"}) assert.Equal(t, newPronouns, user2.Pronouns) }) t.Run("admin settings", func(t *testing.T) { defer tests.PrintCurrentTest(t)() user2 := unittest.AssertExistsAndLoadBean(t, &user_model.User{Name: "user2"}) req := NewRequestf(t, "GET", "/admin/users/%d/edit", user2.ID) resp := adminSession.MakeRequest(t, req, http.StatusOK) htmlDoc := NewHTMLParser(t, resp.Body) // Check that the pronouns field is present pronounField, has := htmlDoc.Find(`input[name="pronouns"]`).Attr("value") assert.True(t, has) assert.NotEmpty(t, pronounField) // Check that updating the field works newPronouns := "it/its" editURI := fmt.Sprintf("/admin/users/%d/edit", user2.ID) req = NewRequestWithValues(t, "POST", editURI, map[string]string{ "_csrf": GetCSRF(t, adminSession, editURI), "login_type": "0-0", "login_name": user2.LoginName, "email": user2.Email, "pronouns": newPronouns, }) adminSession.MakeRequest(t, req, http.StatusSeeOther) user2New := unittest.AssertExistsAndLoadBean(t, &user_model.User{Name: "user2"}) assert.Equal(t, newPronouns, user2New.Pronouns) }) }) t.Run("unspecified", func(t *testing.T) { defer tests.PrintCurrentTest(t)() // Set the pronouns to Unspecified (an empty string) via the API pronouns := "" req := NewRequestWithJSON(t, "PATCH", "/api/v1/admin/users/user2", &api.EditUserOption{ Pronouns: &pronouns, }).AddTokenAuth(adminToken) MakeRequest(t, req, http.StatusOK) // Verify that the profile page does not display any pronouns, nor the separator req = NewRequest(t, "GET", "/user2") resp := MakeRequest(t, req, http.StatusOK) htmlDoc := NewHTMLParser(t, resp.Body) userName := strings.TrimSpace(htmlDoc.Find(".profile-avatar-name .username").Text()) assert.EqualValues(t, "user2", userName) }) } func TestUserTOTPMail(t *testing.T) { defer tests.PrepareTestEnv(t)() user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2}) session := loginUser(t, user.Name) t.Run("No security keys", func(t *testing.T) { defer tests.PrintCurrentTest(t)() called := false defer test.MockVariableValue(&mailer.SendAsync, func(msgs ...*mailer.Message) { assert.Len(t, msgs, 1) assert.Equal(t, user.EmailTo(), msgs[0].To) assert.EqualValues(t, translation.NewLocale("en-US").Tr("mail.totp_disabled.subject"), msgs[0].Subject) assert.Contains(t, msgs[0].Body, translation.NewLocale("en-US").Tr("mail.totp_disabled.no_2fa")) called = true })() unittest.AssertSuccessfulInsert(t, &auth_model.TwoFactor{UID: user.ID}) req := NewRequestWithValues(t, "POST", "/user/settings/security/two_factor/disable", map[string]string{ "_csrf": GetCSRF(t, session, "/user/settings/security"), }) session.MakeRequest(t, req, http.StatusSeeOther) assert.True(t, called) unittest.AssertExistsIf(t, false, &auth_model.TwoFactor{UID: user.ID}) }) t.Run("with security keys", func(t *testing.T) { defer tests.PrintCurrentTest(t)() called := false defer test.MockVariableValue(&mailer.SendAsync, func(msgs ...*mailer.Message) { assert.Len(t, msgs, 1) assert.Equal(t, user.EmailTo(), msgs[0].To) assert.EqualValues(t, translation.NewLocale("en-US").Tr("mail.totp_disabled.subject"), msgs[0].Subject) assert.NotContains(t, msgs[0].Body, translation.NewLocale("en-US").Tr("mail.totp_disabled.no_2fa")) called = true })() unittest.AssertSuccessfulInsert(t, &auth_model.TwoFactor{UID: user.ID}) unittest.AssertSuccessfulInsert(t, &auth_model.WebAuthnCredential{UserID: user.ID}) req := NewRequestWithValues(t, "POST", "/user/settings/security/two_factor/disable", map[string]string{ "_csrf": GetCSRF(t, session, "/user/settings/security"), }) session.MakeRequest(t, req, http.StatusSeeOther) assert.True(t, called) unittest.AssertExistsIf(t, false, &auth_model.TwoFactor{UID: user.ID}) }) } func TestUserSecurityKeyMail(t *testing.T) { defer tests.PrepareTestEnv(t)() user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2}) session := loginUser(t, user.Name) t.Run("Normal", func(t *testing.T) { defer tests.PrintCurrentTest(t)() called := false defer test.MockVariableValue(&mailer.SendAsync, func(msgs ...*mailer.Message) { assert.Len(t, msgs, 1) assert.Equal(t, user.EmailTo(), msgs[0].To) assert.EqualValues(t, translation.NewLocale("en-US").Tr("mail.removed_security_key.subject"), msgs[0].Subject) assert.Contains(t, msgs[0].Body, translation.NewLocale("en-US").Tr("mail.removed_security_key.no_2fa")) assert.Contains(t, msgs[0].Body, "Little Bobby Tables's primary key") called = true })() unittest.AssertSuccessfulInsert(t, &auth_model.WebAuthnCredential{UserID: user.ID, Name: "Little Bobby Tables's primary key"}) id := unittest.AssertExistsAndLoadBean(t, &auth_model.WebAuthnCredential{UserID: user.ID}).ID req := NewRequestWithValues(t, "POST", "/user/settings/security/webauthn/delete", map[string]string{ "_csrf": GetCSRF(t, session, "/user/settings/security"), "id": strconv.FormatInt(id, 10), }) session.MakeRequest(t, req, http.StatusOK) assert.True(t, called) unittest.AssertExistsIf(t, false, &auth_model.WebAuthnCredential{UserID: user.ID}) }) t.Run("With TOTP", func(t *testing.T) { defer tests.PrintCurrentTest(t)() called := false defer test.MockVariableValue(&mailer.SendAsync, func(msgs ...*mailer.Message) { assert.Len(t, msgs, 1) assert.Equal(t, user.EmailTo(), msgs[0].To) assert.EqualValues(t, translation.NewLocale("en-US").Tr("mail.removed_security_key.subject"), msgs[0].Subject) assert.NotContains(t, msgs[0].Body, translation.NewLocale("en-US").Tr("mail.removed_security_key.no_2fa")) assert.Contains(t, msgs[0].Body, "Little Bobby Tables's primary key") called = true })() unittest.AssertSuccessfulInsert(t, &auth_model.WebAuthnCredential{UserID: user.ID, Name: "Little Bobby Tables's primary key"}) id := unittest.AssertExistsAndLoadBean(t, &auth_model.WebAuthnCredential{UserID: user.ID}).ID unittest.AssertSuccessfulInsert(t, &auth_model.TwoFactor{UID: user.ID}) req := NewRequestWithValues(t, "POST", "/user/settings/security/webauthn/delete", map[string]string{ "_csrf": GetCSRF(t, session, "/user/settings/security"), "id": strconv.FormatInt(id, 10), }) session.MakeRequest(t, req, http.StatusOK) assert.True(t, called) unittest.AssertExistsIf(t, false, &auth_model.WebAuthnCredential{UserID: user.ID}) }) t.Run("Two security keys", func(t *testing.T) { defer tests.PrintCurrentTest(t)() called := false defer test.MockVariableValue(&mailer.SendAsync, func(msgs ...*mailer.Message) { assert.Len(t, msgs, 1) assert.Equal(t, user.EmailTo(), msgs[0].To) assert.EqualValues(t, translation.NewLocale("en-US").Tr("mail.removed_security_key.subject"), msgs[0].Subject) assert.NotContains(t, msgs[0].Body, translation.NewLocale("en-US").Tr("mail.removed_security_key.no_2fa")) assert.Contains(t, msgs[0].Body, "Little Bobby Tables's primary key") called = true })() unittest.AssertSuccessfulInsert(t, &auth_model.WebAuthnCredential{UserID: user.ID, Name: "Little Bobby Tables's primary key"}) id := unittest.AssertExistsAndLoadBean(t, &auth_model.WebAuthnCredential{UserID: user.ID}).ID unittest.AssertSuccessfulInsert(t, &auth_model.WebAuthnCredential{UserID: user.ID, Name: "Little Bobby Tables's evil key"}) req := NewRequestWithValues(t, "POST", "/user/settings/security/webauthn/delete", map[string]string{ "_csrf": GetCSRF(t, session, "/user/settings/security"), "id": strconv.FormatInt(id, 10), }) session.MakeRequest(t, req, http.StatusOK) assert.True(t, called) unittest.AssertExistsIf(t, false, &auth_model.WebAuthnCredential{UserID: user.ID, Name: "Little Bobby Tables's primary key"}) unittest.AssertExistsIf(t, true, &auth_model.WebAuthnCredential{UserID: user.ID, Name: "Little Bobby Tables's evil key"}) }) } func TestUserTOTPEnrolled(t *testing.T) { defer tests.PrepareTestEnv(t)() user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2}) session := loginUser(t, user.Name) enrollTOTP := func(t *testing.T) { t.Helper() req := NewRequest(t, "GET", "/user/settings/security/two_factor/enroll") resp := session.MakeRequest(t, req, http.StatusOK) htmlDoc := NewHTMLParser(t, resp.Body) totpSecretKey, has := htmlDoc.Find(".twofa img[src^='data:image/png;base64']").Attr("alt") assert.True(t, has) currentTOTP, err := totp.GenerateCode(totpSecretKey, time.Now()) require.NoError(t, err) req = NewRequestWithValues(t, "POST", "/user/settings/security/two_factor/enroll", map[string]string{ "_csrf": htmlDoc.GetCSRF(), "passcode": currentTOTP, }) session.MakeRequest(t, req, http.StatusSeeOther) flashCookie := session.GetCookie(gitea_context.CookieNameFlash) assert.NotNil(t, flashCookie) assert.Contains(t, flashCookie.Value, "success%3DYour%2Baccount%2Bhas%2Bbeen%2Bsuccessfully%2Benrolled.") unittest.AssertSuccessfulDelete(t, &auth_model.TwoFactor{UID: user.ID}) } t.Run("No WebAuthn enabled", func(t *testing.T) { defer tests.PrintCurrentTest(t)() called := false defer test.MockVariableValue(&mailer.SendAsync, func(msgs ...*mailer.Message) { assert.Len(t, msgs, 1) assert.Equal(t, user.EmailTo(), msgs[0].To) assert.EqualValues(t, translation.NewLocale("en-US").Tr("mail.totp_enrolled.subject"), msgs[0].Subject) assert.Contains(t, msgs[0].Body, translation.NewLocale("en-US").Tr("mail.totp_enrolled.text_1.no_webauthn")) called = true })() enrollTOTP(t) assert.True(t, called) }) t.Run("With WebAuthn enabled", func(t *testing.T) { defer tests.PrintCurrentTest(t)() called := false defer test.MockVariableValue(&mailer.SendAsync, func(msgs ...*mailer.Message) { assert.Len(t, msgs, 1) assert.Equal(t, user.EmailTo(), msgs[0].To) assert.EqualValues(t, translation.NewLocale("en-US").Tr("mail.totp_enrolled.subject"), msgs[0].Subject) assert.Contains(t, msgs[0].Body, translation.NewLocale("en-US").Tr("mail.totp_enrolled.text_1.has_webauthn")) called = true })() unittest.AssertSuccessfulInsert(t, &auth_model.WebAuthnCredential{UserID: user.ID, Name: "Cueball's primary key"}) enrollTOTP(t) assert.True(t, called) }) }