stash/pkg/models/mocks/query.go
WithoutPants 964b559309 Restructure data layer (#2532)
* Add new txn manager interface
* Add txn management to sqlite
* Rename get to getByID
* Add contexts to repository methods
* Update query builders
* Add context to reader writer interfaces
* Use repository in resolver
* Tighten interfaces
* Tighten interfaces in dlna
* Tighten interfaces in match package
* Tighten interfaces in scraper package
* Tighten interfaces in scan code
* Tighten interfaces on autotag package
* Remove ReaderWriter usage
* Merge database package into sqlite
2022-09-06 07:03:40 +00:00

45 lines
926 B
Go

package mocks
import (
context "context"
"github.com/stashapp/stash/pkg/models"
)
type sceneResolver struct {
scenes []*models.Scene
}
func (s *sceneResolver) Find(ctx context.Context, id int) (*models.Scene, error) {
panic("not implemented")
}
func (s *sceneResolver) FindMany(ctx context.Context, ids []int) ([]*models.Scene, error) {
return s.scenes, nil
}
func SceneQueryResult(scenes []*models.Scene, count int) *models.SceneQueryResult {
ret := models.NewSceneQueryResult(&sceneResolver{
scenes: scenes,
})
ret.Count = count
return ret
}
type imageResolver struct {
images []*models.Image
}
func (s *imageResolver) FindMany(ctx context.Context, ids []int) ([]*models.Image, error) {
return s.images, nil
}
func ImageQueryResult(images []*models.Image, count int) *models.ImageQueryResult {
ret := models.NewImageQueryResult(&imageResolver{
images: images,
})
ret.Count = count
return ret
}