mirror of
https://codeberg.org/forgejo/forgejo.git
synced 2024-12-24 08:06:49 -05:00
3fd18838aa
* Repository avatars - first variant of code from old work for gogs - add migration 87 - add new option in app.ini - add en-US locale string - add new class in repository.less * Add changed index.css, remove unused template name * Update en-us doc about configuration options * Add comments to new functions, add new option to docker app.ini * Add comment for lint * Remove variable, not needed * Fix formatting * Update swagger api template * Check if avatar exists * Fix avatar link/path checks * Typo * TEXT column can't have a default value * Fixes: - remove old avatar file on upload - use ID in name of avatar file - users may upload same files - add simple tests * Fix fmt check * Generate PNG instead of "static" GIF * More informative comment * Fix error message * Update avatar upload checks: - add file size check - add new option - update config docs - add new string to en-us locale * Fixes: - use FileHEader field for check file size - add new test - upload big image * Fix formatting * Update comments * Update log message * Removed wrong style - not needed * Use Sync2 to migrate * Update repos list view - bigger avatar - fix html blocks alignment * A little adjust avatar size * Use small icons for explore/repo list * Use new cool avatar preparation func by @lafriks * Missing changes for new function * Remove unused import, move imports * Missed new option definition in app.ini Add file size check in user/profile avatar upload * Use smaller field length for Avatar * Use session to update repo DB data, update DeleteAvatar - use session too * Fix err variable definition * As suggested @lafriks - return as soon as possible, code readability
213 lines
5.8 KiB
Go
213 lines
5.8 KiB
Go
// Copyright 2017 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 models
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/md5"
|
|
"fmt"
|
|
"image"
|
|
"image/png"
|
|
"testing"
|
|
|
|
"code.gitea.io/gitea/modules/markup"
|
|
|
|
"github.com/Unknwon/com"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestRepo(t *testing.T) {
|
|
repo := &Repository{Name: "testRepo"}
|
|
repo.Owner = &User{Name: "testOwner"}
|
|
|
|
repo.Units = nil
|
|
|
|
metas := repo.ComposeMetas()
|
|
assert.Equal(t, "testRepo", metas["repo"])
|
|
assert.Equal(t, "testOwner", metas["user"])
|
|
|
|
externalTracker := RepoUnit{
|
|
Type: UnitTypeExternalTracker,
|
|
Config: &ExternalTrackerConfig{
|
|
ExternalTrackerFormat: "https://someurl.com/{user}/{repo}/{issue}",
|
|
},
|
|
}
|
|
|
|
testSuccess := func(expectedStyle string) {
|
|
repo.Units = []*RepoUnit{&externalTracker}
|
|
repo.ExternalMetas = nil
|
|
metas := repo.ComposeMetas()
|
|
assert.Equal(t, expectedStyle, metas["style"])
|
|
assert.Equal(t, "testRepo", metas["repo"])
|
|
assert.Equal(t, "testOwner", metas["user"])
|
|
assert.Equal(t, "https://someurl.com/{user}/{repo}/{issue}", metas["format"])
|
|
}
|
|
|
|
testSuccess(markup.IssueNameStyleNumeric)
|
|
|
|
externalTracker.ExternalTrackerConfig().ExternalTrackerStyle = markup.IssueNameStyleAlphanumeric
|
|
testSuccess(markup.IssueNameStyleAlphanumeric)
|
|
|
|
externalTracker.ExternalTrackerConfig().ExternalTrackerStyle = markup.IssueNameStyleNumeric
|
|
testSuccess(markup.IssueNameStyleNumeric)
|
|
}
|
|
|
|
func TestGetRepositoryCount(t *testing.T) {
|
|
assert.NoError(t, PrepareTestDatabase())
|
|
|
|
count, err1 := GetRepositoryCount(&User{ID: int64(10)})
|
|
privateCount, err2 := GetPrivateRepositoryCount(&User{ID: int64(10)})
|
|
publicCount, err3 := GetPublicRepositoryCount(&User{ID: int64(10)})
|
|
assert.NoError(t, err1)
|
|
assert.NoError(t, err2)
|
|
assert.NoError(t, err3)
|
|
assert.Equal(t, int64(3), count)
|
|
assert.Equal(t, (privateCount + publicCount), count)
|
|
}
|
|
|
|
func TestGetPublicRepositoryCount(t *testing.T) {
|
|
assert.NoError(t, PrepareTestDatabase())
|
|
|
|
count, err := GetPublicRepositoryCount(&User{ID: int64(10)})
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, int64(1), count)
|
|
}
|
|
|
|
func TestGetPrivateRepositoryCount(t *testing.T) {
|
|
assert.NoError(t, PrepareTestDatabase())
|
|
|
|
count, err := GetPrivateRepositoryCount(&User{ID: int64(10)})
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, int64(2), count)
|
|
}
|
|
|
|
func TestUpdateRepositoryVisibilityChanged(t *testing.T) {
|
|
assert.NoError(t, PrepareTestDatabase())
|
|
|
|
// Get sample repo and change visibility
|
|
repo, err := GetRepositoryByID(9)
|
|
repo.IsPrivate = true
|
|
|
|
// Update it
|
|
err = UpdateRepository(repo, true)
|
|
assert.NoError(t, err)
|
|
|
|
// Check visibility of action has become private
|
|
act := Action{}
|
|
_, err = x.ID(3).Get(&act)
|
|
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, true, act.IsPrivate)
|
|
}
|
|
|
|
func TestGetUserFork(t *testing.T) {
|
|
assert.NoError(t, PrepareTestDatabase())
|
|
|
|
// User13 has repo 11 forked from repo10
|
|
repo, err := GetRepositoryByID(10)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, repo)
|
|
repo, err = repo.GetUserFork(13)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, repo)
|
|
|
|
repo, err = GetRepositoryByID(9)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, repo)
|
|
repo, err = repo.GetUserFork(13)
|
|
assert.NoError(t, err)
|
|
assert.Nil(t, repo)
|
|
}
|
|
|
|
func TestForkRepository(t *testing.T) {
|
|
assert.NoError(t, PrepareTestDatabase())
|
|
|
|
// user 13 has already forked repo10
|
|
user := AssertExistsAndLoadBean(t, &User{ID: 13}).(*User)
|
|
repo := AssertExistsAndLoadBean(t, &Repository{ID: 10}).(*Repository)
|
|
|
|
fork, err := ForkRepository(user, user, repo, "test", "test")
|
|
assert.Nil(t, fork)
|
|
assert.Error(t, err)
|
|
assert.True(t, IsErrRepoAlreadyExist(err))
|
|
}
|
|
|
|
func TestRepoAPIURL(t *testing.T) {
|
|
assert.NoError(t, PrepareTestDatabase())
|
|
repo := AssertExistsAndLoadBean(t, &Repository{ID: 10}).(*Repository)
|
|
|
|
assert.Equal(t, "https://try.gitea.io/api/v1/repos/user12/repo10", repo.APIURL())
|
|
}
|
|
|
|
func TestTransferOwnership(t *testing.T) {
|
|
assert.NoError(t, PrepareTestDatabase())
|
|
|
|
doer := AssertExistsAndLoadBean(t, &User{ID: 2}).(*User)
|
|
repo := AssertExistsAndLoadBean(t, &Repository{ID: 3}).(*Repository)
|
|
repo.Owner = AssertExistsAndLoadBean(t, &User{ID: repo.OwnerID}).(*User)
|
|
assert.NoError(t, TransferOwnership(doer, "user2", repo))
|
|
|
|
transferredRepo := AssertExistsAndLoadBean(t, &Repository{ID: 3}).(*Repository)
|
|
assert.EqualValues(t, 2, transferredRepo.OwnerID)
|
|
|
|
assert.False(t, com.IsExist(RepoPath("user3", "repo3")))
|
|
assert.True(t, com.IsExist(RepoPath("user2", "repo3")))
|
|
AssertExistsAndLoadBean(t, &Action{
|
|
OpType: ActionTransferRepo,
|
|
ActUserID: 2,
|
|
RepoID: 3,
|
|
Content: "user3/repo3",
|
|
})
|
|
|
|
CheckConsistencyFor(t, &Repository{}, &User{}, &Team{})
|
|
}
|
|
|
|
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, PrepareTestDatabase())
|
|
repo := AssertExistsAndLoadBean(t, &Repository{ID: 10}).(*Repository)
|
|
|
|
err := repo.UploadAvatar(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, PrepareTestDatabase())
|
|
repo := AssertExistsAndLoadBean(t, &Repository{ID: 10}).(*Repository)
|
|
|
|
err := repo.UploadAvatar(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, PrepareTestDatabase())
|
|
repo := AssertExistsAndLoadBean(t, &Repository{ID: 10}).(*Repository)
|
|
|
|
err := repo.UploadAvatar(buff.Bytes())
|
|
assert.NoError(t, err)
|
|
|
|
err = repo.DeleteAvatar()
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, "", repo.Avatar)
|
|
}
|