Files
tasks/chasm/component_mock.go
Yichao YangandGitHub 6e9f5cc68c CHASM: improve support for implementing Terminate method (#9351)
## What changed?
- Define TerminableComponent and RootComponent interface
- Make metrics handler available through chasm context
- Add support for passing key value pairs with chasm context similar to
context.Context
- Add support for specifying key values pairs that will always be
available in the Context when a component is accessed.
- Add requestID to TerminateComponentRequest

## Why?
- Improve support for library authors to implement the Terminate method.

## How did you test it?
- [ ] built
- [ ] run locally and tested manually
- [x] covered by existing tests
- [ ] added new unit test(s)
- [x] added new functional test(s)
2026-02-22 23:30:24 -08:00

197 lines
7.0 KiB
Go

// Code generated by MockGen. DO NOT EDIT.
// Source: component.go
//
// Generated by this command:
//
// mockgen -package chasm -source component.go -destination component_mock.go
//
// Package chasm is a generated GoMock package.
package chasm
import (
reflect "reflect"
gomock "go.uber.org/mock/gomock"
)
// MockComponent is a mock of Component interface.
type MockComponent struct {
ctrl *gomock.Controller
recorder *MockComponentMockRecorder
isgomock struct{}
}
// MockComponentMockRecorder is the mock recorder for MockComponent.
type MockComponentMockRecorder struct {
mock *MockComponent
}
// NewMockComponent creates a new mock instance.
func NewMockComponent(ctrl *gomock.Controller) *MockComponent {
mock := &MockComponent{ctrl: ctrl}
mock.recorder = &MockComponentMockRecorder{mock}
return mock
}
// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockComponent) EXPECT() *MockComponentMockRecorder {
return m.recorder
}
// LifecycleState mocks base method.
func (m *MockComponent) LifecycleState(arg0 Context) LifecycleState {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "LifecycleState", arg0)
ret0, _ := ret[0].(LifecycleState)
return ret0
}
// LifecycleState indicates an expected call of LifecycleState.
func (mr *MockComponentMockRecorder) LifecycleState(arg0 any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LifecycleState", reflect.TypeOf((*MockComponent)(nil).LifecycleState), arg0)
}
// mustEmbedUnimplementedComponent mocks base method.
func (m *MockComponent) mustEmbedUnimplementedComponent() {
m.ctrl.T.Helper()
m.ctrl.Call(m, "mustEmbedUnimplementedComponent")
}
// mustEmbedUnimplementedComponent indicates an expected call of mustEmbedUnimplementedComponent.
func (mr *MockComponentMockRecorder) mustEmbedUnimplementedComponent() *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "mustEmbedUnimplementedComponent", reflect.TypeOf((*MockComponent)(nil).mustEmbedUnimplementedComponent))
}
// MockTerminableComponent is a mock of TerminableComponent interface.
type MockTerminableComponent struct {
ctrl *gomock.Controller
recorder *MockTerminableComponentMockRecorder
isgomock struct{}
}
// MockTerminableComponentMockRecorder is the mock recorder for MockTerminableComponent.
type MockTerminableComponentMockRecorder struct {
mock *MockTerminableComponent
}
// NewMockTerminableComponent creates a new mock instance.
func NewMockTerminableComponent(ctrl *gomock.Controller) *MockTerminableComponent {
mock := &MockTerminableComponent{ctrl: ctrl}
mock.recorder = &MockTerminableComponentMockRecorder{mock}
return mock
}
// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockTerminableComponent) EXPECT() *MockTerminableComponentMockRecorder {
return m.recorder
}
// LifecycleState mocks base method.
func (m *MockTerminableComponent) LifecycleState(arg0 Context) LifecycleState {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "LifecycleState", arg0)
ret0, _ := ret[0].(LifecycleState)
return ret0
}
// LifecycleState indicates an expected call of LifecycleState.
func (mr *MockTerminableComponentMockRecorder) LifecycleState(arg0 any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LifecycleState", reflect.TypeOf((*MockTerminableComponent)(nil).LifecycleState), arg0)
}
// Terminate mocks base method.
func (m *MockTerminableComponent) Terminate(arg0 MutableContext, arg1 TerminateComponentRequest) (TerminateComponentResponse, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "Terminate", arg0, arg1)
ret0, _ := ret[0].(TerminateComponentResponse)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// Terminate indicates an expected call of Terminate.
func (mr *MockTerminableComponentMockRecorder) Terminate(arg0, arg1 any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Terminate", reflect.TypeOf((*MockTerminableComponent)(nil).Terminate), arg0, arg1)
}
// mustEmbedUnimplementedComponent mocks base method.
func (m *MockTerminableComponent) mustEmbedUnimplementedComponent() {
m.ctrl.T.Helper()
m.ctrl.Call(m, "mustEmbedUnimplementedComponent")
}
// mustEmbedUnimplementedComponent indicates an expected call of mustEmbedUnimplementedComponent.
func (mr *MockTerminableComponentMockRecorder) mustEmbedUnimplementedComponent() *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "mustEmbedUnimplementedComponent", reflect.TypeOf((*MockTerminableComponent)(nil).mustEmbedUnimplementedComponent))
}
// MockRootComponent is a mock of RootComponent interface.
type MockRootComponent struct {
ctrl *gomock.Controller
recorder *MockRootComponentMockRecorder
isgomock struct{}
}
// MockRootComponentMockRecorder is the mock recorder for MockRootComponent.
type MockRootComponentMockRecorder struct {
mock *MockRootComponent
}
// NewMockRootComponent creates a new mock instance.
func NewMockRootComponent(ctrl *gomock.Controller) *MockRootComponent {
mock := &MockRootComponent{ctrl: ctrl}
mock.recorder = &MockRootComponentMockRecorder{mock}
return mock
}
// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockRootComponent) EXPECT() *MockRootComponentMockRecorder {
return m.recorder
}
// LifecycleState mocks base method.
func (m *MockRootComponent) LifecycleState(arg0 Context) LifecycleState {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "LifecycleState", arg0)
ret0, _ := ret[0].(LifecycleState)
return ret0
}
// LifecycleState indicates an expected call of LifecycleState.
func (mr *MockRootComponentMockRecorder) LifecycleState(arg0 any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LifecycleState", reflect.TypeOf((*MockRootComponent)(nil).LifecycleState), arg0)
}
// Terminate mocks base method.
func (m *MockRootComponent) Terminate(arg0 MutableContext, arg1 TerminateComponentRequest) (TerminateComponentResponse, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "Terminate", arg0, arg1)
ret0, _ := ret[0].(TerminateComponentResponse)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// Terminate indicates an expected call of Terminate.
func (mr *MockRootComponentMockRecorder) Terminate(arg0, arg1 any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Terminate", reflect.TypeOf((*MockRootComponent)(nil).Terminate), arg0, arg1)
}
// mustEmbedUnimplementedComponent mocks base method.
func (m *MockRootComponent) mustEmbedUnimplementedComponent() {
m.ctrl.T.Helper()
m.ctrl.Call(m, "mustEmbedUnimplementedComponent")
}
// mustEmbedUnimplementedComponent indicates an expected call of mustEmbedUnimplementedComponent.
func (mr *MockRootComponentMockRecorder) mustEmbedUnimplementedComponent() *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "mustEmbedUnimplementedComponent", reflect.TypeOf((*MockRootComponent)(nil).mustEmbedUnimplementedComponent))
}