Files
tasks/service/history/ndc/state_rebuilder_mock.go
zeekay 5a601d081b rebrand: go.temporal.io/server -> github.com/hanzoai/tasks
Full module path rebrand from Temporal to Hanzo Tasks:
- go.mod module: github.com/hanzoai/tasks
- All internal Go imports updated (1906 files)
- cmd/server renamed to cmd/tasksd
- Binary name: tasksd (was temporal-server)
- Docker images: hanzoai/tasks (was temporalio/server)
- README rewritten with Hanzo Tasks branding
- LLM.md created with integration notes
- Makefile, goreleaser, CI workflows updated
- go.temporal.io/sdk and go.temporal.io/api imports preserved (separate repos)

Fork of Temporal.io (MIT license).
2026-03-19 10:36:40 -07:00

78 lines
4.0 KiB
Go

// Code generated by MockGen. DO NOT EDIT.
// Source: state_rebuilder.go
//
// Generated by this command:
//
// mockgen -package ndc -source state_rebuilder.go -destination state_rebuilder_mock.go
//
// Package ndc is a generated GoMock package.
package ndc
import (
context "context"
reflect "reflect"
time "time"
persistence "github.com/hanzoai/tasks/api/persistence/v1"
definition "github.com/hanzoai/tasks/common/definition"
interfaces "github.com/hanzoai/tasks/service/history/interfaces"
gomock "go.uber.org/mock/gomock"
)
// MockStateRebuilder is a mock of StateRebuilder interface.
type MockStateRebuilder struct {
ctrl *gomock.Controller
recorder *MockStateRebuilderMockRecorder
isgomock struct{}
}
// MockStateRebuilderMockRecorder is the mock recorder for MockStateRebuilder.
type MockStateRebuilderMockRecorder struct {
mock *MockStateRebuilder
}
// NewMockStateRebuilder creates a new mock instance.
func NewMockStateRebuilder(ctrl *gomock.Controller) *MockStateRebuilder {
mock := &MockStateRebuilder{ctrl: ctrl}
mock.recorder = &MockStateRebuilderMockRecorder{mock}
return mock
}
// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockStateRebuilder) EXPECT() *MockStateRebuilderMockRecorder {
return m.recorder
}
// Rebuild mocks base method.
func (m *MockStateRebuilder) Rebuild(ctx context.Context, now time.Time, baseWorkflowIdentifier definition.WorkflowKey, baseBranchToken []byte, baseLastEventID int64, baseLastEventVersion *int64, targetWorkflowIdentifier definition.WorkflowKey, targetBranchToken []byte, requestID string) (interfaces.MutableState, RebuildStats, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "Rebuild", ctx, now, baseWorkflowIdentifier, baseBranchToken, baseLastEventID, baseLastEventVersion, targetWorkflowIdentifier, targetBranchToken, requestID)
ret0, _ := ret[0].(interfaces.MutableState)
ret1, _ := ret[1].(RebuildStats)
ret2, _ := ret[2].(error)
return ret0, ret1, ret2
}
// Rebuild indicates an expected call of Rebuild.
func (mr *MockStateRebuilderMockRecorder) Rebuild(ctx, now, baseWorkflowIdentifier, baseBranchToken, baseLastEventID, baseLastEventVersion, targetWorkflowIdentifier, targetBranchToken, requestID any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Rebuild", reflect.TypeOf((*MockStateRebuilder)(nil).Rebuild), ctx, now, baseWorkflowIdentifier, baseBranchToken, baseLastEventID, baseLastEventVersion, targetWorkflowIdentifier, targetBranchToken, requestID)
}
// RebuildWithCurrentMutableState mocks base method.
func (m *MockStateRebuilder) RebuildWithCurrentMutableState(ctx context.Context, now time.Time, baseWorkflowIdentifier definition.WorkflowKey, baseBranchToken []byte, baseLastEventID int64, baseLastEventVersion *int64, targetWorkflowIdentifier definition.WorkflowKey, targetBranchToken []byte, currentMutableState *persistence.WorkflowMutableState) (interfaces.MutableState, RebuildStats, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "RebuildWithCurrentMutableState", ctx, now, baseWorkflowIdentifier, baseBranchToken, baseLastEventID, baseLastEventVersion, targetWorkflowIdentifier, targetBranchToken, currentMutableState)
ret0, _ := ret[0].(interfaces.MutableState)
ret1, _ := ret[1].(RebuildStats)
ret2, _ := ret[2].(error)
return ret0, ret1, ret2
}
// RebuildWithCurrentMutableState indicates an expected call of RebuildWithCurrentMutableState.
func (mr *MockStateRebuilderMockRecorder) RebuildWithCurrentMutableState(ctx, now, baseWorkflowIdentifier, baseBranchToken, baseLastEventID, baseLastEventVersion, targetWorkflowIdentifier, targetBranchToken, currentMutableState any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RebuildWithCurrentMutableState", reflect.TypeOf((*MockStateRebuilder)(nil).RebuildWithCurrentMutableState), ctx, now, baseWorkflowIdentifier, baseBranchToken, baseLastEventID, baseLastEventVersion, targetWorkflowIdentifier, targetBranchToken, currentMutableState)
}