f2e7d5477f
The io/ioutil package has been deprecated as of Go 1.16, see https://golang.org/doc/go1.16#ioutil. This commit replaces the existing io/ioutil functions with their new definitions in io and os packages. Signed-off-by: Eng Zer Jun <engzerjun@gmail.com> Co-authored-by: techknowlogick <techknowlogick@gitea.io>
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 (
|
|
"os"
|
|
"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 := os.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 := os.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 := os.ReadFile("testdata/avatar.png")
|
|
assert.NoError(t, err)
|
|
|
|
_, err = Prepare(data)
|
|
assert.EqualError(t, err, "Image width is too large: 10 > 5")
|
|
}
|