2022-08-22 08:21:13 +02:00
// Code generated by mockery v2.14.0. DO NOT EDIT.
2020-06-16 11:58:58 +02:00
package task
import (
context "context"
dao "github.com/goharbor/harbor/src/pkg/task/dao"
mock "github.com/stretchr/testify/mock"
q "github.com/goharbor/harbor/src/lib/q"
time "time"
)
// mockTaskDAO is an autogenerated mock type for the TaskDAO type
type mockTaskDAO struct {
mock . Mock
}
// Count provides a mock function with given fields: ctx, query
func ( _m * mockTaskDAO ) Count ( ctx context . Context , query * q . Query ) ( int64 , error ) {
ret := _m . Called ( ctx , query )
var r0 int64
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * q . Query ) int64 ) ; ok {
r0 = rf ( ctx , query )
} else {
r0 = ret . Get ( 0 ) . ( int64 )
}
var r1 error
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , * q . Query ) error ) ; ok {
r1 = rf ( ctx , query )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// Create provides a mock function with given fields: ctx, _a1
func ( _m * mockTaskDAO ) Create ( ctx context . Context , _a1 * dao . Task ) ( int64 , error ) {
ret := _m . Called ( ctx , _a1 )
var r0 int64
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * dao . Task ) int64 ) ; ok {
r0 = rf ( ctx , _a1 )
} else {
r0 = ret . Get ( 0 ) . ( int64 )
}
var r1 error
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , * dao . Task ) error ) ; ok {
r1 = rf ( ctx , _a1 )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// Delete provides a mock function with given fields: ctx, id
func ( _m * mockTaskDAO ) Delete ( ctx context . Context , id int64 ) error {
ret := _m . Called ( ctx , id )
var r0 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 ) error ) ; ok {
r0 = rf ( ctx , id )
} else {
r0 = ret . Error ( 0 )
}
return r0
}
// Get provides a mock function with given fields: ctx, id
func ( _m * mockTaskDAO ) Get ( ctx context . Context , id int64 ) ( * dao . Task , error ) {
ret := _m . Called ( ctx , id )
var r0 * dao . Task
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 ) * dao . Task ) ; ok {
r0 = rf ( ctx , id )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * dao . Task )
}
}
var r1 error
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , int64 ) error ) ; ok {
r1 = rf ( ctx , id )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// GetMaxEndTime provides a mock function with given fields: ctx, executionID
func ( _m * mockTaskDAO ) GetMaxEndTime ( ctx context . Context , executionID int64 ) ( time . Time , error ) {
ret := _m . Called ( ctx , executionID )
var r0 time . Time
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 ) time . Time ) ; ok {
r0 = rf ( ctx , executionID )
} else {
r0 = ret . Get ( 0 ) . ( time . Time )
}
var r1 error
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , int64 ) error ) ; ok {
r1 = rf ( ctx , executionID )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// List provides a mock function with given fields: ctx, query
func ( _m * mockTaskDAO ) List ( ctx context . Context , query * q . Query ) ( [ ] * dao . Task , error ) {
ret := _m . Called ( ctx , query )
var r0 [ ] * dao . Task
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * q . Query ) [ ] * dao . Task ) ; ok {
r0 = rf ( ctx , query )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( [ ] * dao . Task )
}
}
var r1 error
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , * q . Query ) error ) ; ok {
r1 = rf ( ctx , query )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// ListStatusCount provides a mock function with given fields: ctx, executionID
func ( _m * mockTaskDAO ) ListStatusCount ( ctx context . Context , executionID int64 ) ( [ ] * dao . StatusCount , error ) {
ret := _m . Called ( ctx , executionID )
var r0 [ ] * dao . StatusCount
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 ) [ ] * dao . StatusCount ) ; ok {
r0 = rf ( ctx , executionID )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( [ ] * dao . StatusCount )
}
}
var r1 error
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , int64 ) error ) ; ok {
r1 = rf ( ctx , executionID )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// Update provides a mock function with given fields: ctx, _a1, props
func ( _m * mockTaskDAO ) Update ( ctx context . Context , _a1 * dao . Task , props ... string ) error {
_va := make ( [ ] interface { } , len ( props ) )
for _i := range props {
_va [ _i ] = props [ _i ]
}
var _ca [ ] interface { }
_ca = append ( _ca , ctx , _a1 )
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
var r0 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * dao . Task , ... string ) error ) ; ok {
r0 = rf ( ctx , _a1 , props ... )
} else {
r0 = ret . Error ( 0 )
}
return r0
}
// UpdateStatus provides a mock function with given fields: ctx, id, status, statusRevision
func ( _m * mockTaskDAO ) UpdateStatus ( ctx context . Context , id int64 , status string , statusRevision int64 ) error {
ret := _m . Called ( ctx , id , status , statusRevision )
var r0 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 , string , int64 ) error ) ; ok {
r0 = rf ( ctx , id , status , statusRevision )
} else {
r0 = ret . Error ( 0 )
}
return r0
}
2022-06-07 10:04:33 +02:00
2022-08-22 08:21:13 +02:00
type mockConstructorTestingTnewMockTaskDAO interface {
2022-06-07 10:04:33 +02:00
mock . TestingT
Cleanup ( func ( ) )
}
// newMockTaskDAO creates a new instance of mockTaskDAO. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
2022-08-22 08:21:13 +02:00
func newMockTaskDAO ( t mockConstructorTestingTnewMockTaskDAO ) * mockTaskDAO {
2022-06-07 10:04:33 +02:00
mock := & mockTaskDAO { }
mock . Mock . Test ( t )
t . Cleanup ( func ( ) { mock . AssertExpectations ( t ) } )
return mock
}