52ae0c5802
Some checks failed
cache-seeder / prune (push) Has been skipped
cache-seeder / frontend (push) Failing after 3s
cache-seeder / python (push) Failing after 3s
cache-seeder / gobuild (push) Failing after 4s
cache-seeder / lint (bindata, lint-backend) (push) Failing after 3s
giteabot backport / giteabot (push) Has been skipped
giteabot / giteabot (push) Has been skipped
cache-prune / prune (push) Has been skipped
release-nightly-snapcraft / build-and-publish (ubuntu-24.04) (push) Has been cancelled
release-nightly-snapcraft / build-and-publish (ubuntu-24.04-arm) (push) Has been cancelled
release-nightly / nightly-binary (push) Has been cancelled
release-nightly / nightly-container (push) Has been cancelled
cron-renovate / cron-renovate (push) Has been skipped
67 lines
1.9 KiB
Go
67 lines
1.9 KiB
Go
// Copyright 2024 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package user
|
|
|
|
import (
|
|
"io"
|
|
"strings"
|
|
"testing"
|
|
|
|
"gitea.dev/models/unittest"
|
|
"gitea.dev/modules/setting"
|
|
"gitea.dev/modules/storage"
|
|
"gitea.dev/modules/test"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestUserAvatarLink(t *testing.T) {
|
|
defer test.MockVariableValue(&setting.AppURL, "https://localhost/")()
|
|
defer test.MockVariableValue(&setting.AppSubURL, "")()
|
|
|
|
u := &User{ID: 1, Avatar: "avatar.png"}
|
|
link := u.AvatarLink(t.Context())
|
|
assert.Equal(t, "https://localhost/avatars/avatar.png", link)
|
|
|
|
setting.AppURL = "https://localhost/sub-path/"
|
|
setting.AppSubURL = "/sub-path"
|
|
link = u.AvatarLink(t.Context())
|
|
assert.Equal(t, "https://localhost/sub-path/avatars/avatar.png", link)
|
|
}
|
|
|
|
func TestUserAvatarGenerate(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
var err error
|
|
tmpDir := t.TempDir()
|
|
storage.Avatars, err = storage.NewLocalStorage(t.Context(), &setting.Storage{Path: tmpDir})
|
|
require.NoError(t, err)
|
|
|
|
u := unittest.AssertExistsAndLoadBean(t, &User{ID: 2})
|
|
|
|
// there was no avatar, generate a new one
|
|
assert.Empty(t, u.Avatar)
|
|
err = GenerateRandomAvatar(t.Context(), u)
|
|
require.NoError(t, err)
|
|
assert.NotEmpty(t, u.Avatar)
|
|
|
|
// make sure the generated one exists
|
|
oldAvatarPath := u.CustomAvatarRelativePath()
|
|
_, err = storage.Avatars.Stat(u.CustomAvatarRelativePath())
|
|
require.NoError(t, err)
|
|
// and try to change its content
|
|
_, err = storage.Avatars.Save(u.CustomAvatarRelativePath(), strings.NewReader("abcd"), 4)
|
|
require.NoError(t, err)
|
|
|
|
// try to generate again
|
|
err = GenerateRandomAvatar(t.Context(), u)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, oldAvatarPath, u.CustomAvatarRelativePath())
|
|
f, err := storage.Avatars.Open(u.CustomAvatarRelativePath())
|
|
require.NoError(t, err)
|
|
defer f.Close()
|
|
content, _ := io.ReadAll(f)
|
|
assert.Equal(t, "abcd", string(content))
|
|
}
|