Files
temporal/chasm/task_mock.go
Alan Wu 777c81f865 Add attempt count to task attributes, Validate task on standby (#10985)
## 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)
2026-07-13 20:09:20 +00:00

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)
}