// Code generated by MockGen. DO NOT EDIT.
// Source: workflow.go
//
// Generated by this command:
//
//	mockgen -source=workflow.go -package=mock_workflow -destination=./mock_workflow/mock_workflow.go
//

// Package mock_workflow is a generated GoMock package.
package mock_workflow

import (
	context "context"
	reflect "reflect"

	prometheus "github.com/prometheus/client_golang/prometheus"
	entity "github.com/tidbcloud/fastflow/pkg/entity"
	run "github.com/tidbcloud/fastflow/pkg/entity/run"
	workflow "github.com/tidbcloud/tidb-management-service/utilities/workflow"
	gomock "go.uber.org/mock/gomock"
)

// MockWorkflow is a mock of Workflow interface.
type MockWorkflow struct {
	ctrl     *gomock.Controller
	recorder *MockWorkflowMockRecorder
	isgomock struct{}
}

// MockWorkflowMockRecorder is the mock recorder for MockWorkflow.
type MockWorkflowMockRecorder struct {
	mock *MockWorkflow
}

// NewMockWorkflow creates a new mock instance.
func NewMockWorkflow(ctrl *gomock.Controller) *MockWorkflow {
	mock := &MockWorkflow{ctrl: ctrl}
	mock.recorder = &MockWorkflowMockRecorder{mock}
	return mock
}

// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockWorkflow) EXPECT() *MockWorkflowMockRecorder {
	return m.recorder
}

// Close mocks base method.
func (m *MockWorkflow) Close() {
	m.ctrl.T.Helper()
	m.ctrl.Call(m, "Close")
}

// Close indicates an expected call of Close.
func (mr *MockWorkflowMockRecorder) Close() *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*MockWorkflow)(nil).Close))
}

// DeleteDagInstances mocks base method.
func (m *MockWorkflow) DeleteDagInstances(ctx context.Context, bizId string, excludeDags map[string]bool) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "DeleteDagInstances", ctx, bizId, excludeDags)
	ret0, _ := ret[0].(error)
	return ret0
}

// DeleteDagInstances indicates an expected call of DeleteDagInstances.
func (mr *MockWorkflowMockRecorder) DeleteDagInstances(ctx, bizId, excludeDags any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteDagInstances", reflect.TypeOf((*MockWorkflow)(nil).DeleteDagInstances), ctx, bizId, excludeDags)
}

// DeleteDagInstancesWithDagId mocks base method.
func (m *MockWorkflow) DeleteDagInstancesWithDagId(ctx context.Context, bizId, dagId string) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "DeleteDagInstancesWithDagId", ctx, bizId, dagId)
	ret0, _ := ret[0].(error)
	return ret0
}

// DeleteDagInstancesWithDagId indicates an expected call of DeleteDagInstancesWithDagId.
func (mr *MockWorkflowMockRecorder) DeleteDagInstancesWithDagId(ctx, bizId, dagId any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteDagInstancesWithDagId", reflect.TypeOf((*MockWorkflow)(nil).DeleteDagInstancesWithDagId), ctx, bizId, dagId)
}

// IsLeader mocks base method.
func (m *MockWorkflow) IsLeader() bool {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "IsLeader")
	ret0, _ := ret[0].(bool)
	return ret0
}

// IsLeader indicates an expected call of IsLeader.
func (mr *MockWorkflowMockRecorder) IsLeader() *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IsLeader", reflect.TypeOf((*MockWorkflow)(nil).IsLeader))
}

// ListAllRunningDagInstances mocks base method.
func (m *MockWorkflow) ListAllRunningDagInstances() ([]*entity.DagInstance, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ListAllRunningDagInstances")
	ret0, _ := ret[0].([]*entity.DagInstance)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// ListAllRunningDagInstances indicates an expected call of ListAllRunningDagInstances.
func (mr *MockWorkflowMockRecorder) ListAllRunningDagInstances() *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListAllRunningDagInstances", reflect.TypeOf((*MockWorkflow)(nil).ListAllRunningDagInstances))
}

// ListAllRunningTaskInstances mocks base method.
func (m *MockWorkflow) ListAllRunningTaskInstances() ([]*entity.TaskInstance, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ListAllRunningTaskInstances")
	ret0, _ := ret[0].([]*entity.TaskInstance)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// ListAllRunningTaskInstances indicates an expected call of ListAllRunningTaskInstances.
func (mr *MockWorkflowMockRecorder) ListAllRunningTaskInstances() *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListAllRunningTaskInstances", reflect.TypeOf((*MockWorkflow)(nil).ListAllRunningTaskInstances))
}

// ListRunningDagInstances mocks base method.
func (m *MockWorkflow) ListRunningDagInstances(ctx context.Context, dagIds []string, bizId string) ([]*entity.DagInstance, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ListRunningDagInstances", ctx, dagIds, bizId)
	ret0, _ := ret[0].([]*entity.DagInstance)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// ListRunningDagInstances indicates an expected call of ListRunningDagInstances.
func (mr *MockWorkflowMockRecorder) ListRunningDagInstances(ctx, dagIds, bizId any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListRunningDagInstances", reflect.TypeOf((*MockWorkflow)(nil).ListRunningDagInstances), ctx, dagIds, bizId)
}

// ListRunningDagInstancesByBizIds mocks base method.
func (m *MockWorkflow) ListRunningDagInstancesByBizIds(ctx context.Context, dagIds, bizIds []string) ([]*entity.DagInstance, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ListRunningDagInstancesByBizIds", ctx, dagIds, bizIds)
	ret0, _ := ret[0].([]*entity.DagInstance)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// ListRunningDagInstancesByBizIds indicates an expected call of ListRunningDagInstancesByBizIds.
func (mr *MockWorkflowMockRecorder) ListRunningDagInstancesByBizIds(ctx, dagIds, bizIds any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListRunningDagInstancesByBizIds", reflect.TypeOf((*MockWorkflow)(nil).ListRunningDagInstancesByBizIds), ctx, dagIds, bizIds)
}

// ListRunningDagInstancesByNamePrefix mocks base method.
func (m *MockWorkflow) ListRunningDagInstancesByNamePrefix(ctx context.Context, namePrefix, bizId string) ([]*entity.DagInstance, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ListRunningDagInstancesByNamePrefix", ctx, namePrefix, bizId)
	ret0, _ := ret[0].([]*entity.DagInstance)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// ListRunningDagInstancesByNamePrefix indicates an expected call of ListRunningDagInstancesByNamePrefix.
func (mr *MockWorkflowMockRecorder) ListRunningDagInstancesByNamePrefix(ctx, namePrefix, bizId any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListRunningDagInstancesByNamePrefix", reflect.TypeOf((*MockWorkflow)(nil).ListRunningDagInstancesByNamePrefix), ctx, namePrefix, bizId)
}

// ListRunningDagInstancesByOrgId mocks base method.
func (m *MockWorkflow) ListRunningDagInstancesByOrgId(ctx context.Context, dagIds []string, orgId string) ([]*entity.DagInstance, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ListRunningDagInstancesByOrgId", ctx, dagIds, orgId)
	ret0, _ := ret[0].([]*entity.DagInstance)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// ListRunningDagInstancesByOrgId indicates an expected call of ListRunningDagInstancesByOrgId.
func (mr *MockWorkflowMockRecorder) ListRunningDagInstancesByOrgId(ctx, dagIds, orgId any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListRunningDagInstancesByOrgId", reflect.TypeOf((*MockWorkflow)(nil).ListRunningDagInstancesByOrgId), ctx, dagIds, orgId)
}

// RegisterDag mocks base method.
func (m *MockWorkflow) RegisterDag(dag *entity.Dag, actions []run.Action) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "RegisterDag", dag, actions)
	ret0, _ := ret[0].(error)
	return ret0
}

// RegisterDag indicates an expected call of RegisterDag.
func (mr *MockWorkflowMockRecorder) RegisterDag(dag, actions any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RegisterDag", reflect.TypeOf((*MockWorkflow)(nil).RegisterDag), dag, actions)
}

// RegisterDynamicDagThenRun mocks base method.
func (m *MockWorkflow) RegisterDynamicDagThenRun(ctx context.Context, prefix string, opt *workflow.CreateDynamicDagOption) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "RegisterDynamicDagThenRun", ctx, prefix, opt)
	ret0, _ := ret[0].(error)
	return ret0
}

// RegisterDynamicDagThenRun indicates an expected call of RegisterDynamicDagThenRun.
func (mr *MockWorkflowMockRecorder) RegisterDynamicDagThenRun(ctx, prefix, opt any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RegisterDynamicDagThenRun", reflect.TypeOf((*MockWorkflow)(nil).RegisterDynamicDagThenRun), ctx, prefix, opt)
}

// RegisterMetrics mocks base method.
func (m *MockWorkflow) RegisterMetrics(registry prometheus.Registerer) {
	m.ctrl.T.Helper()
	m.ctrl.Call(m, "RegisterMetrics", registry)
}

// RegisterMetrics indicates an expected call of RegisterMetrics.
func (mr *MockWorkflowMockRecorder) RegisterMetrics(registry any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RegisterMetrics", reflect.TypeOf((*MockWorkflow)(nil).RegisterMetrics), registry)
}

// RunDag mocks base method.
func (m *MockWorkflow) RunDag(ctx context.Context, dagId, orgId, projectId string, specVar map[string]string) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "RunDag", ctx, dagId, orgId, projectId, specVar)
	ret0, _ := ret[0].(error)
	return ret0
}

// RunDag indicates an expected call of RunDag.
func (mr *MockWorkflowMockRecorder) RunDag(ctx, dagId, orgId, projectId, specVar any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RunDag", reflect.TypeOf((*MockWorkflow)(nil).RunDag), ctx, dagId, orgId, projectId, specVar)
}

// RunDagWithBizId mocks base method.
func (m *MockWorkflow) RunDagWithBizId(ctx context.Context, dagId, orgId, projectId, bizId string, specVar map[string]string) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "RunDagWithBizId", ctx, dagId, orgId, projectId, bizId, specVar)
	ret0, _ := ret[0].(error)
	return ret0
}

// RunDagWithBizId indicates an expected call of RunDagWithBizId.
func (mr *MockWorkflowMockRecorder) RunDagWithBizId(ctx, dagId, orgId, projectId, bizId, specVar any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RunDagWithBizId", reflect.TypeOf((*MockWorkflow)(nil).RunDagWithBizId), ctx, dagId, orgId, projectId, bizId, specVar)
}

// SetDagInstanceLifecycleHook mocks base method.
func (m *MockWorkflow) SetDagInstanceLifecycleHook(hook entity.DagInstanceLifecycleHook) {
	m.ctrl.T.Helper()
	m.ctrl.Call(m, "SetDagInstanceLifecycleHook", hook)
}

// SetDagInstanceLifecycleHook indicates an expected call of SetDagInstanceLifecycleHook.
func (mr *MockWorkflowMockRecorder) SetDagInstanceLifecycleHook(hook any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetDagInstanceLifecycleHook", reflect.TypeOf((*MockWorkflow)(nil).SetDagInstanceLifecycleHook), hook)
}

// Start mocks base method.
func (m *MockWorkflow) Start() error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "Start")
	ret0, _ := ret[0].(error)
	return ret0
}

// Start indicates an expected call of Start.
func (mr *MockWorkflowMockRecorder) Start() *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Start", reflect.TypeOf((*MockWorkflow)(nil).Start))
}