parent
							
								
									91fa0eb9d7
								
							
						
					
					
						commit
						a8c4f8cebc
					
				
					 2 changed files with 50 additions and 2 deletions
				
			
		| 
						 | 
				
			
			@ -134,7 +134,7 @@ func Find[T any](ctx context.Context, opts FindOptions, objects *[]T) error {
 | 
			
		|||
	if !opts.IsListAll() {
 | 
			
		||||
		sess.Limit(opts.GetSkipTake())
 | 
			
		||||
	}
 | 
			
		||||
	return sess.Find(&objects)
 | 
			
		||||
	return sess.Find(objects)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Count represents a common count function which accept an options interface
 | 
			
		||||
| 
						 | 
				
			
			@ -148,5 +148,5 @@ func FindAndCount[T any](ctx context.Context, opts FindOptions, objects *[]T) (i
 | 
			
		|||
	if !opts.IsListAll() {
 | 
			
		||||
		sess.Limit(opts.GetSkipTake())
 | 
			
		||||
	}
 | 
			
		||||
	return sess.FindAndCount(&objects)
 | 
			
		||||
	return sess.FindAndCount(objects)
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										48
									
								
								models/db/list_test.go
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										48
									
								
								models/db/list_test.go
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,48 @@
 | 
			
		|||
// 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{}))
 | 
			
		||||
 | 
			
		||||
	opts := mockListOptions{}
 | 
			
		||||
	var repoUnits []repo_model.RepoUnit
 | 
			
		||||
	err := db.Find(db.DefaultContext, &opts, &repoUnits)
 | 
			
		||||
	assert.NoError(t, err)
 | 
			
		||||
	assert.EqualValues(t, 83, len(repoUnits))
 | 
			
		||||
 | 
			
		||||
	cnt, err := db.Count(db.DefaultContext, &opts, new(repo_model.RepoUnit))
 | 
			
		||||
	assert.NoError(t, err)
 | 
			
		||||
	assert.EqualValues(t, 83, cnt)
 | 
			
		||||
 | 
			
		||||
	repoUnits = make([]repo_model.RepoUnit, 0, 10)
 | 
			
		||||
	newCnt, err := db.FindAndCount(db.DefaultContext, &opts, &repoUnits)
 | 
			
		||||
	assert.NoError(t, err)
 | 
			
		||||
	assert.EqualValues(t, cnt, newCnt)
 | 
			
		||||
}
 | 
			
		||||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue