task_event_recorder.go

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

package mocks

import (
	context "context"

	config "github.com/flyteorg/flyte/flytepropeller/pkg/controller/config"

	event "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/event"

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

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

type TaskEventRecorder_RecordTaskEvent struct {
	*mock.Call
}

func (_m TaskEventRecorder_RecordTaskEvent) Return(_a0 error) *TaskEventRecorder_RecordTaskEvent {
	return &TaskEventRecorder_RecordTaskEvent{Call: _m.Call.Return(_a0)}
}

func (_m *TaskEventRecorder) OnRecordTaskEvent(ctx context.Context, _a1 *event.TaskExecutionEvent, eventConfig *config.EventConfig) *TaskEventRecorder_RecordTaskEvent {
	c_call := _m.On("RecordTaskEvent", ctx, _a1, eventConfig)
	return &TaskEventRecorder_RecordTaskEvent{Call: c_call}
}

func (_m *TaskEventRecorder) OnRecordTaskEventMatch(matchers ...interface{}) *TaskEventRecorder_RecordTaskEvent {
	c_call := _m.On("RecordTaskEvent", matchers...)
	return &TaskEventRecorder_RecordTaskEvent{Call: c_call}
}

// RecordTaskEvent provides a mock function with given fields: ctx, _a1, eventConfig
func (_m *TaskEventRecorder) RecordTaskEvent(ctx context.Context, _a1 *event.TaskExecutionEvent, eventConfig *config.EventConfig) error {
	ret := _m.Called(ctx, _a1, eventConfig)

	var r0 error
	if rf, ok := ret.Get(0).(func(context.Context, *event.TaskExecutionEvent, *config.EventConfig) error); ok {
		r0 = rf(ctx, _a1, eventConfig)
	} else {
		r0 = ret.Error(0)
	}

	return r0
}