forked from NYANDEV/forgejo
1e6fa57acb
* Use single shared random string generation function - Replace 3 functions that do the same with 1 shared one - Use crypto/rand over math/rand for a stronger RNG - Output only alphanumerical for URL compatibilty Fixes: #15536 * use const string method * Update modules/avatar/avatar.go Co-authored-by: a1012112796 <1012112796@qq.com> Co-authored-by: a1012112796 <1012112796@qq.com>
73 lines
1.7 KiB
Go
73 lines
1.7 KiB
Go
// Copyright 2016 The Gogs Authors. All rights reserved.
|
|
// Use of this source code is governed by a MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package avatar
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"testing"
|
|
|
|
"code.gitea.io/gitea/modules/setting"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func Test_RandomImageSize(t *testing.T) {
|
|
_, err := RandomImageSize(0, []byte("gitea@local"))
|
|
assert.Error(t, err)
|
|
|
|
_, err = RandomImageSize(64, []byte("gitea@local"))
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func Test_RandomImage(t *testing.T) {
|
|
_, err := RandomImage([]byte("gitea@local"))
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func Test_PrepareWithPNG(t *testing.T) {
|
|
setting.Avatar.MaxWidth = 4096
|
|
setting.Avatar.MaxHeight = 4096
|
|
|
|
data, err := ioutil.ReadFile("testdata/avatar.png")
|
|
assert.NoError(t, err)
|
|
|
|
imgPtr, err := Prepare(data)
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, 290, (*imgPtr).Bounds().Max.X)
|
|
assert.Equal(t, 290, (*imgPtr).Bounds().Max.Y)
|
|
}
|
|
|
|
func Test_PrepareWithJPEG(t *testing.T) {
|
|
setting.Avatar.MaxWidth = 4096
|
|
setting.Avatar.MaxHeight = 4096
|
|
|
|
data, err := ioutil.ReadFile("testdata/avatar.jpeg")
|
|
assert.NoError(t, err)
|
|
|
|
imgPtr, err := Prepare(data)
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, 290, (*imgPtr).Bounds().Max.X)
|
|
assert.Equal(t, 290, (*imgPtr).Bounds().Max.Y)
|
|
}
|
|
|
|
func Test_PrepareWithInvalidImage(t *testing.T) {
|
|
setting.Avatar.MaxWidth = 5
|
|
setting.Avatar.MaxHeight = 5
|
|
|
|
_, err := Prepare([]byte{})
|
|
assert.EqualError(t, err, "DecodeConfig: image: unknown format")
|
|
}
|
|
func Test_PrepareWithInvalidImageSize(t *testing.T) {
|
|
setting.Avatar.MaxWidth = 5
|
|
setting.Avatar.MaxHeight = 5
|
|
|
|
data, err := ioutil.ReadFile("testdata/avatar.png")
|
|
assert.NoError(t, err)
|
|
|
|
_, err = Prepare(data)
|
|
assert.EqualError(t, err, "Image width is too large: 10 > 5")
|
|
}
|