mirror of
https://github.com/temporalio/temporal.git
synced 2026-08-30 18:41:49 -07:00
## What changed? Plumb executable attempt count to task attributes during task validation/execution. Check task validator in side effect standby task executor. ## Why? Add a field to allow task validators to decide if task is best effort, meaning they attempt execution a number of times before being invalidated. ## How did you test it? - [X] built - [X] run locally and tested manually - [X] covered by existing tests - [ ] added new unit test(s) - [ ] added new functional test(s)
201 lines
7.5 KiB
Go
201 lines
7.5 KiB
Go
// Code generated by MockGen. DO NOT EDIT.
|
|
// Source: task.go
|
|
//
|
|
// Generated by this command:
|
|
//
|
|
// mockgen -package chasm -source task.go -destination task_mock.go
|
|
//
|
|
|
|
// Package chasm is a generated GoMock package.
|
|
package chasm
|
|
|
|
import (
|
|
context "context"
|
|
reflect "reflect"
|
|
|
|
gomock "go.uber.org/mock/gomock"
|
|
)
|
|
|
|
// MockSideEffectTaskHandler is a mock of SideEffectTaskHandler interface.
|
|
type MockSideEffectTaskHandler[C any, T any] struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockSideEffectTaskHandlerMockRecorder[C, T]
|
|
isgomock struct{}
|
|
}
|
|
|
|
// MockSideEffectTaskHandlerMockRecorder is the mock recorder for MockSideEffectTaskHandler.
|
|
type MockSideEffectTaskHandlerMockRecorder[C any, T any] struct {
|
|
mock *MockSideEffectTaskHandler[C, T]
|
|
}
|
|
|
|
// NewMockSideEffectTaskHandler creates a new mock instance.
|
|
func NewMockSideEffectTaskHandler[C any, T any](ctrl *gomock.Controller) *MockSideEffectTaskHandler[C, T] {
|
|
mock := &MockSideEffectTaskHandler[C, T]{ctrl: ctrl}
|
|
mock.recorder = &MockSideEffectTaskHandlerMockRecorder[C, T]{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockSideEffectTaskHandler[C, T]) EXPECT() *MockSideEffectTaskHandlerMockRecorder[C, T] {
|
|
return m.recorder
|
|
}
|
|
|
|
// Discard mocks base method.
|
|
func (m *MockSideEffectTaskHandler[C, T]) Discard(arg0 context.Context, arg1 ComponentRef, arg2 TaskAttributes, arg3 T) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Discard", arg0, arg1, arg2, arg3)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// Discard indicates an expected call of Discard.
|
|
func (mr *MockSideEffectTaskHandlerMockRecorder[C, T]) Discard(arg0, arg1, arg2, arg3 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Discard", reflect.TypeOf((*MockSideEffectTaskHandler[C, T])(nil).Discard), arg0, arg1, arg2, arg3)
|
|
}
|
|
|
|
// Execute mocks base method.
|
|
func (m *MockSideEffectTaskHandler[C, T]) Execute(arg0 context.Context, arg1 ComponentRef, arg2 TaskAttributes, arg3 T) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Execute", arg0, arg1, arg2, arg3)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// Execute indicates an expected call of Execute.
|
|
func (mr *MockSideEffectTaskHandlerMockRecorder[C, T]) Execute(arg0, arg1, arg2, arg3 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Execute", reflect.TypeOf((*MockSideEffectTaskHandler[C, T])(nil).Execute), arg0, arg1, arg2, arg3)
|
|
}
|
|
|
|
// Validate mocks base method.
|
|
func (m *MockSideEffectTaskHandler[C, T]) Validate(arg0 Context, arg1 C, arg2 TaskInvocation, arg3 T) (bool, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Validate", arg0, arg1, arg2, arg3)
|
|
ret0, _ := ret[0].(bool)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// Validate indicates an expected call of Validate.
|
|
func (mr *MockSideEffectTaskHandlerMockRecorder[C, T]) Validate(arg0, arg1, arg2, arg3 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Validate", reflect.TypeOf((*MockSideEffectTaskHandler[C, T])(nil).Validate), arg0, arg1, arg2, arg3)
|
|
}
|
|
|
|
// sideEffectTaskHandler mocks base method.
|
|
func (m *MockSideEffectTaskHandler[C, T]) sideEffectTaskHandler() {
|
|
m.ctrl.T.Helper()
|
|
m.ctrl.Call(m, "sideEffectTaskHandler")
|
|
}
|
|
|
|
// sideEffectTaskHandler indicates an expected call of sideEffectTaskHandler.
|
|
func (mr *MockSideEffectTaskHandlerMockRecorder[C, T]) sideEffectTaskHandler() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "sideEffectTaskHandler", reflect.TypeOf((*MockSideEffectTaskHandler[C, T])(nil).sideEffectTaskHandler))
|
|
}
|
|
|
|
// MockPureTaskHandler is a mock of PureTaskHandler interface.
|
|
type MockPureTaskHandler[C any, T any] struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockPureTaskHandlerMockRecorder[C, T]
|
|
isgomock struct{}
|
|
}
|
|
|
|
// MockPureTaskHandlerMockRecorder is the mock recorder for MockPureTaskHandler.
|
|
type MockPureTaskHandlerMockRecorder[C any, T any] struct {
|
|
mock *MockPureTaskHandler[C, T]
|
|
}
|
|
|
|
// NewMockPureTaskHandler creates a new mock instance.
|
|
func NewMockPureTaskHandler[C any, T any](ctrl *gomock.Controller) *MockPureTaskHandler[C, T] {
|
|
mock := &MockPureTaskHandler[C, T]{ctrl: ctrl}
|
|
mock.recorder = &MockPureTaskHandlerMockRecorder[C, T]{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockPureTaskHandler[C, T]) EXPECT() *MockPureTaskHandlerMockRecorder[C, T] {
|
|
return m.recorder
|
|
}
|
|
|
|
// Execute mocks base method.
|
|
func (m *MockPureTaskHandler[C, T]) Execute(arg0 MutableContext, arg1 C, arg2 TaskAttributes, arg3 T) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Execute", arg0, arg1, arg2, arg3)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// Execute indicates an expected call of Execute.
|
|
func (mr *MockPureTaskHandlerMockRecorder[C, T]) Execute(arg0, arg1, arg2, arg3 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Execute", reflect.TypeOf((*MockPureTaskHandler[C, T])(nil).Execute), arg0, arg1, arg2, arg3)
|
|
}
|
|
|
|
// Validate mocks base method.
|
|
func (m *MockPureTaskHandler[C, T]) Validate(arg0 Context, arg1 C, arg2 TaskInvocation, arg3 T) (bool, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Validate", arg0, arg1, arg2, arg3)
|
|
ret0, _ := ret[0].(bool)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// Validate indicates an expected call of Validate.
|
|
func (mr *MockPureTaskHandlerMockRecorder[C, T]) Validate(arg0, arg1, arg2, arg3 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Validate", reflect.TypeOf((*MockPureTaskHandler[C, T])(nil).Validate), arg0, arg1, arg2, arg3)
|
|
}
|
|
|
|
// pureTaskHandler mocks base method.
|
|
func (m *MockPureTaskHandler[C, T]) pureTaskHandler() {
|
|
m.ctrl.T.Helper()
|
|
m.ctrl.Call(m, "pureTaskHandler")
|
|
}
|
|
|
|
// pureTaskHandler indicates an expected call of pureTaskHandler.
|
|
func (mr *MockPureTaskHandlerMockRecorder[C, T]) pureTaskHandler() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "pureTaskHandler", reflect.TypeOf((*MockPureTaskHandler[C, T])(nil).pureTaskHandler))
|
|
}
|
|
|
|
// MockTaskValidator is a mock of TaskValidator interface.
|
|
type MockTaskValidator[C any, T any] struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockTaskValidatorMockRecorder[C, T]
|
|
isgomock struct{}
|
|
}
|
|
|
|
// MockTaskValidatorMockRecorder is the mock recorder for MockTaskValidator.
|
|
type MockTaskValidatorMockRecorder[C any, T any] struct {
|
|
mock *MockTaskValidator[C, T]
|
|
}
|
|
|
|
// NewMockTaskValidator creates a new mock instance.
|
|
func NewMockTaskValidator[C any, T any](ctrl *gomock.Controller) *MockTaskValidator[C, T] {
|
|
mock := &MockTaskValidator[C, T]{ctrl: ctrl}
|
|
mock.recorder = &MockTaskValidatorMockRecorder[C, T]{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockTaskValidator[C, T]) EXPECT() *MockTaskValidatorMockRecorder[C, T] {
|
|
return m.recorder
|
|
}
|
|
|
|
// Validate mocks base method.
|
|
func (m *MockTaskValidator[C, T]) Validate(arg0 Context, arg1 C, arg2 TaskInvocation, arg3 T) (bool, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Validate", arg0, arg1, arg2, arg3)
|
|
ret0, _ := ret[0].(bool)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// Validate indicates an expected call of Validate.
|
|
func (mr *MockTaskValidatorMockRecorder[C, T]) Validate(arg0, arg1, arg2, arg3 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Validate", reflect.TypeOf((*MockTaskValidator[C, T])(nil).Validate), arg0, arg1, arg2, arg3)
|
|
}
|