forked from NYANDEV/forgejo
254a82842a
This adds an API for uploading and Deleting Avatars for of Users, Repos and Organisations. I'm not sure, if this should also be added to the Admin API. Resolves #25344 --------- Co-authored-by: silverwind <me@silverwind.io> Co-authored-by: Giteabot <teabot@gitea.io>
72 lines
1.9 KiB
Go
72 lines
1.9 KiB
Go
// Copyright 2023 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package integration
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"net/http"
|
|
"os"
|
|
"testing"
|
|
|
|
auth_model "code.gitea.io/gitea/models/auth"
|
|
api "code.gitea.io/gitea/modules/structs"
|
|
"code.gitea.io/gitea/tests"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestAPIUpdateOrgAvatar(t *testing.T) {
|
|
defer tests.PrepareTestEnv(t)()
|
|
|
|
session := loginUser(t, "user1")
|
|
|
|
token := getTokenForLoggedInUser(t, session, auth_model.AccessTokenScopeWriteOrganization)
|
|
|
|
// 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.UpdateUserAvatarOption{
|
|
Image: base64.StdEncoding.EncodeToString(avatar),
|
|
}
|
|
|
|
req := NewRequestWithJSON(t, "POST", "/api/v1/orgs/user3/avatar?token="+token, &opts)
|
|
MakeRequest(t, req, http.StatusNoContent)
|
|
|
|
// Test what happens if you don't have a valid Base64 string
|
|
opts = api.UpdateUserAvatarOption{
|
|
Image: "Invalid",
|
|
}
|
|
|
|
req = NewRequestWithJSON(t, "POST", "/api/v1/orgs/user3/avatar?token="+token, &opts)
|
|
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.UpdateUserAvatarOption{
|
|
Image: base64.StdEncoding.EncodeToString(text),
|
|
}
|
|
|
|
req = NewRequestWithJSON(t, "POST", "/api/v1/orgs/user3/avatar?token="+token, &opts)
|
|
MakeRequest(t, req, http.StatusInternalServerError)
|
|
}
|
|
|
|
func TestAPIDeleteOrgAvatar(t *testing.T) {
|
|
defer tests.PrepareTestEnv(t)()
|
|
|
|
session := loginUser(t, "user1")
|
|
|
|
token := getTokenForLoggedInUser(t, session, auth_model.AccessTokenScopeWriteOrganization)
|
|
|
|
req := NewRequest(t, "DELETE", "/api/v1/orgs/user3/avatar?token="+token)
|
|
MakeRequest(t, req, http.StatusNoContent)
|
|
}
|