582 lines · 14.1 KB · Go
RawDownload
// Copyright 2018 The Gitea Authors. All rights reserved.// SPDX-License-Identifier: MIT package integration import (	"fmt"	"net/http"	"testing" 	auth_model "forgejo.org/models/auth"	"forgejo.org/models/unittest"	user_model "forgejo.org/models/user"	"forgejo.org/modules/log"	api "forgejo.org/modules/structs"	"forgejo.org/tests" 	"github.com/stretchr/testify/assert") // TestAPICreateAndDeleteToken tests that token that was just created can be deletedfunc TestAPICreateAndDeleteToken(t *testing.T) {	defer tests.PrepareTestEnv(t)()	user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1}) 	newAccessToken := createAPIAccessTokenWithoutCleanUp(t, "test-key-1", user, []auth_model.AccessTokenScope{auth_model.AccessTokenScopeAll})	deleteAPIAccessToken(t, newAccessToken, user) 	newAccessToken = createAPIAccessTokenWithoutCleanUp(t, "test-key-2", user, []auth_model.AccessTokenScope{auth_model.AccessTokenScopeAll})	deleteAPIAccessToken(t, newAccessToken, user)} func TestAPIGetTokens(t *testing.T) {	defer tests.PrepareTestEnv(t)()	user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1}) 	// with basic auth...	req := NewRequest(t, "GET", "/api/v1/users/user2/tokens").		AddBasicAuth(user.Name)	MakeRequest(t, req, http.StatusOK) 	// ... or with a token.	newAccessToken := createAPIAccessTokenWithoutCleanUp(t, "test-key-1", user, []auth_model.AccessTokenScope{auth_model.AccessTokenScopeAll})	req = NewRequest(t, "GET", "/api/v1/users/user2/tokens").		AddTokenAuth(newAccessToken.Token)	MakeRequest(t, req, http.StatusOK)	deleteAPIAccessToken(t, newAccessToken, user)} // TestAPIDeleteMissingToken ensures that error is thrown when token not foundfunc TestAPIDeleteMissingToken(t *testing.T) {	defer tests.PrepareTestEnv(t)()	user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1}) 	req := NewRequestf(t, "DELETE", "/api/v1/users/user1/tokens/%d", unittest.NonexistentID).		AddBasicAuth(user.Name)	MakeRequest(t, req, http.StatusNotFound)} // TestAPIGetTokensPermission ensures that only the admin can get tokens from other usersfunc TestAPIGetTokensPermission(t *testing.T) {	defer tests.PrepareTestEnv(t)() 	// admin can get tokens for other users	user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1})	req := NewRequest(t, "GET", "/api/v1/users/user2/tokens").		AddBasicAuth(user.Name)	MakeRequest(t, req, http.StatusOK) 	// non-admin can get tokens for himself	user = unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2})	req = NewRequest(t, "GET", "/api/v1/users/user2/tokens").		AddBasicAuth(user.Name)	MakeRequest(t, req, http.StatusOK) 	// non-admin can't get tokens for other users	user = unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 4})	req = NewRequest(t, "GET", "/api/v1/users/user2/tokens").		AddBasicAuth(user.Name)	MakeRequest(t, req, http.StatusForbidden)} // TestAPIDeleteTokensPermission ensures that only the admin can delete tokens from other usersfunc TestAPIDeleteTokensPermission(t *testing.T) {	defer tests.PrepareTestEnv(t)() 	admin := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1})	user2 := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2})	user4 := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 4}) 	// admin can delete tokens for other users	createAPIAccessTokenWithoutCleanUp(t, "test-key-1", user2, []auth_model.AccessTokenScope{auth_model.AccessTokenScopeAll})	req := NewRequest(t, "DELETE", "/api/v1/users/"+user2.LoginName+"/tokens/test-key-1").		AddBasicAuth(admin.Name)	MakeRequest(t, req, http.StatusNoContent) 	// non-admin can delete tokens for himself	createAPIAccessTokenWithoutCleanUp(t, "test-key-2", user2, []auth_model.AccessTokenScope{auth_model.AccessTokenScopeAll})	req = NewRequest(t, "DELETE", "/api/v1/users/"+user2.LoginName+"/tokens/test-key-2").		AddBasicAuth(user2.Name)	MakeRequest(t, req, http.StatusNoContent) 	// non-admin can't delete tokens for other users	createAPIAccessTokenWithoutCleanUp(t, "test-key-3", user2, []auth_model.AccessTokenScope{auth_model.AccessTokenScopeAll})	req = NewRequest(t, "DELETE", "/api/v1/users/"+user2.LoginName+"/tokens/test-key-3").		AddBasicAuth(user4.Name)	MakeRequest(t, req, http.StatusForbidden)} type permission struct {	category auth_model.AccessTokenScopeCategory	level    auth_model.AccessTokenScopeLevel} type requiredScopeTestCase struct {	url                 string	method              string	requiredPermissions []permission} func (c *requiredScopeTestCase) Name() string {	return fmt.Sprintf("%v %v", c.method, c.url)} // TestAPIDeniesPermissionBasedOnTokenScope tests that API routes forbid access// when the correct token scope is not included.func TestAPIDeniesPermissionBasedOnTokenScope(t *testing.T) {	defer tests.PrepareTestEnv(t)() 	// We'll assert that each endpoint, when fetched with a token with all	// scopes *except* the ones specified, a forbidden status code is returned.	//	// This is to protect against endpoints having their access check copied	// from other endpoints and not updated.	//	// Test cases are in alphabetical order by URL.	testCases := []requiredScopeTestCase{		{			"/api/v1/admin/emails",			"GET",			[]permission{				{					auth_model.AccessTokenScopeCategoryAdmin,					auth_model.Read,				},			},		},		{			"/api/v1/admin/users",			"GET",			[]permission{				{					auth_model.AccessTokenScopeCategoryAdmin,					auth_model.Read,				},			},		},		{			"/api/v1/admin/users",			"POST",			[]permission{				{					auth_model.AccessTokenScopeCategoryAdmin,					auth_model.Write,				},			},		},		{			"/api/v1/admin/users/user2",			"PATCH",			[]permission{				{					auth_model.AccessTokenScopeCategoryAdmin,					auth_model.Write,				},			},		},		{			"/api/v1/admin/users/user2/orgs",			"GET",			[]permission{				{					auth_model.AccessTokenScopeCategoryAdmin,					auth_model.Read,				},			},		},		{			"/api/v1/admin/users/user2/orgs",			"POST",			[]permission{				{					auth_model.AccessTokenScopeCategoryAdmin,					auth_model.Write,				},			},		},		{			"/api/v1/admin/orgs",			"GET",			[]permission{				{					auth_model.AccessTokenScopeCategoryAdmin,					auth_model.Read,				},			},		},		{			"/api/v1/notifications",			"GET",			[]permission{				{					auth_model.AccessTokenScopeCategoryNotification,					auth_model.Read,				},			},		},		{			"/api/v1/notifications",			"PUT",			[]permission{				{					auth_model.AccessTokenScopeCategoryNotification,					auth_model.Write,				},			},		},		{			"/api/v1/org/org1/repos",			"POST",			[]permission{				{					auth_model.AccessTokenScopeCategoryOrganization,					auth_model.Write,				},				{					auth_model.AccessTokenScopeCategoryRepository,					auth_model.Write,				},			},		},		{			"/api/v1/packages/user1/type/name/1",			"GET",			[]permission{				{					auth_model.AccessTokenScopeCategoryPackage,					auth_model.Read,				},			},		},		{			"/api/v1/packages/user1/type/name/1",			"DELETE",			[]permission{				{					auth_model.AccessTokenScopeCategoryPackage,					auth_model.Write,				},			},		},		{			"/api/v1/repos/user1/repo1",			"GET",			[]permission{				{					auth_model.AccessTokenScopeCategoryRepository,					auth_model.Read,				},			},		},		{			"/api/v1/repos/user1/repo1",			"PATCH",			[]permission{				{					auth_model.AccessTokenScopeCategoryRepository,					auth_model.Write,				},			},		},		{			"/api/v1/repos/user1/repo1",			"DELETE",			[]permission{				{					auth_model.AccessTokenScopeCategoryRepository,					auth_model.Write,				},			},		},		{			"/api/v1/repos/user1/repo1/branches",			"GET",			[]permission{				{					auth_model.AccessTokenScopeCategoryRepository,					auth_model.Read,				},			},		},		{			"/api/v1/repos/user1/repo1/archive/foo",			"GET",			[]permission{				{					auth_model.AccessTokenScopeCategoryRepository,					auth_model.Read,				},			},		},		{			"/api/v1/repos/user1/repo1/issues",			"GET",			[]permission{				{					auth_model.AccessTokenScopeCategoryIssue,					auth_model.Read,				},			},		},		{			"/api/v1/repos/user1/repo1/media/foo",			"GET",			[]permission{				{					auth_model.AccessTokenScopeCategoryRepository,					auth_model.Read,				},			},		},		{			"/api/v1/repos/user1/repo1/raw/foo",			"GET",			[]permission{				{					auth_model.AccessTokenScopeCategoryRepository,					auth_model.Read,				},			},		},		{			"/api/v1/repos/user1/repo1/teams",			"GET",			[]permission{				{					auth_model.AccessTokenScopeCategoryRepository,					auth_model.Read,				},			},		},		{			"/api/v1/repos/user1/repo1/teams/team1",			"PUT",			[]permission{				{					auth_model.AccessTokenScopeCategoryRepository,					auth_model.Write,				},			},		},		{			"/api/v1/repos/user1/repo1/transfer",			"POST",			[]permission{				{					auth_model.AccessTokenScopeCategoryRepository,					auth_model.Write,				},			},		},		// Private repo		{			"/api/v1/repos/user2/repo2",			"GET",			[]permission{				{					auth_model.AccessTokenScopeCategoryRepository,					auth_model.Read,				},			},		},		// Private repo		{			"/api/v1/repos/user2/repo2",			"GET",			[]permission{				{					auth_model.AccessTokenScopeCategoryRepository,					auth_model.Read,				},			},		},		{			"/api/v1/user",			"GET",			[]permission{				{					auth_model.AccessTokenScopeCategoryUser,					auth_model.Read,				},			},		},		{			"/api/v1/user/emails",			"GET",			[]permission{				{					auth_model.AccessTokenScopeCategoryUser,					auth_model.Read,				},			},		},		{			"/api/v1/user/emails",			"POST",			[]permission{				{					auth_model.AccessTokenScopeCategoryUser,					auth_model.Write,				},			},		},		{			"/api/v1/user/emails",			"DELETE",			[]permission{				{					auth_model.AccessTokenScopeCategoryUser,					auth_model.Write,				},			},		},		{			"/api/v1/user/applications/oauth2",			"GET",			[]permission{				{					auth_model.AccessTokenScopeCategoryUser,					auth_model.Read,				},			},		},		{			"/api/v1/user/applications/oauth2",			"POST",			[]permission{				{					auth_model.AccessTokenScopeCategoryUser,					auth_model.Write,				},			},		},		{			"/api/v1/users/search",			"GET",			[]permission{				{					auth_model.AccessTokenScopeCategoryUser,					auth_model.Read,				},			},		},		// Private user		{			"/api/v1/users/user31",			"GET",			[]permission{				{					auth_model.AccessTokenScopeCategoryUser,					auth_model.Read,				},			},		},		// Private user		{			"/api/v1/users/user31/gpg_keys",			"GET",			[]permission{				{					auth_model.AccessTokenScopeCategoryUser,					auth_model.Read,				},			},		},	} 	// User needs to be admin so that we can verify that tokens without admin	// scopes correctly deny access.	user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1})	assert.True(t, user.IsAdmin, "User needs to be admin") 	for _, testCase := range testCases {		runTestCase(t, &testCase, user)	}} // runTestCase Helper function to run a single test case.func runTestCase(t *testing.T, testCase *requiredScopeTestCase, user *user_model.User) {	t.Run(testCase.Name(), func(t *testing.T) {

Lines 1–500 of 582.