mocks

Documentation

Index

• func NewMockCompiler() interfaces.Compiler
• type FlyteWorkflowBuilder
  • func (_m *FlyteWorkflowBuilder) Build(wfClosure *core.CompiledWorkflowClosure, inputs *core.LiteralMap, ...) (*v1alpha1.FlyteWorkflow, error)
  • func (_m *FlyteWorkflowBuilder) OnBuild(wfClosure *core.CompiledWorkflowClosure, inputs *core.LiteralMap, ...) *FlyteWorkflowBuilder_Build
  • func (_m *FlyteWorkflowBuilder) OnBuildMatch(matchers ...interface{}) *FlyteWorkflowBuilder_Build
• type FlyteWorkflowBuilder_Build
  • func (_m FlyteWorkflowBuilder_Build) Return(_a0 *v1alpha1.FlyteWorkflow, _a1 error) *FlyteWorkflowBuilder_Build
• type MockCompiler
  • func (c *MockCompiler) AddCompileTaskCallback(callback func(task *core.TaskTemplate) (*core.CompiledTask, error))
  • func (c *MockCompiler) AddCompileWorkflowCallback(...)
  • func (c *MockCompiler) AddGetRequirementCallback(...)
  • func (c *MockCompiler) CompileTask(task *core.TaskTemplate) (*core.CompiledTask, error)
  • func (c *MockCompiler) CompileWorkflow(primaryWf *core.WorkflowTemplate, subworkflows []*core.WorkflowTemplate, ...) (*core.CompiledWorkflowClosure, error)
  • func (c *MockCompiler) GetRequirements(fg *core.WorkflowTemplate, subWfs []*core.WorkflowTemplate) (reqs compiler.WorkflowExecutionRequirements, err error)
• type WorkflowExecutor
  • func (_m *WorkflowExecutor) Abort(ctx context.Context, data interfaces.AbortData) error
  • func (_m *WorkflowExecutor) Execute(ctx context.Context, data interfaces.ExecutionData) (interfaces.ExecutionResponse, error)
  • func (_m *WorkflowExecutor) ID() string
  • func (_m *WorkflowExecutor) OnAbort(ctx context.Context, data interfaces.AbortData) *WorkflowExecutor_Abort
  • func (_m *WorkflowExecutor) OnAbortMatch(matchers ...interface{}) *WorkflowExecutor_Abort
  • func (_m *WorkflowExecutor) OnExecute(ctx context.Context, data interfaces.ExecutionData) *WorkflowExecutor_Execute
  • func (_m *WorkflowExecutor) OnExecuteMatch(matchers ...interface{}) *WorkflowExecutor_Execute
  • func (_m *WorkflowExecutor) OnID() *WorkflowExecutor_ID
  • func (_m *WorkflowExecutor) OnIDMatch(matchers ...interface{}) *WorkflowExecutor_ID
• type WorkflowExecutor_Abort
  • func (_m WorkflowExecutor_Abort) Return(_a0 error) *WorkflowExecutor_Abort
• type WorkflowExecutor_Execute
  • func (_m WorkflowExecutor_Execute) Return(_a0 interfaces.ExecutionResponse, _a1 error) *WorkflowExecutor_Execute
• type WorkflowExecutor_ID
  • func (_m WorkflowExecutor_ID) Return(_a0 string) *WorkflowExecutor_ID

Functions

func NewMockCompiler

func NewMockCompiler() interfaces.Compiler

Types

type FlyteWorkflowBuilder

type FlyteWorkflowBuilder struct {
	mock.Mock
}

FlyteWorkflowBuilder is an autogenerated mock type for the FlyteWorkflowBuilder type

func (*FlyteWorkflowBuilder) Build

func (_m *FlyteWorkflowBuilder) Build(wfClosure *core.CompiledWorkflowClosure, inputs *core.LiteralMap, executionID *core.WorkflowExecutionIdentifier, namespace string) (*v1alpha1.FlyteWorkflow, error)

Build provides a mock function with given fields: wfClosure, inputs, executionID, namespace

func (*FlyteWorkflowBuilder) OnBuild

func (_m *FlyteWorkflowBuilder) OnBuild(wfClosure *core.CompiledWorkflowClosure, inputs *core.LiteralMap, executionID *core.WorkflowExecutionIdentifier, namespace string) *FlyteWorkflowBuilder_Build

func (*FlyteWorkflowBuilder) OnBuildMatch

func (_m *FlyteWorkflowBuilder) OnBuildMatch(matchers ...interface{}) *FlyteWorkflowBuilder_Build

type FlyteWorkflowBuilder_Build

type FlyteWorkflowBuilder_Build struct {
	*mock.Call
}

func (FlyteWorkflowBuilder_Build) Return

func (_m FlyteWorkflowBuilder_Build) Return(_a0 *v1alpha1.FlyteWorkflow, _a1 error) *FlyteWorkflowBuilder_Build

type MockCompiler

type MockCompiler struct {
	// contains filtered or unexported fields
}

func (*MockCompiler) AddCompileTaskCallback

func (c *MockCompiler) AddCompileTaskCallback(
	callback func(task *core.TaskTemplate) (*core.CompiledTask, error))

func (*MockCompiler) AddCompileWorkflowCallback

func (c *MockCompiler) AddCompileWorkflowCallback(callback func(
	primaryWf *core.WorkflowTemplate, subworkflows []*core.WorkflowTemplate, tasks []*core.CompiledTask,
	launchPlans []common.InterfaceProvider) (*core.CompiledWorkflowClosure, error))

func (*MockCompiler) AddGetRequirementCallback

func (c *MockCompiler) AddGetRequirementCallback(
	callback func(fg *core.WorkflowTemplate, subWfs []*core.WorkflowTemplate) (
		reqs compiler.WorkflowExecutionRequirements, err error))

func (*MockCompiler) CompileTask

func (c *MockCompiler) CompileTask(task *core.TaskTemplate) (*core.CompiledTask, error)

func (*MockCompiler) CompileWorkflow

func (c *MockCompiler) CompileWorkflow(
	primaryWf *core.WorkflowTemplate, subworkflows []*core.WorkflowTemplate, tasks []*core.CompiledTask,
	launchPlans []common.InterfaceProvider) (*core.CompiledWorkflowClosure, error)

func (*MockCompiler) GetRequirements

func (c *MockCompiler) GetRequirements(fg *core.WorkflowTemplate, subWfs []*core.WorkflowTemplate) (
	reqs compiler.WorkflowExecutionRequirements, err error)

type WorkflowExecutor

type WorkflowExecutor struct {
	mock.Mock
}

WorkflowExecutor is an autogenerated mock type for the WorkflowExecutor type

func (*WorkflowExecutor) Abort

func (_m *WorkflowExecutor) Abort(ctx context.Context, data interfaces.AbortData) error

Abort provides a mock function with given fields: ctx, data

func (*WorkflowExecutor) Execute

func (_m *WorkflowExecutor) Execute(ctx context.Context, data interfaces.ExecutionData) (interfaces.ExecutionResponse, error)

Execute provides a mock function with given fields: ctx, data

func (*WorkflowExecutor) ID

func (_m *WorkflowExecutor) ID() string

ID provides a mock function with given fields:

func (*WorkflowExecutor) OnAbort

func (_m *WorkflowExecutor) OnAbort(ctx context.Context, data interfaces.AbortData) *WorkflowExecutor_Abort

func (*WorkflowExecutor) OnAbortMatch

func (_m *WorkflowExecutor) OnAbortMatch(matchers ...interface{}) *WorkflowExecutor_Abort

func (*WorkflowExecutor) OnExecute

func (_m *WorkflowExecutor) OnExecute(ctx context.Context, data interfaces.ExecutionData) *WorkflowExecutor_Execute

func (*WorkflowExecutor) OnExecuteMatch

func (_m *WorkflowExecutor) OnExecuteMatch(matchers ...interface{}) *WorkflowExecutor_Execute

func (*WorkflowExecutor) OnID

func (_m *WorkflowExecutor) OnID() *WorkflowExecutor_ID

func (*WorkflowExecutor) OnIDMatch

func (_m *WorkflowExecutor) OnIDMatch(matchers ...interface{}) *WorkflowExecutor_ID

type WorkflowExecutor_Abort

type WorkflowExecutor_Abort struct {
	*mock.Call
}

func (WorkflowExecutor_Abort) Return

func (_m WorkflowExecutor_Abort) Return(_a0 error) *WorkflowExecutor_Abort

type WorkflowExecutor_Execute

type WorkflowExecutor_Execute struct {
	*mock.Call
}

func (WorkflowExecutor_Execute) Return

func (_m WorkflowExecutor_Execute) Return(_a0 interfaces.ExecutionResponse, _a1 error) *WorkflowExecutor_Execute

type WorkflowExecutor_ID

type WorkflowExecutor_ID struct {
	*mock.Call
}

func (WorkflowExecutor_ID) Return

func (_m WorkflowExecutor_ID) Return(_a0 string) *WorkflowExecutor_ID