Платформа ЦРНП "Мирокод" для разработки проектов
https://git.mirocod.ru
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
64 lines
1.7 KiB
64 lines
1.7 KiB
// Copyright 2021 The Gitea 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 repository |
|
|
|
import ( |
|
"bytes" |
|
"crypto/md5" |
|
"fmt" |
|
"image" |
|
"image/png" |
|
"testing" |
|
|
|
repo_model "code.gitea.io/gitea/models/repo" |
|
"code.gitea.io/gitea/models/unittest" |
|
|
|
"github.com/stretchr/testify/assert" |
|
) |
|
|
|
func TestUploadAvatar(t *testing.T) { |
|
// Generate image |
|
myImage := image.NewRGBA(image.Rect(0, 0, 1, 1)) |
|
var buff bytes.Buffer |
|
png.Encode(&buff, myImage) |
|
|
|
assert.NoError(t, unittest.PrepareTestDatabase()) |
|
repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 10}).(*repo_model.Repository) |
|
|
|
err := UploadAvatar(repo, buff.Bytes()) |
|
assert.NoError(t, err) |
|
assert.Equal(t, fmt.Sprintf("%d-%x", 10, md5.Sum(buff.Bytes())), repo.Avatar) |
|
} |
|
|
|
func TestUploadBigAvatar(t *testing.T) { |
|
// Generate BIG image |
|
myImage := image.NewRGBA(image.Rect(0, 0, 5000, 1)) |
|
var buff bytes.Buffer |
|
png.Encode(&buff, myImage) |
|
|
|
assert.NoError(t, unittest.PrepareTestDatabase()) |
|
repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 10}).(*repo_model.Repository) |
|
|
|
err := UploadAvatar(repo, buff.Bytes()) |
|
assert.Error(t, err) |
|
} |
|
|
|
func TestDeleteAvatar(t *testing.T) { |
|
// Generate image |
|
myImage := image.NewRGBA(image.Rect(0, 0, 1, 1)) |
|
var buff bytes.Buffer |
|
png.Encode(&buff, myImage) |
|
|
|
assert.NoError(t, unittest.PrepareTestDatabase()) |
|
repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 10}).(*repo_model.Repository) |
|
|
|
err := UploadAvatar(repo, buff.Bytes()) |
|
assert.NoError(t, err) |
|
|
|
err = DeleteAvatar(repo) |
|
assert.NoError(t, err) |
|
|
|
assert.Equal(t, "", repo.Avatar) |
|
}
|
|
|