gotenberg/pkg/gotenberg/mocks.go

package gotenberg

import (
	"context"
	"log/slog"
	"os"
)

// ModuleMock is a mock for the [Module] interface.
type ModuleMock struct {
	DescriptorMock func() ModuleDescriptor
}

func (mod *ModuleMock) Descriptor() ModuleDescriptor {
	return mod.DescriptorMock()
}

// ProvisionerMock is a mock for the [Provisioner] interface.
type ProvisionerMock struct {
	ProvisionMock func(*Context) error
}

func (mod *ProvisionerMock) Provision(ctx *Context) error {
	return mod.ProvisionMock(ctx)
}

// ValidatorMock is a mock for the [Validator] interface.
type ValidatorMock struct {
	ValidateMock func() error
}

func (mod *ValidatorMock) Validate() error {
	return mod.ValidateMock()
}

type DebuggableMock struct {
	DebugMock func() map[string]any
}

func (mod *DebuggableMock) Debug() map[string]any {
	return mod.DebugMock()
}

// PdfEngineMock is a mock for the [PdfEngine] interface.
//
//nolint:dupl
type PdfEngineMock struct {
	MergeMock          func(ctx context.Context, logger *slog.Logger, inputPaths []string, outputPath string) error
	SplitMock          func(ctx context.Context, logger *slog.Logger, mode SplitMode, inputPath, outputDirPath string) ([]string, error)
	FlattenMock        func(ctx context.Context, logger *slog.Logger, inputPath string) error
	ConvertMock        func(ctx context.Context, logger *slog.Logger, formats PdfFormats, inputPath, outputPath string) error
	ReadMetadataMock   func(ctx context.Context, logger *slog.Logger, inputPath string) (map[string]any, error)
	PageCountMock      func(ctx context.Context, logger *slog.Logger, inputPath string) (int, error)
	WriteMetadataMock  func(ctx context.Context, logger *slog.Logger, metadata map[string]any, inputPath string) error
	ReadBookmarksMock  func(ctx context.Context, logger *slog.Logger, inputPath string) ([]Bookmark, error)
	EncryptMock        func(ctx context.Context, logger *slog.Logger, inputPath, userPassword, ownerPassword string) error
	EmbedFilesMock     func(ctx context.Context, logger *slog.Logger, filePaths []string, inputPath string) error
	WriteBookmarksMock func(ctx context.Context, logger *slog.Logger, inputPath string, bookmarks []Bookmark) error
	WatermarkMock      func(ctx context.Context, logger *slog.Logger, inputPath string, stamp Stamp) error
	StampMock          func(ctx context.Context, logger *slog.Logger, inputPath string, stamp Stamp) error
	RotateMock         func(ctx context.Context, logger *slog.Logger, inputPath string, angle int, pages string) error
}

func (engine *PdfEngineMock) Merge(ctx context.Context, logger *slog.Logger, inputPaths []string, outputPath string) error {
	return engine.MergeMock(ctx, logger, inputPaths, outputPath)
}

func (engine *PdfEngineMock) Split(ctx context.Context, logger *slog.Logger, mode SplitMode, inputPath, outputDirPath string) ([]string, error) {
	return engine.SplitMock(ctx, logger, mode, inputPath, outputDirPath)
}

func (engine *PdfEngineMock) Flatten(ctx context.Context, logger *slog.Logger, inputPath string) error {
	return engine.FlattenMock(ctx, logger, inputPath)
}

func (engine *PdfEngineMock) Convert(ctx context.Context, logger *slog.Logger, formats PdfFormats, inputPath, outputPath string) error {
	return engine.ConvertMock(ctx, logger, formats, inputPath, outputPath)
}

func (engine *PdfEngineMock) ReadMetadata(ctx context.Context, logger *slog.Logger, inputPath string) (map[string]any, error) {
	return engine.ReadMetadataMock(ctx, logger, inputPath)
}

func (engine *PdfEngineMock) PageCount(ctx context.Context, logger *slog.Logger, inputPath string) (int, error) {
	return engine.PageCountMock(ctx, logger, inputPath)
}

func (engine *PdfEngineMock) WriteMetadata(ctx context.Context, logger *slog.Logger, metadata map[string]any, inputPath string) error {
	return engine.WriteMetadataMock(ctx, logger, metadata, inputPath)
}

func (engine *PdfEngineMock) ReadBookmarks(ctx context.Context, logger *slog.Logger, inputPath string) ([]Bookmark, error) {
	return engine.ReadBookmarksMock(ctx, logger, inputPath)
}

func (engine *PdfEngineMock) Encrypt(ctx context.Context, logger *slog.Logger, inputPath, userPassword, ownerPassword string) error {
	return engine.EncryptMock(ctx, logger, inputPath, userPassword, ownerPassword)
}

func (engine *PdfEngineMock) EmbedFiles(ctx context.Context, logger *slog.Logger, filePaths []string, inputPath string) error {
	return engine.EmbedFilesMock(ctx, logger, filePaths, inputPath)
}

func (engine *PdfEngineMock) WriteBookmarks(ctx context.Context, logger *slog.Logger, inputPath string, bookmarks []Bookmark) error {
	return engine.WriteBookmarksMock(ctx, logger, inputPath, bookmarks)
}

func (engine *PdfEngineMock) Watermark(ctx context.Context, logger *slog.Logger, inputPath string, stamp Stamp) error {
	return engine.WatermarkMock(ctx, logger, inputPath, stamp)
}

func (engine *PdfEngineMock) Stamp(ctx context.Context, logger *slog.Logger, inputPath string, stamp Stamp) error {
	return engine.StampMock(ctx, logger, inputPath, stamp)
}

func (engine *PdfEngineMock) Rotate(ctx context.Context, logger *slog.Logger, inputPath string, angle int, pages string) error {
	return engine.RotateMock(ctx, logger, inputPath, angle, pages)
}

// PdfEngineProviderMock is a mock for the [PdfEngineProvider] interface.
type PdfEngineProviderMock struct {
	PdfEngineMock func() (PdfEngine, error)
}

func (provider *PdfEngineProviderMock) PdfEngine() (PdfEngine, error) {
	return provider.PdfEngineMock()
}

// ProcessMock is a mock for the [Process] interface.
type ProcessMock struct {
	StartMock   func(logger *slog.Logger) error
	StopMock    func(logger *slog.Logger) error
	HealthyMock func(logger *slog.Logger) bool
}

func (p *ProcessMock) Start(logger *slog.Logger) error {
	return p.StartMock(logger)
}

func (p *ProcessMock) Stop(logger *slog.Logger) error {
	return p.StopMock(logger)
}

func (p *ProcessMock) Healthy(logger *slog.Logger) bool {
	return p.HealthyMock(logger)
}

// ProcessSupervisorMock is a mock for the [ProcessSupervisor] interface.
type ProcessSupervisorMock struct {
	LaunchMock           func() error
	ShutdownMock         func() error
	HealthyMock          func() bool
	RunMock              func(ctx context.Context, logger *slog.Logger, task func() error) error
	ReqQueueSizeMock     func() int64
	RestartsCountMock    func() int64
	ActiveTasksCountMock func() int64
}

func (s *ProcessSupervisorMock) Launch() error {
	return s.LaunchMock()
}

func (s *ProcessSupervisorMock) Shutdown() error {
	return s.ShutdownMock()
}

func (s *ProcessSupervisorMock) Healthy() bool {
	return s.HealthyMock()
}

func (s *ProcessSupervisorMock) Run(ctx context.Context, logger *slog.Logger, task func() error) error {
	return s.RunMock(ctx, logger, task)
}

func (s *ProcessSupervisorMock) ReqQueueSize() int64 {
	return s.ReqQueueSizeMock()
}

func (s *ProcessSupervisorMock) RestartsCount() int64 {
	return s.RestartsCountMock()
}

func (s *ProcessSupervisorMock) ActiveTasksCount() int64 {
	return s.ActiveTasksCountMock()
}

// MetricsProviderMock is a mock for the [MetricsProvider] interface.
type MetricsProviderMock struct {
	MetricsMock func() ([]Metric, error)
}

func (provider *MetricsProviderMock) Metrics() ([]Metric, error) {
	return provider.MetricsMock()
}

// MkdirAllMock is a mock for the [MkdirAll] interface.
type MkdirAllMock struct {
	MkdirAllMock func(path string, perm os.FileMode) error
}

func (mkdirAll *MkdirAllMock) MkdirAll(path string, perm os.FileMode) error {
	return mkdirAll.MkdirAllMock(path, perm)
}

// PathRenameMock is a mock for the [PathRename] interface.
type PathRenameMock struct {
	RenameMock func(oldpath, newpath string) error
}

func (rename *PathRenameMock) Rename(oldpath, newpath string) error {
	return rename.RenameMock(oldpath, newpath)
}

// Interface guards.
var (
	_ Module            = (*ModuleMock)(nil)
	_ Validator         = (*ValidatorMock)(nil)
	_ PdfEngine         = (*PdfEngineMock)(nil)
	_ PdfEngineProvider = (*PdfEngineProviderMock)(nil)
	_ Process           = (*ProcessMock)(nil)
	_ ProcessSupervisor = (*ProcessSupervisorMock)(nil)
	_ MetricsProvider   = (*MetricsProviderMock)(nil)
	_ MkdirAll          = (*MkdirAllMock)(nil)
	_ PathRename        = (*PathRenameMock)(nil)
)