func (_m *TaskExecutionContext) InputReader() io.InputReader {
	ret := _m.Called()
	var r0 io.InputReader
	if rf, ok := ret.Get(0).(func() io.InputReader); ok {
		r0 = rf()
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(io.InputReader)
		}
	}
	return r0
}
InputReader provides a mock function with given fields:
package mocks
import (
	core "github.com/flyteorg/flyteplugins/go/tasks/pluginmachinery/core"
	io "github.com/flyteorg/flyteplugins/go/tasks/pluginmachinery/io"
	mock "github.com/stretchr/testify/mock"
	storage "github.com/flyteorg/flytestdlib/storage"
)
type TaskExecutionContext struct {
	mock.Mock
}
type TaskExecutionContext_DataStore struct {
	*mock.Call
}
func (_m TaskExecutionContext_DataStore) Return(_a0 *storage.DataStore) *TaskExecutionContext_DataStore {
	return &TaskExecutionContext_DataStore{Call: _m.Call.Return(_a0)}
}
func (_m *TaskExecutionContext) OnDataStore() *TaskExecutionContext_DataStore {
	c := _m.On("DataStore")
	return &TaskExecutionContext_DataStore{Call: c}
}
func (_m *TaskExecutionContext) OnDataStoreMatch(matchers ...interface{}) *TaskExecutionContext_DataStore {
	c := _m.On("DataStore", matchers...)
	return &TaskExecutionContext_DataStore{Call: c}
}
func (_m *TaskExecutionContext) DataStore() *storage.DataStore {
	ret := _m.Called()
	var r0 *storage.DataStore
	if rf, ok := ret.Get(0).(func() *storage.DataStore); ok {
		r0 = rf()
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*storage.DataStore)
		}
	}
	return r0
}
type TaskExecutionContext_InputReader struct {
	*mock.Call
}
func (_m TaskExecutionContext_InputReader) Return(_a0 io.InputReader) *TaskExecutionContext_InputReader {
	return &TaskExecutionContext_InputReader{Call: _m.Call.Return(_a0)}
}
func (_m *TaskExecutionContext) OnInputReader() *TaskExecutionContext_InputReader {
	c := _m.On("InputReader")
	return &TaskExecutionContext_InputReader{Call: c}
}
func (_m *TaskExecutionContext) OnInputReaderMatch(matchers ...interface{}) *TaskExecutionContext_InputReader {
	c := _m.On("InputReader", matchers...)
	return &TaskExecutionContext_InputReader{Call: c}
}