mock

type JRClient struct {
	NewJobChainFunc    func(proto.JobChain) error
	StartRequestFunc   func(string) error
	StopRequestFunc    func(string) error
	RequestStatusFunc  func(string) (proto.JobChainStatus, error)
	SysStatRunningFunc func() ([]proto.JobStatus, error)
}

type JRStatus struct {
	RunningFunc func() ([]proto.JobStatus, error)
}

type Job struct {
	CreateErr       error
	SerializeBytes  []byte
	SerializeErr    error
	DeserializeErr  error
	RunReturn       job.Return
	RunErr          error
	RunFunc         func(jobData map[string]interface{}) (job.Return, error) // can use this instead of RunErr and RunFunc for more involved mocks
	StopErr         error
	StatusResp      string
	NameResp        string
	TypeResp        string
	CreatedWithArgs map[string]interface{}
	SetJobArgs      map[string]interface{}
}

type JobFactory struct {
	MockJobs map[string]*Job // keyed on job type
	MakeErr  error

	Created map[string]*Job // keyed on name
	// contains filtered or unexported fields
}

type RMClient struct {
	CreateRequestFunc  func(string, map[string]interface{}) (string, error)
	GetRequestFunc     func(string) (proto.Request, error)
	StartRequestFunc   func(string) error
	FinishRequestFunc  func(string, byte) error
	StopRequestFunc    func(string) error
	RequestStatusFunc  func(string) (proto.RequestStatus, error)
	GetJobChainFunc    func(string) (proto.JobChain, error)
	GetJLFunc          func(string) ([]proto.JobLog, error)
	CreateJLFunc       func(string, proto.JobLog) error
	SysStatRunningFunc func() (proto.RunningStatus, error)
}

type RMStatus struct {
	RunningFunc func(status.Filter) (proto.RunningStatus, error)
}

type RequestDBAccessor struct {
	SaveRequestFunc                  func(proto.Request, proto.CreateRequestParams) error
	GetRequestFunc                   func(string) (proto.Request, error)
	UpdateRequestFunc                func(proto.Request) error
	IncrementRequestFinishedJobsFunc func(string) error
	GetRequestJobStatusesFunc        func(string) (proto.JobStatuses, error)
	GetJobChainFunc                  func(string) (proto.JobChain, error)
	GetLatestJLFunc                  func(string, string) (proto.JobLog, error)
	GetFullJLFunc                    func(string) ([]proto.JobLog, error)
	SaveJLFunc                       func(proto.JobLog) error
}

type RequestManager struct {
	CreateRequestFunc func(proto.CreateRequestParams) (proto.Request, error)
	GetRequestFunc    func(string) (proto.Request, error)
	StartRequestFunc  func(string) error
	FinishRequestFunc func(string, proto.FinishRequestParams) error
	StopRequestFunc   func(string) error
	RequestStatusFunc func(string) (proto.RequestStatus, error)
	GetJobChainFunc   func(string) (proto.JobChain, error)
	GetJLFunc         func(string, string) (proto.JobLog, error)
	GetFullJLFunc     func(string) ([]proto.JobLog, error)
	CreateJLFunc      func(string, proto.JobLog) (proto.JobLog, error)
	RequestListFunc   func() []proto.RequestSpec
}

type Runner struct {
	RunReturn    byte
	RunErr       error
	RunFunc      func(jobData map[string]interface{}) byte // can use this instead of RunErr and RunFunc for more involved mocks
	AddedJobData map[string]interface{}                    // Data to add to jobData.
	RunWg        *sync.WaitGroup                           // WaitGroup that gets released from when a runner starts running.
	RunBlock     chan struct{}                             // Channel that runner.Run() will block on, if defined.
	StopErr      error
	StatusResp   string
	// contains filtered or unexported fields
}

type RunnerFactory struct {
	RunnersToReturn map[string]*Runner // Keyed on job name.
	MakeErr         error
}

type Traverser struct {
	RunErr     error
	StopErr    error
	StatusResp proto.JobChainStatus
	StatusErr  error
}

type TraverserFactory struct {
	MakeFunc func(proto.JobChain) (chain.Traverser, error)
}