123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255 |
- // 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/models/db"
- repo_model "code.gitea.io/gitea/models/repo"
- "code.gitea.io/gitea/models/unit"
- "code.gitea.io/gitea/models/unittest"
- user_model "code.gitea.io/gitea/models/user"
- "code.gitea.io/gitea/modules/markup"
-
- "github.com/stretchr/testify/assert"
- )
-
- func TestMetas(t *testing.T) {
- assert.NoError(t, unittest.PrepareTestDatabase())
-
- repo := &Repository{Name: "testRepo"}
- repo.Owner = &user_model.User{Name: "testOwner"}
- repo.OwnerName = repo.Owner.Name
-
- repo.Units = nil
-
- metas := repo.ComposeMetas()
- assert.Equal(t, "testRepo", metas["repo"])
- assert.Equal(t, "testOwner", metas["user"])
-
- externalTracker := RepoUnit{
- Type: unit.TypeExternalTracker,
- Config: &ExternalTrackerConfig{
- ExternalTrackerFormat: "https://someurl.com/{user}/{repo}/{issue}",
- },
- }
-
- testSuccess := func(expectedStyle string) {
- repo.Units = []*RepoUnit{&externalTracker}
- repo.RenderingMetas = 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)
-
- repo, err := GetRepositoryByID(3)
- assert.NoError(t, err)
-
- metas = repo.ComposeMetas()
- assert.Contains(t, metas, "org")
- assert.Contains(t, metas, "teams")
- assert.Equal(t, "user3", metas["org"])
- assert.Equal(t, ",owners,team1,", metas["teams"])
- }
-
- func TestGetRepositoryCount(t *testing.T) {
- assert.NoError(t, unittest.PrepareTestDatabase())
-
- count, err1 := GetRepositoryCount(db.DefaultContext, 10)
- privateCount, err2 := GetPrivateRepositoryCount(&user_model.User{ID: int64(10)})
- publicCount, err3 := GetPublicRepositoryCount(&user_model.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, unittest.PrepareTestDatabase())
-
- count, err := GetPublicRepositoryCount(&user_model.User{ID: int64(10)})
- assert.NoError(t, err)
- assert.Equal(t, int64(1), count)
- }
-
- func TestGetPrivateRepositoryCount(t *testing.T) {
- assert.NoError(t, unittest.PrepareTestDatabase())
-
- count, err := GetPrivateRepositoryCount(&user_model.User{ID: int64(10)})
- assert.NoError(t, err)
- assert.Equal(t, int64(2), count)
- }
-
- func TestUpdateRepositoryVisibilityChanged(t *testing.T) {
- assert.NoError(t, unittest.PrepareTestDatabase())
-
- // Get sample repo and change visibility
- repo, err := GetRepositoryByID(9)
- assert.NoError(t, err)
- repo.IsPrivate = true
-
- // Update it
- err = UpdateRepository(repo, true)
- assert.NoError(t, err)
-
- // Check visibility of action has become private
- act := Action{}
- _, err = db.GetEngine(db.DefaultContext).ID(3).Get(&act)
-
- assert.NoError(t, err)
- assert.True(t, act.IsPrivate)
- }
-
- func TestGetUserFork(t *testing.T) {
- assert.NoError(t, unittest.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 TestRepoAPIURL(t *testing.T) {
- assert.NoError(t, unittest.PrepareTestDatabase())
- repo := unittest.AssertExistsAndLoadBean(t, &Repository{ID: 10}).(*Repository)
-
- assert.Equal(t, "https://try.gitea.io/api/v1/repos/user12/repo10", repo.APIURL())
- }
-
- 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, &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, unittest.PrepareTestDatabase())
- repo := unittest.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, unittest.PrepareTestDatabase())
- repo := unittest.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)
- }
-
- func TestDoctorUserStarNum(t *testing.T) {
- assert.NoError(t, unittest.PrepareTestDatabase())
-
- assert.NoError(t, DoctorUserStarNum())
- }
-
- func TestRepoGetReviewers(t *testing.T) {
- assert.NoError(t, unittest.PrepareTestDatabase())
-
- // test public repo
- repo1 := unittest.AssertExistsAndLoadBean(t, &Repository{ID: 1}).(*Repository)
-
- reviewers, err := repo1.GetReviewers(2, 2)
- assert.NoError(t, err)
- assert.Len(t, reviewers, 4)
-
- // test private repo
- repo2 := unittest.AssertExistsAndLoadBean(t, &Repository{ID: 2}).(*Repository)
- reviewers, err = repo2.GetReviewers(2, 2)
- assert.NoError(t, err)
- assert.Empty(t, reviewers)
- }
-
- func TestRepoGetReviewerTeams(t *testing.T) {
- assert.NoError(t, unittest.PrepareTestDatabase())
-
- repo2 := unittest.AssertExistsAndLoadBean(t, &Repository{ID: 2}).(*Repository)
- teams, err := repo2.GetReviewerTeams()
- assert.NoError(t, err)
- assert.Empty(t, teams)
-
- repo3 := unittest.AssertExistsAndLoadBean(t, &Repository{ID: 3}).(*Repository)
- teams, err = repo3.GetReviewerTeams()
- assert.NoError(t, err)
- assert.Len(t, teams, 2)
- }
-
- func TestLinkedRepository(t *testing.T) {
- assert.NoError(t, unittest.PrepareTestDatabase())
- testCases := []struct {
- name string
- attachID int64
- expectedRepo *Repository
- expectedUnitType unit.Type
- }{
- {"LinkedIssue", 1, &Repository{ID: 1}, unit.TypeIssues},
- {"LinkedComment", 3, &Repository{ID: 1}, unit.TypePullRequests},
- {"LinkedRelease", 9, &Repository{ID: 1}, unit.TypeReleases},
- {"Notlinked", 10, nil, -1},
- }
- for _, tc := range testCases {
- t.Run(tc.name, func(t *testing.T) {
- attach, err := repo_model.GetAttachmentByID(tc.attachID)
- assert.NoError(t, err)
- repo, unitType, err := LinkedRepository(attach)
- assert.NoError(t, err)
- if tc.expectedRepo != nil {
- assert.Equal(t, tc.expectedRepo.ID, repo.ID)
- }
- assert.Equal(t, tc.expectedUnitType, unitType)
- })
- }
- }
|