stash/pkg/models/mocks/query.go
DingDongSoLong4 c364346a59
Model refactor (#3915)
* Add mockery config file
* Move basic file/folder structs to models
* Fix hack due to import loop
* Move file interfaces to models
* Move folder interfaces to models
* Move scene interfaces to models
* Move scene marker interfaces to models
* Move image interfaces to models
* Move gallery interfaces to models
* Move gallery chapter interfaces to models
* Move studio interfaces to models
* Move movie interfaces to models
* Move performer interfaces to models
* Move tag interfaces to models
* Move autotag interfaces to models
* Regenerate mocks
2023-09-01 10:39:29 +10:00

49 lines
1 KiB
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) Find(ctx context.Context, id int) (*models.Image, error) {
panic("not implemented")
}
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
}