2023-06-30 07:22:55 +08:00
|
|
|
// 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),
|
|
|
|
}
|
|
|
|
|
2023-09-14 10:59:53 +08:00
|
|
|
req := NewRequestWithJSON(t, "POST", "/api/v1/orgs/org3/avatar?token="+token, &opts)
|
2023-06-30 07:22:55 +08:00
|
|
|
MakeRequest(t, req, http.StatusNoContent)
|
|
|
|
|
|
|
|
// Test what happens if you don't have a valid Base64 string
|
|
|
|
opts = api.UpdateUserAvatarOption{
|
|
|
|
Image: "Invalid",
|
|
|
|
}
|
|
|
|
|
2023-09-14 10:59:53 +08:00
|
|
|
req = NewRequestWithJSON(t, "POST", "/api/v1/orgs/org3/avatar?token="+token, &opts)
|
2023-06-30 07:22:55 +08:00
|
|
|
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),
|
|
|
|
}
|
|
|
|
|
2023-09-14 10:59:53 +08:00
|
|
|
req = NewRequestWithJSON(t, "POST", "/api/v1/orgs/org3/avatar?token="+token, &opts)
|
2023-06-30 07:22:55 +08:00
|
|
|
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)
|
|
|
|
|
2023-09-14 10:59:53 +08:00
|
|
|
req := NewRequest(t, "DELETE", "/api/v1/orgs/org3/avatar?token="+token)
|
2023-06-30 07:22:55 +08:00
|
|
|
MakeRequest(t, req, http.StatusNoContent)
|
|
|
|
}
|