mirror of https://github.com/go-gitea/gitea.git
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.
212 lines
9.2 KiB
212 lines
9.2 KiB
// 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", "/api/v1/user/keys?fingerprint="+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", "/api/v1/user/keys?fingerprint="+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) |
|
}
|
|
|