fleet/server/mock/datastore_installers.go
Roberto Dip 69f8f2a73b
add API endpoints to retrieve pre-built installers (#6672)
Rel: #6365, this adds a new endpoint to check and download pre-built installers.
2022-07-18 13:44:30 -03:00

34 lines
825 B
Go

// Automatically generated by mockimpl. DO NOT EDIT!
package mock
import (
"context"
"io"
"github.com/fleetdm/fleet/v4/server/fleet"
)
var _ fleet.InstallerStore = (*InstallerStore)(nil)
type GetFunc func(ctx context.Context, installer fleet.Installer) (io.ReadCloser, int64, error)
type PutFunc func(ctx context.Context, installer fleet.Installer) (string, error)
type InstallerStore struct {
GetFunc GetFunc
GetFuncInvoked bool
PutFunc PutFunc
PutFuncInvoked bool
}
func (s *InstallerStore) Get(ctx context.Context, installer fleet.Installer) (io.ReadCloser, int64, error) {
s.GetFuncInvoked = true
return s.GetFunc(ctx, installer)
}
func (s *InstallerStore) Put(ctx context.Context, installer fleet.Installer) (string, error) {
s.PutFuncInvoked = true
return s.PutFunc(ctx, installer)
}