2023-11-05 07:37:25 +01:00
// Code generated by mockery v2.35.4. DO NOT EDIT.
2023-03-21 03:10:55 +01:00
package task
import (
context "context"
mock "github.com/stretchr/testify/mock"
)
// mockSweepManager is an autogenerated mock type for the SweepManager type
type mockSweepManager struct {
mock . Mock
}
// Clean provides a mock function with given fields: ctx, execID
func ( _m * mockSweepManager ) Clean ( ctx context . Context , execID [ ] int64 ) error {
ret := _m . Called ( ctx , execID )
var r0 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , [ ] int64 ) error ) ; ok {
r0 = rf ( ctx , execID )
} else {
r0 = ret . Error ( 0 )
}
return r0
}
2023-03-21 11:48:14 +01:00
// FixDanglingStateExecution provides a mock function with given fields: ctx
func ( _m * mockSweepManager ) FixDanglingStateExecution ( ctx context . Context ) error {
ret := _m . Called ( ctx )
var r0 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context ) error ) ; ok {
r0 = rf ( ctx )
} else {
r0 = ret . Error ( 0 )
}
return r0
}
2023-03-21 03:10:55 +01:00
// ListCandidates provides a mock function with given fields: ctx, vendorType, retainCnt
func ( _m * mockSweepManager ) ListCandidates ( ctx context . Context , vendorType string , retainCnt int64 ) ( [ ] int64 , error ) {
ret := _m . Called ( ctx , vendorType , retainCnt )
var r0 [ ] int64
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , string , int64 ) ( [ ] int64 , error ) ) ; ok {
return rf ( ctx , vendorType , retainCnt )
}
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , string , int64 ) [ ] int64 ) ; ok {
r0 = rf ( ctx , vendorType , retainCnt )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( [ ] int64 )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , string , int64 ) error ) ; ok {
r1 = rf ( ctx , vendorType , retainCnt )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
2023-11-05 07:37:25 +01:00
// newMockSweepManager creates a new instance of mockSweepManager. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
// The first argument is typically a *testing.T value.
func newMockSweepManager ( t interface {
2023-03-21 03:10:55 +01:00
mock . TestingT
Cleanup ( func ( ) )
2023-11-05 07:37:25 +01:00
} ) * mockSweepManager {
2023-03-21 03:10:55 +01:00
mock := & mockSweepManager { }
mock . Mock . Test ( t )
t . Cleanup ( func ( ) { mock . AssertExpectations ( t ) } )
return mock
}