From 8e9a76ea9f621e3cabfb5d4bd5016a01257a3620 Mon Sep 17 00:00:00 2001 From: Thomas Newton Date: Thu, 21 Dec 2023 18:32:22 +0000 Subject: [PATCH] Update generated code --- .../v1alpha1/mocks/ExecutableNodeStatus.go | 5 +++ .../v1alpha1/mocks/MutableNodeStatus.go | 5 +++ .../nodes/interfaces/mocks/node_executor.go | 39 ++++--------------- 3 files changed, 17 insertions(+), 32 deletions(-) diff --git a/flytepropeller/pkg/apis/flyteworkflow/v1alpha1/mocks/ExecutableNodeStatus.go b/flytepropeller/pkg/apis/flyteworkflow/v1alpha1/mocks/ExecutableNodeStatus.go index e6d2309df16..a436537d6cc 100644 --- a/flytepropeller/pkg/apis/flyteworkflow/v1alpha1/mocks/ExecutableNodeStatus.go +++ b/flytepropeller/pkg/apis/flyteworkflow/v1alpha1/mocks/ExecutableNodeStatus.go @@ -30,6 +30,11 @@ func (_m *ExecutableNodeStatus) ClearDynamicNodeStatus() { _m.Called() } +// ClearExecutionError provides a mock function with given fields: +func (_m *ExecutableNodeStatus) ClearExecutionError() { + _m.Called() +} + // ClearGateNodeStatus provides a mock function with given fields: func (_m *ExecutableNodeStatus) ClearGateNodeStatus() { _m.Called() diff --git a/flytepropeller/pkg/apis/flyteworkflow/v1alpha1/mocks/MutableNodeStatus.go b/flytepropeller/pkg/apis/flyteworkflow/v1alpha1/mocks/MutableNodeStatus.go index d6bdd0510ea..538f59131e7 100644 --- a/flytepropeller/pkg/apis/flyteworkflow/v1alpha1/mocks/MutableNodeStatus.go +++ b/flytepropeller/pkg/apis/flyteworkflow/v1alpha1/mocks/MutableNodeStatus.go @@ -28,6 +28,11 @@ func (_m *MutableNodeStatus) ClearDynamicNodeStatus() { _m.Called() } +// ClearExecutionError provides a mock function with given fields: +func (_m *MutableNodeStatus) ClearExecutionError() { + _m.Called() +} + // ClearGateNodeStatus provides a mock function with given fields: func (_m *MutableNodeStatus) ClearGateNodeStatus() { _m.Called() diff --git a/flytepropeller/pkg/controller/nodes/interfaces/mocks/node_executor.go b/flytepropeller/pkg/controller/nodes/interfaces/mocks/node_executor.go index 1b4f056b517..f3b8c007d19 100644 --- a/flytepropeller/pkg/controller/nodes/interfaces/mocks/node_executor.go +++ b/flytepropeller/pkg/controller/nodes/interfaces/mocks/node_executor.go @@ -9,6 +9,8 @@ import ( interfaces "github.com/flyteorg/flyte/flytepropeller/pkg/controller/nodes/interfaces" mock "github.com/stretchr/testify/mock" + + v1alpha1 "github.com/flyteorg/flyte/flytepropeller/pkg/apis/flyteworkflow/v1alpha1" ) // NodeExecutor is an autogenerated mock type for the NodeExecutor type @@ -48,6 +50,11 @@ func (_m *NodeExecutor) Abort(ctx context.Context, h interfaces.NodeHandler, nCt return r0 } +// Clear provides a mock function with given fields: executableNodeStatus +func (_m *NodeExecutor) Clear(executableNodeStatus v1alpha1.ExecutableNodeStatus) { + _m.Called(executableNodeStatus) +} + type NodeExecutor_Finalize struct { *mock.Call } @@ -80,38 +87,6 @@ func (_m *NodeExecutor) Finalize(ctx context.Context, h interfaces.NodeHandler, return r0 } -type NodeExecutor_GetClearPreviousError struct { - *mock.Call -} - -func (_m NodeExecutor_GetClearPreviousError) Return(_a0 bool) *NodeExecutor_GetClearPreviousError { - return &NodeExecutor_GetClearPreviousError{Call: _m.Call.Return(_a0)} -} - -func (_m *NodeExecutor) OnGetClearPreviousError() *NodeExecutor_GetClearPreviousError { - c_call := _m.On("GetClearPreviousError") - return &NodeExecutor_GetClearPreviousError{Call: c_call} -} - -func (_m *NodeExecutor) OnGetClearPreviousErrorMatch(matchers ...interface{}) *NodeExecutor_GetClearPreviousError { - c_call := _m.On("GetClearPreviousError", matchers...) - return &NodeExecutor_GetClearPreviousError{Call: c_call} -} - -// GetClearPreviousError provides a mock function with given fields: -func (_m *NodeExecutor) GetClearPreviousError() bool { - ret := _m.Called() - - var r0 bool - if rf, ok := ret.Get(0).(func() bool); ok { - r0 = rf() - } else { - r0 = ret.Get(0).(bool) - } - - return r0 -} - type NodeExecutor_HandleNode struct { *mock.Call }