2023-11-05 07:37:25 +01:00
// Code generated by mockery v2.35.4. DO NOT EDIT.
2020-07-09 04:53:32 +02:00
package scheduler
import (
context "context"
q "github.com/goharbor/harbor/src/lib/q"
mock "github.com/stretchr/testify/mock"
)
// mockDAO is an autogenerated mock type for the DAO type
type mockDAO struct {
mock . Mock
}
2022-11-17 13:12:07 +01:00
// Count provides a mock function with given fields: ctx, query
func ( _m * mockDAO ) Count ( ctx context . Context , query * q . Query ) ( int64 , error ) {
ret := _m . Called ( ctx , query )
var r0 int64
2023-03-11 01:16:11 +01:00
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * q . Query ) ( int64 , error ) ) ; ok {
return rf ( ctx , query )
}
2022-11-17 13:12:07 +01:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * q . Query ) int64 ) ; ok {
r0 = rf ( ctx , query )
} else {
r0 = ret . Get ( 0 ) . ( int64 )
}
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
}
2020-07-22 03:59:32 +02:00
// Create provides a mock function with given fields: ctx, s
func ( _m * mockDAO ) Create ( ctx context . Context , s * schedule ) ( int64 , error ) {
ret := _m . Called ( ctx , s )
2020-07-09 04:53:32 +02:00
var r0 int64
2023-03-11 01:16:11 +01:00
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * schedule ) ( int64 , error ) ) ; ok {
return rf ( ctx , s )
}
2020-07-09 04:53:32 +02:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * schedule ) int64 ) ; ok {
2020-07-22 03:59:32 +02:00
r0 = rf ( ctx , s )
2020-07-09 04:53:32 +02:00
} else {
r0 = ret . Get ( 0 ) . ( int64 )
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , * schedule ) error ) ; ok {
2020-07-22 03:59:32 +02:00
r1 = rf ( ctx , s )
2020-07-09 04:53:32 +02:00
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// Delete provides a mock function with given fields: ctx, id
func ( _m * mockDAO ) 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 * mockDAO ) Get ( ctx context . Context , id int64 ) ( * schedule , error ) {
ret := _m . Called ( ctx , id )
var r0 * schedule
2023-03-11 01:16:11 +01:00
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 ) ( * schedule , error ) ) ; ok {
return rf ( ctx , id )
}
2020-07-09 04:53:32 +02:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 ) * schedule ) ; ok {
r0 = rf ( ctx , id )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * schedule )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , int64 ) error ) ; ok {
r1 = rf ( ctx , id )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// List provides a mock function with given fields: ctx, query
func ( _m * mockDAO ) List ( ctx context . Context , query * q . Query ) ( [ ] * schedule , error ) {
ret := _m . Called ( ctx , query )
var r0 [ ] * schedule
2023-03-11 01:16:11 +01:00
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * q . Query ) ( [ ] * schedule , error ) ) ; ok {
return rf ( ctx , query )
}
2020-07-09 04:53:32 +02:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * q . Query ) [ ] * schedule ) ; ok {
r0 = rf ( ctx , query )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( [ ] * schedule )
}
}
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
}
2020-07-22 03:59:32 +02:00
// Update provides a mock function with given fields: ctx, s, props
func ( _m * mockDAO ) Update ( ctx context . Context , s * schedule , props ... string ) error {
2020-07-09 04:53:32 +02:00
_va := make ( [ ] interface { } , len ( props ) )
for _i := range props {
_va [ _i ] = props [ _i ]
}
var _ca [ ] interface { }
2020-07-22 03:59:32 +02:00
_ca = append ( _ca , ctx , s )
2020-07-09 04:53:32 +02:00
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
var r0 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * schedule , ... string ) error ) ; ok {
2020-07-22 03:59:32 +02:00
r0 = rf ( ctx , s , props ... )
2020-07-09 04:53:32 +02:00
} else {
r0 = ret . Error ( 0 )
}
return r0
}
2021-05-11 15:19:36 +02:00
// UpdateRevision provides a mock function with given fields: ctx, id, revision
func ( _m * mockDAO ) UpdateRevision ( ctx context . Context , id int64 , revision int64 ) ( int64 , error ) {
ret := _m . Called ( ctx , id , revision )
var r0 int64
2023-03-11 01:16:11 +01:00
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 , int64 ) ( int64 , error ) ) ; ok {
return rf ( ctx , id , revision )
}
2021-05-11 15:19:36 +02:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 , int64 ) int64 ) ; ok {
r0 = rf ( ctx , id , revision )
} else {
r0 = ret . Get ( 0 ) . ( int64 )
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , int64 , int64 ) error ) ; ok {
r1 = rf ( ctx , id , revision )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
2022-06-07 10:04:33 +02:00
2023-11-05 07:37:25 +01:00
// newMockDAO creates a new instance of mockDAO. 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 newMockDAO ( t interface {
2022-06-07 10:04:33 +02:00
mock . TestingT
Cleanup ( func ( ) )
2023-11-05 07:37:25 +01:00
} ) * mockDAO {
2022-06-07 10:04:33 +02:00
mock := & mockDAO { }
mock . Mock . Test ( t )
t . Cleanup ( func ( ) { mock . AssertExpectations ( t ) } )
return mock
}