forked from gitea/gitea
Lunny Xiao
df1e7d0067
For those simple objects, it's unnecessary to write the find and count methods again and again.
124 lines
2.8 KiB
Go
124 lines
2.8 KiB
Go
// Copyright 2020 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package project
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"code.gitea.io/gitea/models/db"
|
|
"code.gitea.io/gitea/models/unittest"
|
|
"code.gitea.io/gitea/modules/timeutil"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestIsProjectTypeValid(t *testing.T) {
|
|
const UnknownType Type = 15
|
|
|
|
cases := []struct {
|
|
typ Type
|
|
valid bool
|
|
}{
|
|
{TypeIndividual, true},
|
|
{TypeRepository, true},
|
|
{TypeOrganization, true},
|
|
{UnknownType, false},
|
|
}
|
|
|
|
for _, v := range cases {
|
|
assert.Equal(t, v.valid, IsTypeValid(v.typ))
|
|
}
|
|
}
|
|
|
|
func TestGetProjects(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
projects, err := db.Find[Project](db.DefaultContext, SearchOptions{RepoID: 1})
|
|
assert.NoError(t, err)
|
|
|
|
// 1 value for this repo exists in the fixtures
|
|
assert.Len(t, projects, 1)
|
|
|
|
projects, err = db.Find[Project](db.DefaultContext, SearchOptions{RepoID: 3})
|
|
assert.NoError(t, err)
|
|
|
|
// 1 value for this repo exists in the fixtures
|
|
assert.Len(t, projects, 1)
|
|
}
|
|
|
|
func TestProject(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
project := &Project{
|
|
Type: TypeRepository,
|
|
BoardType: BoardTypeBasicKanban,
|
|
CardType: CardTypeTextOnly,
|
|
Title: "New Project",
|
|
RepoID: 1,
|
|
CreatedUnix: timeutil.TimeStampNow(),
|
|
CreatorID: 2,
|
|
}
|
|
|
|
assert.NoError(t, NewProject(db.DefaultContext, project))
|
|
|
|
_, err := GetProjectByID(db.DefaultContext, project.ID)
|
|
assert.NoError(t, err)
|
|
|
|
// Update project
|
|
project.Title = "Updated title"
|
|
assert.NoError(t, UpdateProject(db.DefaultContext, project))
|
|
|
|
projectFromDB, err := GetProjectByID(db.DefaultContext, project.ID)
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, project.Title, projectFromDB.Title)
|
|
|
|
assert.NoError(t, ChangeProjectStatus(db.DefaultContext, project, true))
|
|
|
|
// Retrieve from DB afresh to check if it is truly closed
|
|
projectFromDB, err = GetProjectByID(db.DefaultContext, project.ID)
|
|
assert.NoError(t, err)
|
|
|
|
assert.True(t, projectFromDB.IsClosed)
|
|
}
|
|
|
|
func TestProjectsSort(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
tests := []struct {
|
|
sortType string
|
|
wants []int64
|
|
}{
|
|
{
|
|
sortType: "default",
|
|
wants: []int64{1, 3, 2, 4},
|
|
},
|
|
{
|
|
sortType: "oldest",
|
|
wants: []int64{4, 2, 3, 1},
|
|
},
|
|
{
|
|
sortType: "recentupdate",
|
|
wants: []int64{1, 3, 2, 4},
|
|
},
|
|
{
|
|
sortType: "leastupdate",
|
|
wants: []int64{4, 2, 3, 1},
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
projects, count, err := db.FindAndCount[Project](db.DefaultContext, SearchOptions{
|
|
OrderBy: GetSearchOrderByBySortType(tt.sortType),
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, int64(4), count)
|
|
if assert.Len(t, projects, 4) {
|
|
for i := range projects {
|
|
assert.EqualValues(t, tt.wants[i], projects[i].ID)
|
|
}
|
|
}
|
|
}
|
|
}
|