mirror of https://github.com/go-gitea/gitea.git
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.
52 lines
1.3 KiB
52 lines
1.3 KiB
// Copyright 2023 The Gitea Authors. All rights reserved. |
|
// SPDX-License-Identifier: MIT |
|
|
|
package db_test |
|
|
|
import ( |
|
"testing" |
|
|
|
"code.gitea.io/gitea/models/db" |
|
repo_model "code.gitea.io/gitea/models/repo" |
|
"code.gitea.io/gitea/models/unittest" |
|
|
|
"github.com/stretchr/testify/assert" |
|
"xorm.io/builder" |
|
) |
|
|
|
type mockListOptions struct { |
|
db.ListOptions |
|
} |
|
|
|
func (opts mockListOptions) IsListAll() bool { |
|
return true |
|
} |
|
|
|
func (opts mockListOptions) ToConds() builder.Cond { |
|
return builder.NewCond() |
|
} |
|
|
|
func TestFind(t *testing.T) { |
|
assert.NoError(t, unittest.PrepareTestDatabase()) |
|
xe := unittest.GetXORMEngine() |
|
assert.NoError(t, xe.Sync(&repo_model.RepoUnit{})) |
|
|
|
var repoUnitCount int |
|
_, err := db.GetEngine(t.Context()).SQL("SELECT COUNT(*) FROM repo_unit").Get(&repoUnitCount) |
|
assert.NoError(t, err) |
|
assert.NotEmpty(t, repoUnitCount) |
|
|
|
opts := mockListOptions{} |
|
repoUnits, err := db.Find[repo_model.RepoUnit](t.Context(), opts) |
|
assert.NoError(t, err) |
|
assert.Len(t, repoUnits, repoUnitCount) |
|
|
|
cnt, err := db.Count[repo_model.RepoUnit](t.Context(), opts) |
|
assert.NoError(t, err) |
|
assert.EqualValues(t, repoUnitCount, cnt) |
|
|
|
repoUnits, newCnt, err := db.FindAndCount[repo_model.RepoUnit](t.Context(), opts) |
|
assert.NoError(t, err) |
|
assert.Equal(t, cnt, newCnt) |
|
assert.Len(t, repoUnits, repoUnitCount) |
|
}
|
|
|