stash/pkg/models/mocks/query.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 (s *sceneResolver) FindByIDs(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 audioResolver struct {
	audios []*models.Audio
}

func (s *audioResolver) Find(ctx context.Context, id int) (*models.Audio, error) {
	panic("not implemented")
}

func (s *audioResolver) FindMany(ctx context.Context, ids []int) ([]*models.Audio, error) {
	return s.audios, nil
}

func (s *audioResolver) FindByIDs(ctx context.Context, ids []int) ([]*models.Audio, error) {
	return s.audios, nil
}

func AudioQueryResult(audios []*models.Audio, count int) *models.AudioQueryResult {
	ret := models.NewAudioQueryResult(&audioResolver{
		audios: audios,
	})

	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
}