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.
44 lines
1.1 KiB
44 lines
1.1 KiB
// Copyright 2022 The Gitea Authors. All rights reserved. |
|
// SPDX-License-Identifier: MIT |
|
|
|
package db_test |
|
|
|
import ( |
|
"context" |
|
"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" |
|
) |
|
|
|
func TestIterate(t *testing.T) { |
|
assert.NoError(t, unittest.PrepareTestDatabase()) |
|
xe := unittest.GetXORMEngine() |
|
assert.NoError(t, xe.Sync(&repo_model.RepoUnit{})) |
|
|
|
cnt, err := db.GetEngine(t.Context()).Count(&repo_model.RepoUnit{}) |
|
assert.NoError(t, err) |
|
|
|
var repoUnitCnt int |
|
err = db.Iterate(t.Context(), nil, func(ctx context.Context, repo *repo_model.RepoUnit) error { |
|
repoUnitCnt++ |
|
return nil |
|
}) |
|
assert.NoError(t, err) |
|
assert.EqualValues(t, cnt, repoUnitCnt) |
|
|
|
err = db.Iterate(t.Context(), nil, func(ctx context.Context, repoUnit *repo_model.RepoUnit) error { |
|
has, err := db.ExistByID[repo_model.RepoUnit](ctx, repoUnit.ID) |
|
if err != nil { |
|
return err |
|
} |
|
if !has { |
|
return db.ErrNotExist{Resource: "repo_unit", ID: repoUnit.ID} |
|
} |
|
return nil |
|
}) |
|
assert.NoError(t, err) |
|
}
|
|
|