// Copyright 2017 The Gogs Authors. All rights reserved.// SPDX-License-Identifier: MIT package integration import (	"fmt"	"net/http"	"net/url"	"testing" 	asymkey_model "forgejo.org/models/asymkey"	auth_model "forgejo.org/models/auth"	"forgejo.org/models/perm"	repo_model "forgejo.org/models/repo"	"forgejo.org/models/unittest"	user_model "forgejo.org/models/user"	api "forgejo.org/modules/structs"	"forgejo.org/tests" 	"github.com/stretchr/testify/assert"	"github.com/stretchr/testify/require") 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)	require.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)}