mirror of
				https://github.com/go-gitea/gitea.git
				synced 2025-11-04 12:53:43 +01:00 
			
		
		
		
	Enables tenv and testifylint linters closes: https://github.com/go-gitea/gitea/issues/32842
		
			
				
	
	
		
			213 lines
		
	
	
		
			9.2 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			213 lines
		
	
	
		
			9.2 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
// Copyright 2017 The Gogs Authors. All rights reserved.
 | 
						|
// SPDX-License-Identifier: MIT
 | 
						|
 | 
						|
package integration
 | 
						|
 | 
						|
import (
 | 
						|
	"fmt"
 | 
						|
	"net/http"
 | 
						|
	"net/url"
 | 
						|
	"testing"
 | 
						|
 | 
						|
	asymkey_model "code.gitea.io/gitea/models/asymkey"
 | 
						|
	auth_model "code.gitea.io/gitea/models/auth"
 | 
						|
	"code.gitea.io/gitea/models/perm"
 | 
						|
	repo_model "code.gitea.io/gitea/models/repo"
 | 
						|
	"code.gitea.io/gitea/models/unittest"
 | 
						|
	user_model "code.gitea.io/gitea/models/user"
 | 
						|
	api "code.gitea.io/gitea/modules/structs"
 | 
						|
	"code.gitea.io/gitea/tests"
 | 
						|
 | 
						|
	"github.com/stretchr/testify/assert"
 | 
						|
)
 | 
						|
 | 
						|
func TestViewDeployKeysNoLogin(t *testing.T) {
 | 
						|
	defer tests.PrepareTestEnv(t)()
 | 
						|
	req := NewRequest(t, "GET", "/api/v1/repos/user2/repo1/keys")
 | 
						|
	MakeRequest(t, req, http.StatusUnauthorized)
 | 
						|
}
 | 
						|
 | 
						|
func TestCreateDeployKeyNoLogin(t *testing.T) {
 | 
						|
	defer tests.PrepareTestEnv(t)()
 | 
						|
	req := NewRequestWithJSON(t, "POST", "/api/v1/repos/user2/repo1/keys", api.CreateKeyOption{
 | 
						|
		Title: "title",
 | 
						|
		Key:   "key",
 | 
						|
	})
 | 
						|
	MakeRequest(t, req, http.StatusUnauthorized)
 | 
						|
}
 | 
						|
 | 
						|
func TestGetDeployKeyNoLogin(t *testing.T) {
 | 
						|
	defer tests.PrepareTestEnv(t)()
 | 
						|
	req := NewRequest(t, "GET", "/api/v1/repos/user2/repo1/keys/1")
 | 
						|
	MakeRequest(t, req, http.StatusUnauthorized)
 | 
						|
}
 | 
						|
 | 
						|
func TestDeleteDeployKeyNoLogin(t *testing.T) {
 | 
						|
	defer tests.PrepareTestEnv(t)()
 | 
						|
	req := NewRequest(t, "DELETE", "/api/v1/repos/user2/repo1/keys/1")
 | 
						|
	MakeRequest(t, req, http.StatusUnauthorized)
 | 
						|
}
 | 
						|
 | 
						|
func TestCreateReadOnlyDeployKey(t *testing.T) {
 | 
						|
	defer tests.PrepareTestEnv(t)()
 | 
						|
	repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{Name: "repo1"})
 | 
						|
	repoOwner := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: repo.OwnerID})
 | 
						|
 | 
						|
	session := loginUser(t, repoOwner.Name)
 | 
						|
	token := getTokenForLoggedInUser(t, session, auth_model.AccessTokenScopeWriteRepository)
 | 
						|
	keysURL := fmt.Sprintf("/api/v1/repos/%s/%s/keys", repoOwner.Name, repo.Name)
 | 
						|
	rawKeyBody := api.CreateKeyOption{
 | 
						|
		Title:    "read-only",
 | 
						|
		Key:      "ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABgQC4cn+iXnA4KvcQYSV88vGn0Yi91vG47t1P7okprVmhNTkipNRIHWr6WdCO4VDr/cvsRkuVJAsLO2enwjGWWueOO6BodiBgyAOZ/5t5nJNMCNuLGT5UIo/RI1b0WRQwxEZTRjt6mFNw6lH14wRd8ulsr9toSWBPMOGWoYs1PDeDL0JuTjL+tr1SZi/EyxCngpYszKdXllJEHyI79KQgeD0Vt3pTrkbNVTOEcCNqZePSVmUH8X8Vhugz3bnE0/iE9Pb5fkWO9c4AnM1FgI/8Bvp27Fw2ShryIXuR6kKvUqhVMTuOSDHwu6A8jLE5Owt3GAYugDpDYuwTVNGrHLXKpPzrGGPE/jPmaLCMZcsdkec95dYeU3zKODEm8UQZFhmJmDeWVJ36nGrGZHL4J5aTTaeFUJmmXDaJYiJ+K2/ioKgXqnXvltu0A9R8/LGy4nrTJRr4JMLuJFoUXvGm1gXQ70w2LSpk6yl71RNC0hCtsBe8BP8IhYCM0EP5jh7eCMQZNvM= nocomment\n",
 | 
						|
		ReadOnly: true,
 | 
						|
	}
 | 
						|
	req := NewRequestWithJSON(t, "POST", keysURL, rawKeyBody).
 | 
						|
		AddTokenAuth(token)
 | 
						|
	resp := MakeRequest(t, req, http.StatusCreated)
 | 
						|
 | 
						|
	var newDeployKey api.DeployKey
 | 
						|
	DecodeJSON(t, resp, &newDeployKey)
 | 
						|
	unittest.AssertExistsAndLoadBean(t, &asymkey_model.DeployKey{
 | 
						|
		ID:      newDeployKey.ID,
 | 
						|
		Name:    rawKeyBody.Title,
 | 
						|
		Content: rawKeyBody.Key,
 | 
						|
		Mode:    perm.AccessModeRead,
 | 
						|
	})
 | 
						|
 | 
						|
	// Using the ID of a key that does not belong to the repository must fail
 | 
						|
	{
 | 
						|
		req := NewRequest(t, "GET", fmt.Sprintf("/api/v1/repos/%s/%s/keys/%d", repoOwner.Name, repo.Name, newDeployKey.ID)).
 | 
						|
			AddTokenAuth(token)
 | 
						|
		MakeRequest(t, req, http.StatusOK)
 | 
						|
 | 
						|
		session5 := loginUser(t, "user5")
 | 
						|
		token5 := getTokenForLoggedInUser(t, session5, auth_model.AccessTokenScopeWriteRepository)
 | 
						|
		req = NewRequest(t, "GET", fmt.Sprintf("/api/v1/repos/user5/repo4/keys/%d", newDeployKey.ID)).
 | 
						|
			AddTokenAuth(token5)
 | 
						|
		MakeRequest(t, req, http.StatusNotFound)
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestCreateReadWriteDeployKey(t *testing.T) {
 | 
						|
	defer tests.PrepareTestEnv(t)()
 | 
						|
	repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{Name: "repo1"})
 | 
						|
	repoOwner := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: repo.OwnerID})
 | 
						|
 | 
						|
	session := loginUser(t, repoOwner.Name)
 | 
						|
	token := getTokenForLoggedInUser(t, session, auth_model.AccessTokenScopeWriteRepository)
 | 
						|
	keysURL := fmt.Sprintf("/api/v1/repos/%s/%s/keys", repoOwner.Name, repo.Name)
 | 
						|
	rawKeyBody := api.CreateKeyOption{
 | 
						|
		Title: "read-write",
 | 
						|
		Key:   "ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABgQC4cn+iXnA4KvcQYSV88vGn0Yi91vG47t1P7okprVmhNTkipNRIHWr6WdCO4VDr/cvsRkuVJAsLO2enwjGWWueOO6BodiBgyAOZ/5t5nJNMCNuLGT5UIo/RI1b0WRQwxEZTRjt6mFNw6lH14wRd8ulsr9toSWBPMOGWoYs1PDeDL0JuTjL+tr1SZi/EyxCngpYszKdXllJEHyI79KQgeD0Vt3pTrkbNVTOEcCNqZePSVmUH8X8Vhugz3bnE0/iE9Pb5fkWO9c4AnM1FgI/8Bvp27Fw2ShryIXuR6kKvUqhVMTuOSDHwu6A8jLE5Owt3GAYugDpDYuwTVNGrHLXKpPzrGGPE/jPmaLCMZcsdkec95dYeU3zKODEm8UQZFhmJmDeWVJ36nGrGZHL4J5aTTaeFUJmmXDaJYiJ+K2/ioKgXqnXvltu0A9R8/LGy4nrTJRr4JMLuJFoUXvGm1gXQ70w2LSpk6yl71RNC0hCtsBe8BP8IhYCM0EP5jh7eCMQZNvM= nocomment\n",
 | 
						|
	}
 | 
						|
	req := NewRequestWithJSON(t, "POST", keysURL, rawKeyBody).
 | 
						|
		AddTokenAuth(token)
 | 
						|
	resp := MakeRequest(t, req, http.StatusCreated)
 | 
						|
 | 
						|
	var newDeployKey api.DeployKey
 | 
						|
	DecodeJSON(t, resp, &newDeployKey)
 | 
						|
	unittest.AssertExistsAndLoadBean(t, &asymkey_model.DeployKey{
 | 
						|
		ID:      newDeployKey.ID,
 | 
						|
		Name:    rawKeyBody.Title,
 | 
						|
		Content: rawKeyBody.Key,
 | 
						|
		Mode:    perm.AccessModeWrite,
 | 
						|
	})
 | 
						|
}
 | 
						|
 | 
						|
func TestCreateUserKey(t *testing.T) {
 | 
						|
	defer tests.PrepareTestEnv(t)()
 | 
						|
	user := unittest.AssertExistsAndLoadBean(t, &user_model.User{Name: "user1"})
 | 
						|
 | 
						|
	session := loginUser(t, "user1")
 | 
						|
	token := url.QueryEscape(getTokenForLoggedInUser(t, session, auth_model.AccessTokenScopeWriteUser))
 | 
						|
	keyType := "ssh-rsa"
 | 
						|
	keyContent := "AAAAB3NzaC1yc2EAAAADAQABAAABgQC4cn+iXnA4KvcQYSV88vGn0Yi91vG47t1P7okprVmhNTkipNRIHWr6WdCO4VDr/cvsRkuVJAsLO2enwjGWWueOO6BodiBgyAOZ/5t5nJNMCNuLGT5UIo/RI1b0WRQwxEZTRjt6mFNw6lH14wRd8ulsr9toSWBPMOGWoYs1PDeDL0JuTjL+tr1SZi/EyxCngpYszKdXllJEHyI79KQgeD0Vt3pTrkbNVTOEcCNqZePSVmUH8X8Vhugz3bnE0/iE9Pb5fkWO9c4AnM1FgI/8Bvp27Fw2ShryIXuR6kKvUqhVMTuOSDHwu6A8jLE5Owt3GAYugDpDYuwTVNGrHLXKpPzrGGPE/jPmaLCMZcsdkec95dYeU3zKODEm8UQZFhmJmDeWVJ36nGrGZHL4J5aTTaeFUJmmXDaJYiJ+K2/ioKgXqnXvltu0A9R8/LGy4nrTJRr4JMLuJFoUXvGm1gXQ70w2LSpk6yl71RNC0hCtsBe8BP8IhYCM0EP5jh7eCMQZNvM="
 | 
						|
	rawKeyBody := api.CreateKeyOption{
 | 
						|
		Title: "test-key",
 | 
						|
		Key:   keyType + " " + keyContent,
 | 
						|
	}
 | 
						|
	req := NewRequestWithJSON(t, "POST", "/api/v1/user/keys", rawKeyBody).
 | 
						|
		AddTokenAuth(token)
 | 
						|
	resp := MakeRequest(t, req, http.StatusCreated)
 | 
						|
 | 
						|
	var newPublicKey api.PublicKey
 | 
						|
	DecodeJSON(t, resp, &newPublicKey)
 | 
						|
	fingerprint, err := asymkey_model.CalcFingerprint(rawKeyBody.Key)
 | 
						|
	assert.NoError(t, err)
 | 
						|
	unittest.AssertExistsAndLoadBean(t, &asymkey_model.PublicKey{
 | 
						|
		ID:          newPublicKey.ID,
 | 
						|
		OwnerID:     user.ID,
 | 
						|
		Name:        rawKeyBody.Title,
 | 
						|
		Fingerprint: fingerprint,
 | 
						|
		Mode:        perm.AccessModeWrite,
 | 
						|
	})
 | 
						|
 | 
						|
	// Search by fingerprint
 | 
						|
	req = NewRequest(t, "GET", fmt.Sprintf("/api/v1/user/keys?fingerprint=%s", newPublicKey.Fingerprint)).
 | 
						|
		AddTokenAuth(token)
 | 
						|
	resp = MakeRequest(t, req, http.StatusOK)
 | 
						|
 | 
						|
	var fingerprintPublicKeys []api.PublicKey
 | 
						|
	DecodeJSON(t, resp, &fingerprintPublicKeys)
 | 
						|
	assert.Equal(t, newPublicKey.Fingerprint, fingerprintPublicKeys[0].Fingerprint)
 | 
						|
	assert.Equal(t, newPublicKey.ID, fingerprintPublicKeys[0].ID)
 | 
						|
	assert.Equal(t, user.ID, fingerprintPublicKeys[0].Owner.ID)
 | 
						|
 | 
						|
	req = NewRequest(t, "GET", fmt.Sprintf("/api/v1/users/%s/keys?fingerprint=%s", user.Name, newPublicKey.Fingerprint)).
 | 
						|
		AddTokenAuth(token)
 | 
						|
	resp = MakeRequest(t, req, http.StatusOK)
 | 
						|
 | 
						|
	DecodeJSON(t, resp, &fingerprintPublicKeys)
 | 
						|
	assert.Equal(t, newPublicKey.Fingerprint, fingerprintPublicKeys[0].Fingerprint)
 | 
						|
	assert.Equal(t, newPublicKey.ID, fingerprintPublicKeys[0].ID)
 | 
						|
	assert.Equal(t, user.ID, fingerprintPublicKeys[0].Owner.ID)
 | 
						|
 | 
						|
	// Fail search by fingerprint
 | 
						|
	req = NewRequest(t, "GET", fmt.Sprintf("/api/v1/user/keys?fingerprint=%sA", newPublicKey.Fingerprint)).
 | 
						|
		AddTokenAuth(token)
 | 
						|
	resp = MakeRequest(t, req, http.StatusOK)
 | 
						|
 | 
						|
	DecodeJSON(t, resp, &fingerprintPublicKeys)
 | 
						|
	assert.Empty(t, fingerprintPublicKeys)
 | 
						|
 | 
						|
	// Fail searching for wrong users key
 | 
						|
	req = NewRequest(t, "GET", fmt.Sprintf("/api/v1/users/%s/keys?fingerprint=%s", "user2", newPublicKey.Fingerprint)).
 | 
						|
		AddTokenAuth(token)
 | 
						|
	resp = MakeRequest(t, req, http.StatusOK)
 | 
						|
 | 
						|
	DecodeJSON(t, resp, &fingerprintPublicKeys)
 | 
						|
	assert.Empty(t, fingerprintPublicKeys)
 | 
						|
 | 
						|
	// Now login as user 2
 | 
						|
	session2 := loginUser(t, "user2")
 | 
						|
	token2 := getTokenForLoggedInUser(t, session2, auth_model.AccessTokenScopeWriteUser)
 | 
						|
 | 
						|
	// Should find key even though not ours, but we shouldn't know whose it is
 | 
						|
	req = NewRequest(t, "GET", fmt.Sprintf("/api/v1/user/keys?fingerprint=%s", newPublicKey.Fingerprint)).
 | 
						|
		AddTokenAuth(token2)
 | 
						|
	resp = MakeRequest(t, req, http.StatusOK)
 | 
						|
 | 
						|
	DecodeJSON(t, resp, &fingerprintPublicKeys)
 | 
						|
	assert.Equal(t, newPublicKey.Fingerprint, fingerprintPublicKeys[0].Fingerprint)
 | 
						|
	assert.Equal(t, newPublicKey.ID, fingerprintPublicKeys[0].ID)
 | 
						|
	assert.Nil(t, fingerprintPublicKeys[0].Owner)
 | 
						|
 | 
						|
	// Should find key even though not ours, but we shouldn't know whose it is
 | 
						|
	req = NewRequest(t, "GET", fmt.Sprintf("/api/v1/users/%s/keys?fingerprint=%s", user.Name, newPublicKey.Fingerprint)).
 | 
						|
		AddTokenAuth(token2)
 | 
						|
	resp = MakeRequest(t, req, http.StatusOK)
 | 
						|
 | 
						|
	DecodeJSON(t, resp, &fingerprintPublicKeys)
 | 
						|
	assert.Equal(t, newPublicKey.Fingerprint, fingerprintPublicKeys[0].Fingerprint)
 | 
						|
	assert.Equal(t, newPublicKey.ID, fingerprintPublicKeys[0].ID)
 | 
						|
	assert.Nil(t, fingerprintPublicKeys[0].Owner)
 | 
						|
 | 
						|
	// Fail when searching for key if it is not ours
 | 
						|
	req = NewRequest(t, "GET", fmt.Sprintf("/api/v1/users/%s/keys?fingerprint=%s", "user2", newPublicKey.Fingerprint)).
 | 
						|
		AddTokenAuth(token2)
 | 
						|
	resp = MakeRequest(t, req, http.StatusOK)
 | 
						|
 | 
						|
	DecodeJSON(t, resp, &fingerprintPublicKeys)
 | 
						|
	assert.Empty(t, fingerprintPublicKeys)
 | 
						|
}
 |