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.
80 lines
2.5 KiB
80 lines
2.5 KiB
// Copyright 2018 The Gitea Authors. All rights reserved. |
|
// SPDX-License-Identifier: MIT |
|
|
|
package integration |
|
|
|
import ( |
|
"encoding/base64" |
|
"fmt" |
|
"net/http" |
|
"os" |
|
"testing" |
|
|
|
auth_model "code.gitea.io/gitea/models/auth" |
|
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 TestAPIUpdateRepoAvatar(t *testing.T) { |
|
defer tests.PrepareTestEnv(t)() |
|
|
|
repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 1}) |
|
user2 := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2}) |
|
token := getUserToken(t, user2.LowerName, auth_model.AccessTokenScopeWriteRepository) |
|
|
|
// Test what happens if you use a valid image |
|
avatar, err := os.ReadFile("tests/integration/avatar.png") |
|
assert.NoError(t, err) |
|
if err != nil { |
|
assert.FailNow(t, "Unable to open avatar.png") |
|
} |
|
|
|
opts := api.UpdateRepoAvatarOption{ |
|
Image: base64.StdEncoding.EncodeToString(avatar), |
|
} |
|
|
|
req := NewRequestWithJSON(t, "POST", fmt.Sprintf("/api/v1/repos/%s/%s/avatar", repo.OwnerName, repo.Name), &opts). |
|
AddTokenAuth(token) |
|
MakeRequest(t, req, http.StatusNoContent) |
|
|
|
// Test what happens if you don't have a valid Base64 string |
|
opts = api.UpdateRepoAvatarOption{ |
|
Image: "Invalid", |
|
} |
|
|
|
req = NewRequestWithJSON(t, "POST", fmt.Sprintf("/api/v1/repos/%s/%s/avatar", repo.OwnerName, repo.Name), &opts). |
|
AddTokenAuth(token) |
|
MakeRequest(t, req, http.StatusBadRequest) |
|
|
|
// Test what happens if you use a file that is not an image |
|
text, err := os.ReadFile("tests/integration/README.md") |
|
assert.NoError(t, err) |
|
if err != nil { |
|
assert.FailNow(t, "Unable to open README.md") |
|
} |
|
|
|
opts = api.UpdateRepoAvatarOption{ |
|
Image: base64.StdEncoding.EncodeToString(text), |
|
} |
|
|
|
req = NewRequestWithJSON(t, "POST", fmt.Sprintf("/api/v1/repos/%s/%s/avatar", repo.OwnerName, repo.Name), &opts). |
|
AddTokenAuth(token) |
|
MakeRequest(t, req, http.StatusInternalServerError) |
|
} |
|
|
|
func TestAPIDeleteRepoAvatar(t *testing.T) { |
|
defer tests.PrepareTestEnv(t)() |
|
|
|
repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 1}) |
|
user2 := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2}) |
|
token := getUserToken(t, user2.LowerName, auth_model.AccessTokenScopeWriteRepository) |
|
|
|
req := NewRequest(t, "DELETE", fmt.Sprintf("/api/v1/repos/%s/%s/avatar", repo.OwnerName, repo.Name)). |
|
AddTokenAuth(token) |
|
MakeRequest(t, req, http.StatusNoContent) |
|
}
|
|
|