Платформа ЦРНП "Мирокод" для разработки проектов
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.
194 lines
7.7 KiB
194 lines
7.7 KiB
// Copyright 2017 The Gogs 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 ( |
|
"fmt" |
|
"net/http" |
|
"net/url" |
|
"testing" |
|
|
|
"github.com/stretchr/testify/assert" |
|
|
|
"code.gitea.io/gitea/models" |
|
api "code.gitea.io/sdk/gitea" |
|
) |
|
|
|
func TestViewDeployKeysNoLogin(t *testing.T) { |
|
prepareTestEnv(t) |
|
req := NewRequest(t, "GET", "/api/v1/repos/user2/repo1/keys") |
|
MakeRequest(t, req, http.StatusUnauthorized) |
|
} |
|
|
|
func TestCreateDeployKeyNoLogin(t *testing.T) { |
|
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) { |
|
prepareTestEnv(t) |
|
req := NewRequest(t, "GET", "/api/v1/repos/user2/repo1/keys/1") |
|
MakeRequest(t, req, http.StatusUnauthorized) |
|
} |
|
|
|
func TestDeleteDeployKeyNoLogin(t *testing.T) { |
|
prepareTestEnv(t) |
|
req := NewRequest(t, "DELETE", "/api/v1/repos/user2/repo1/keys/1") |
|
MakeRequest(t, req, http.StatusUnauthorized) |
|
} |
|
|
|
func TestCreateReadOnlyDeployKey(t *testing.T) { |
|
prepareTestEnv(t) |
|
repo := models.AssertExistsAndLoadBean(t, &models.Repository{Name: "repo1"}).(*models.Repository) |
|
repoOwner := models.AssertExistsAndLoadBean(t, &models.User{ID: repo.OwnerID}).(*models.User) |
|
|
|
session := loginUser(t, repoOwner.Name) |
|
token := getTokenForLoggedInUser(t, session) |
|
keysURL := fmt.Sprintf("/api/v1/repos/%s/%s/keys?token=%s", repoOwner.Name, repo.Name, token) |
|
rawKeyBody := api.CreateKeyOption{ |
|
Title: "read-only", |
|
Key: "ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAAAgQDAu7tvIvX6ZHrRXuZNfkR3XLHSsuCK9Zn3X58lxBcQzuo5xZgB6vRwwm/QtJuF+zZPtY5hsQILBLmF+BZ5WpKZp1jBeSjH2G7lxet9kbcH+kIVj0tPFEoyKI9wvWqIwC4prx/WVk2wLTJjzBAhyNxfEq7C9CeiX9pQEbEqJfkKCQ== nocomment\n", |
|
ReadOnly: true, |
|
} |
|
req := NewRequestWithJSON(t, "POST", keysURL, rawKeyBody) |
|
resp := session.MakeRequest(t, req, http.StatusCreated) |
|
|
|
var newDeployKey api.DeployKey |
|
DecodeJSON(t, resp, &newDeployKey) |
|
models.AssertExistsAndLoadBean(t, &models.DeployKey{ |
|
ID: newDeployKey.ID, |
|
Name: rawKeyBody.Title, |
|
Content: rawKeyBody.Key, |
|
Mode: models.AccessModeRead, |
|
}) |
|
} |
|
|
|
func TestCreateReadWriteDeployKey(t *testing.T) { |
|
prepareTestEnv(t) |
|
repo := models.AssertExistsAndLoadBean(t, &models.Repository{Name: "repo1"}).(*models.Repository) |
|
repoOwner := models.AssertExistsAndLoadBean(t, &models.User{ID: repo.OwnerID}).(*models.User) |
|
|
|
session := loginUser(t, repoOwner.Name) |
|
token := getTokenForLoggedInUser(t, session) |
|
keysURL := fmt.Sprintf("/api/v1/repos/%s/%s/keys?token=%s", repoOwner.Name, repo.Name, token) |
|
rawKeyBody := api.CreateKeyOption{ |
|
Title: "read-write", |
|
Key: "ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABAQDsufOCrDDlT8DLkodnnJtbq7uGflcPae7euTfM+Laq4So+v4WeSV362Rg0O/+Sje1UthrhN6lQkfRkdWIlCRQEXg+LMqr6RhvDfZquE2Xwqv/itlz7LjbdAUdYoO1iH7rMSmYvQh4WEnC/DAacKGbhdGIM/ZBz0z6tHm7bPgbI9ykEKekTmPwQFP1Qebvf5NYOFMWqQ2sCEAI9dBMVLoojsIpV+KADf+BotiIi8yNfTG2rzmzpxBpW9fYjd1Sy1yd4NSUpoPbEJJYJ1TrjiSWlYOVq9Ar8xW1O87i6gBjL/3zN7ANeoYhaAXupdOS6YL22YOK/yC0tJtXwwdh/eSrh", |
|
} |
|
req := NewRequestWithJSON(t, "POST", keysURL, rawKeyBody) |
|
resp := session.MakeRequest(t, req, http.StatusCreated) |
|
|
|
var newDeployKey api.DeployKey |
|
DecodeJSON(t, resp, &newDeployKey) |
|
models.AssertExistsAndLoadBean(t, &models.DeployKey{ |
|
ID: newDeployKey.ID, |
|
Name: rawKeyBody.Title, |
|
Content: rawKeyBody.Key, |
|
Mode: models.AccessModeWrite, |
|
}) |
|
} |
|
|
|
func TestCreateUserKey(t *testing.T) { |
|
prepareTestEnv(t) |
|
user := models.AssertExistsAndLoadBean(t, &models.User{Name: "user1"}).(*models.User) |
|
|
|
session := loginUser(t, "user1") |
|
token := url.QueryEscape(getTokenForLoggedInUser(t, session)) |
|
keysURL := fmt.Sprintf("/api/v1/user/keys?token=%s", token) |
|
keyType := "ssh-rsa" |
|
keyContent := "AAAAB3NzaC1yc2EAAAADAQABAAABAQCyTiPTeHJl6Gs5D1FyHT0qTWpVkAy9+LIKjctQXklrePTvUNVrSpt4r2exFYXNMPeA8V0zCrc3Kzs1SZw3jWkG3i53te9onCp85DqyatxOD2pyZ30/gPn1ZUg40WowlFM8gsUFMZqaH7ax6d8nsBKW7N/cRyqesiOQEV9up3tnKjIB8XMTVvC5X4rBWgywz7AFxSv8mmaTHnUgVW4LgMPwnTWo0pxtiIWbeMLyrEE4hIM74gSwp6CRQYo6xnG3fn4yWkcK2X2mT9adQ241IDdwpENJHcry/T6AJ8dNXduEZ67egnk+rVlQ2HM4LpymAv9DAAFFeaQK0hT+3aMDoumV" |
|
rawKeyBody := api.CreateKeyOption{ |
|
Title: "test-key", |
|
Key: keyType + " " + keyContent, |
|
} |
|
req := NewRequestWithJSON(t, "POST", keysURL, rawKeyBody) |
|
resp := session.MakeRequest(t, req, http.StatusCreated) |
|
|
|
var newPublicKey api.PublicKey |
|
DecodeJSON(t, resp, &newPublicKey) |
|
models.AssertExistsAndLoadBean(t, &models.PublicKey{ |
|
ID: newPublicKey.ID, |
|
OwnerID: user.ID, |
|
Name: rawKeyBody.Title, |
|
Content: rawKeyBody.Key, |
|
Mode: models.AccessModeWrite, |
|
}) |
|
|
|
// Search by fingerprint |
|
fingerprintURL := fmt.Sprintf("/api/v1/user/keys?token=%s&fingerprint=%s", token, newPublicKey.Fingerprint) |
|
|
|
req = NewRequest(t, "GET", fingerprintURL) |
|
resp = session.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) |
|
|
|
fingerprintURL = fmt.Sprintf("/api/v1/users/%s/keys?token=%s&fingerprint=%s", user.Name, token, newPublicKey.Fingerprint) |
|
|
|
req = NewRequest(t, "GET", fingerprintURL) |
|
resp = session.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 |
|
fingerprintURL = fmt.Sprintf("/api/v1/user/keys?token=%s&fingerprint=%sA", token, newPublicKey.Fingerprint) |
|
|
|
req = NewRequest(t, "GET", fingerprintURL) |
|
resp = session.MakeRequest(t, req, http.StatusOK) |
|
|
|
DecodeJSON(t, resp, &fingerprintPublicKeys) |
|
assert.Len(t, fingerprintPublicKeys, 0) |
|
|
|
// Fail searching for wrong users key |
|
fingerprintURL = fmt.Sprintf("/api/v1/users/%s/keys?token=%s&fingerprint=%s", "user2", token, newPublicKey.Fingerprint) |
|
req = NewRequest(t, "GET", fingerprintURL) |
|
resp = session.MakeRequest(t, req, http.StatusOK) |
|
|
|
DecodeJSON(t, resp, &fingerprintPublicKeys) |
|
assert.Len(t, fingerprintPublicKeys, 0) |
|
|
|
// Now login as user 2 |
|
session2 := loginUser(t, "user2") |
|
token2 := url.QueryEscape(getTokenForLoggedInUser(t, session2)) |
|
|
|
// Should find key even though not ours, but we shouldn't know whose it is |
|
fingerprintURL = fmt.Sprintf("/api/v1/user/keys?token=%s&fingerprint=%s", token2, newPublicKey.Fingerprint) |
|
req = NewRequest(t, "GET", fingerprintURL) |
|
resp = session.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 |
|
fingerprintURL = fmt.Sprintf("/api/v1/users/%s/keys?token=%s&fingerprint=%s", user.Name, token2, newPublicKey.Fingerprint) |
|
|
|
req = NewRequest(t, "GET", fingerprintURL) |
|
resp = session.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 |
|
fingerprintURL = fmt.Sprintf("/api/v1/users/%s/keys?token=%s&fingerprint=%s", "user2", token2, newPublicKey.Fingerprint) |
|
req = NewRequest(t, "GET", fingerprintURL) |
|
resp = session.MakeRequest(t, req, http.StatusOK) |
|
|
|
DecodeJSON(t, resp, &fingerprintPublicKeys) |
|
assert.Len(t, fingerprintPublicKeys, 0) |
|
}
|
|
|