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() {
 | 
						if !opts.IsListAll() {
 | 
				
			||||||
		sess.Limit(opts.GetSkipTake())
 | 
							sess.Limit(opts.GetSkipTake())
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	return sess.Find(&objects)
 | 
						return sess.Find(objects)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// Count represents a common count function which accept an options interface
 | 
					// 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() {
 | 
						if !opts.IsListAll() {
 | 
				
			||||||
		sess.Limit(opts.GetSkipTake())
 | 
							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