mirror of
https://codeberg.org/forgejo/forgejo.git
synced 2024-11-22 08:42:32 -05:00
ce563ade3d
Reviewed-on: https://codeberg.org/forgejo/forgejo/pulls/4573 Co-authored-by: TheFox0x7 <thefox0x7@gmail.com> Co-committed-by: TheFox0x7 <thefox0x7@gmail.com>
77 lines
2 KiB
Go
77 lines
2 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"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestAPIUpdateUserAvatar(t *testing.T) {
|
|
defer tests.PrepareTestEnv(t)()
|
|
|
|
normalUsername := "user2"
|
|
session := loginUser(t, normalUsername)
|
|
token := getTokenForLoggedInUser(t, session, auth_model.AccessTokenScopeWriteUser)
|
|
|
|
// Test what happens if you use a valid image
|
|
avatar, err := os.ReadFile("tests/integration/avatar.png")
|
|
require.NoError(t, err)
|
|
if err != nil {
|
|
assert.FailNow(t, "Unable to open avatar.png")
|
|
}
|
|
|
|
// Test what happens if you don't have a valid Base64 string
|
|
opts := api.UpdateUserAvatarOption{
|
|
Image: base64.StdEncoding.EncodeToString(avatar),
|
|
}
|
|
|
|
req := NewRequestWithJSON(t, "POST", "/api/v1/user/avatar", &opts).
|
|
AddTokenAuth(token)
|
|
MakeRequest(t, req, http.StatusNoContent)
|
|
|
|
opts = api.UpdateUserAvatarOption{
|
|
Image: "Invalid",
|
|
}
|
|
|
|
req = NewRequestWithJSON(t, "POST", "/api/v1/user/avatar", &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")
|
|
require.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/user/avatar", &opts).
|
|
AddTokenAuth(token)
|
|
MakeRequest(t, req, http.StatusInternalServerError)
|
|
}
|
|
|
|
func TestAPIDeleteUserAvatar(t *testing.T) {
|
|
defer tests.PrepareTestEnv(t)()
|
|
|
|
normalUsername := "user2"
|
|
session := loginUser(t, normalUsername)
|
|
token := getTokenForLoggedInUser(t, session, auth_model.AccessTokenScopeWriteUser)
|
|
|
|
req := NewRequest(t, "DELETE", "/api/v1/user/avatar").
|
|
AddTokenAuth(token)
|
|
MakeRequest(t, req, http.StatusNoContent)
|
|
}
|