plugin.go

// Code generated by mockery v1.0.1. DO NOT EDIT.

package mocks

import (
	context "context"

	client "sigs.k8s.io/controller-runtime/pkg/client"

	core "github.com/flyteorg/flyteplugins/go/tasks/pluginmachinery/core"

	k8s "github.com/flyteorg/flyteplugins/go/tasks/pluginmachinery/k8s"

	mock "github.com/stretchr/testify/mock"
)

// Plugin is an autogenerated mock type for the Plugin type
type Plugin struct {
	mock.Mock
}

type Plugin_BuildIdentityResource struct {
	*mock.Call
}

func (_m Plugin_BuildIdentityResource) Return(_a0 client.Object, _a1 error) *Plugin_BuildIdentityResource {
	return &Plugin_BuildIdentityResource{Call: _m.Call.Return(_a0, _a1)}
}

func (_m *Plugin) OnBuildIdentityResource(ctx context.Context, taskCtx core.TaskExecutionMetadata) *Plugin_BuildIdentityResource {
	c_call := _m.On("BuildIdentityResource", ctx, taskCtx)
	return &Plugin_BuildIdentityResource{Call: c_call}
}

func (_m *Plugin) OnBuildIdentityResourceMatch(matchers ...interface{}) *Plugin_BuildIdentityResource {
	c_call := _m.On("BuildIdentityResource", matchers...)
	return &Plugin_BuildIdentityResource{Call: c_call}
}

// BuildIdentityResource provides a mock function with given fields: ctx, taskCtx
func (_m *Plugin) BuildIdentityResource(ctx context.Context, taskCtx core.TaskExecutionMetadata) (client.Object, error) {
	ret := _m.Called(ctx, taskCtx)

	var r0 client.Object
	if rf, ok := ret.Get(0).(func(context.Context, core.TaskExecutionMetadata) client.Object); ok {
		r0 = rf(ctx, taskCtx)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(client.Object)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, core.TaskExecutionMetadata) error); ok {
		r1 = rf(ctx, taskCtx)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

type Plugin_BuildResource struct {
	*mock.Call
}

func (_m Plugin_BuildResource) Return(_a0 client.Object, _a1 error) *Plugin_BuildResource {
	return &Plugin_BuildResource{Call: _m.Call.Return(_a0, _a1)}
}

func (_m *Plugin) OnBuildResource(ctx context.Context, taskCtx core.TaskExecutionContext) *Plugin_BuildResource {
	c_call := _m.On("BuildResource", ctx, taskCtx)
	return &Plugin_BuildResource{Call: c_call}
}

func (_m *Plugin) OnBuildResourceMatch(matchers ...interface{}) *Plugin_BuildResource {
	c_call := _m.On("BuildResource", matchers...)
	return &Plugin_BuildResource{Call: c_call}
}

// BuildResource provides a mock function with given fields: ctx, taskCtx
func (_m *Plugin) BuildResource(ctx context.Context, taskCtx core.TaskExecutionContext) (client.Object, error) {
	ret := _m.Called(ctx, taskCtx)

	var r0 client.Object
	if rf, ok := ret.Get(0).(func(context.Context, core.TaskExecutionContext) client.Object); ok {
		r0 = rf(ctx, taskCtx)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(client.Object)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, core.TaskExecutionContext) error); ok {
		r1 = rf(ctx, taskCtx)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

type Plugin_GetProperties struct {
	*mock.Call
}

func (_m Plugin_GetProperties) Return(_a0 k8s.PluginProperties) *Plugin_GetProperties {
	return &Plugin_GetProperties{Call: _m.Call.Return(_a0)}
}

func (_m *Plugin) OnGetProperties() *Plugin_GetProperties {
	c_call := _m.On("GetProperties")
	return &Plugin_GetProperties{Call: c_call}
}

func (_m *Plugin) OnGetPropertiesMatch(matchers ...interface{}) *Plugin_GetProperties {
	c_call := _m.On("GetProperties", matchers...)
	return &Plugin_GetProperties{Call: c_call}
}

// GetProperties provides a mock function with given fields:
func (_m *Plugin) GetProperties() k8s.PluginProperties {
	ret := _m.Called()

	var r0 k8s.PluginProperties
	if rf, ok := ret.Get(0).(func() k8s.PluginProperties); ok {
		r0 = rf()
	} else {
		r0 = ret.Get(0).(k8s.PluginProperties)
	}

	return r0
}

type Plugin_GetTaskPhase struct {
	*mock.Call
}

func (_m Plugin_GetTaskPhase) Return(_a0 core.PhaseInfo, _a1 error) *Plugin_GetTaskPhase {
	return &Plugin_GetTaskPhase{Call: _m.Call.Return(_a0, _a1)}
}

func (_m *Plugin) OnGetTaskPhase(ctx context.Context, pluginContext k8s.PluginContext, resource client.Object) *Plugin_GetTaskPhase {
	c_call := _m.On("GetTaskPhase", ctx, pluginContext, resource)
	return &Plugin_GetTaskPhase{Call: c_call}
}

func (_m *Plugin) OnGetTaskPhaseMatch(matchers ...interface{}) *Plugin_GetTaskPhase {
	c_call := _m.On("GetTaskPhase", matchers...)
	return &Plugin_GetTaskPhase{Call: c_call}
}

// GetTaskPhase provides a mock function with given fields: ctx, pluginContext, resource
func (_m *Plugin) GetTaskPhase(ctx context.Context, pluginContext k8s.PluginContext, resource client.Object) (core.PhaseInfo, error) {
	ret := _m.Called(ctx, pluginContext, resource)

	var r0 core.PhaseInfo
	if rf, ok := ret.Get(0).(func(context.Context, k8s.PluginContext, client.Object) core.PhaseInfo); ok {
		r0 = rf(ctx, pluginContext, resource)
	} else {
		r0 = ret.Get(0).(core.PhaseInfo)
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, k8s.PluginContext, client.Object) error); ok {
		r1 = rf(ctx, pluginContext, resource)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}