// Code generated by mockery v2.26.0. DO NOT EDIT.

package mock

import (
	run "github.com/shiningrush/fastflow/pkg/entity/run"
	mock "github.com/stretchr/testify/mock"
	zap "go.uber.org/zap"
)

// DagWaiter is an autogenerated mock type for the DagWaiter type
type DagWaiter struct {
	mock.Mock
}

// WaitUntilSuccess provides a mock function with given fields: ctx, logger, insIDKey
func (_m *DagWaiter) WaitUntilSuccess(ctx run.ExecuteContext, logger *zap.Logger, insIDKey string) error {
	ret := _m.Called(ctx, logger, insIDKey)

	var r0 error
	if rf, ok := ret.Get(0).(func(run.ExecuteContext, *zap.Logger, string) error); ok {
		r0 = rf(ctx, logger, insIDKey)
	} else {
		r0 = ret.Error(0)
	}

	return r0
}

type mockConstructorTestingTNewDagWaiter interface {
	mock.TestingT
	Cleanup(func())
}

// NewDagWaiter creates a new instance of DagWaiter. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
func NewDagWaiter(t mockConstructorTestingTNewDagWaiter) *DagWaiter {
	mock := &DagWaiter{}
	mock.Mock.Test(t)

	t.Cleanup(func() { mock.AssertExpectations(t) })

	return mock
}