2024-04-11 05:37:59 +02:00
// Code generated by mockery v2.42.2. DO NOT EDIT.
2021-03-22 10:27:23 +01:00
2023-03-03 03:17:47 +01:00
package policy
2021-03-22 10:27:23 +01:00
import (
context "context"
model "github.com/goharbor/harbor/src/pkg/notification/policy/model"
mock "github.com/stretchr/testify/mock"
q "github.com/goharbor/harbor/src/lib/q"
)
// Manager is an autogenerated mock type for the Manager type
type Manager struct {
mock . Mock
}
// Count provides a mock function with given fields: ctx, query
func ( _m * Manager ) Count ( ctx context . Context , query * q . Query ) ( int64 , error ) {
ret := _m . Called ( ctx , query )
2024-04-11 05:37:59 +02:00
if len ( ret ) == 0 {
panic ( "no return value specified for Count" )
}
2021-03-22 10:27:23 +01:00
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 )
}
2021-03-22 10:27:23 +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
}
// Create provides a mock function with given fields: ctx, _a1
func ( _m * Manager ) Create ( ctx context . Context , _a1 * model . Policy ) ( int64 , error ) {
ret := _m . Called ( ctx , _a1 )
2024-04-11 05:37:59 +02:00
if len ( ret ) == 0 {
panic ( "no return value specified for Create" )
}
2021-03-22 10:27:23 +01:00
var r0 int64
2023-03-11 01:16:11 +01:00
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * model . Policy ) ( int64 , error ) ) ; ok {
return rf ( ctx , _a1 )
}
2021-03-22 10:27:23 +01:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * model . Policy ) int64 ) ; ok {
r0 = rf ( ctx , _a1 )
} else {
r0 = ret . Get ( 0 ) . ( int64 )
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , * model . Policy ) error ) ; ok {
r1 = rf ( ctx , _a1 )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// Delete provides a mock function with given fields: ctx, policyID
func ( _m * Manager ) Delete ( ctx context . Context , policyID int64 ) error {
ret := _m . Called ( ctx , policyID )
2024-04-11 05:37:59 +02:00
if len ( ret ) == 0 {
panic ( "no return value specified for Delete" )
}
2021-03-22 10:27:23 +01:00
var r0 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 ) error ) ; ok {
r0 = rf ( ctx , policyID )
} else {
r0 = ret . Error ( 0 )
}
return r0
}
// Get provides a mock function with given fields: ctx, id
func ( _m * Manager ) Get ( ctx context . Context , id int64 ) ( * model . Policy , error ) {
ret := _m . Called ( ctx , id )
2024-04-11 05:37:59 +02:00
if len ( ret ) == 0 {
panic ( "no return value specified for Get" )
}
2021-03-22 10:27:23 +01:00
var r0 * model . Policy
2023-03-11 01:16:11 +01:00
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 ) ( * model . Policy , error ) ) ; ok {
return rf ( ctx , id )
}
2021-03-22 10:27:23 +01:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 ) * model . Policy ) ; ok {
r0 = rf ( ctx , id )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * model . Policy )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , int64 ) error ) ; ok {
r1 = rf ( ctx , id )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// GetRelatedPolices provides a mock function with given fields: ctx, projectID, eventType
func ( _m * Manager ) GetRelatedPolices ( ctx context . Context , projectID int64 , eventType string ) ( [ ] * model . Policy , error ) {
ret := _m . Called ( ctx , projectID , eventType )
2024-04-11 05:37:59 +02:00
if len ( ret ) == 0 {
panic ( "no return value specified for GetRelatedPolices" )
}
2021-03-22 10:27:23 +01:00
var r0 [ ] * model . Policy
2023-03-11 01:16:11 +01:00
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 , string ) ( [ ] * model . Policy , error ) ) ; ok {
return rf ( ctx , projectID , eventType )
}
2021-03-22 10:27:23 +01:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 , string ) [ ] * model . Policy ) ; ok {
r0 = rf ( ctx , projectID , eventType )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( [ ] * model . Policy )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , int64 , string ) error ) ; ok {
r1 = rf ( ctx , projectID , eventType )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// List provides a mock function with given fields: ctx, query
func ( _m * Manager ) List ( ctx context . Context , query * q . Query ) ( [ ] * model . Policy , error ) {
ret := _m . Called ( ctx , query )
2024-04-11 05:37:59 +02:00
if len ( ret ) == 0 {
panic ( "no return value specified for List" )
}
2021-03-22 10:27:23 +01:00
var r0 [ ] * model . Policy
2023-03-11 01:16:11 +01:00
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * q . Query ) ( [ ] * model . Policy , error ) ) ; ok {
return rf ( ctx , query )
}
2021-03-22 10:27:23 +01:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * q . Query ) [ ] * model . Policy ) ; ok {
r0 = rf ( ctx , query )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( [ ] * model . Policy )
}
}
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
}
// Update provides a mock function with given fields: ctx, _a1
func ( _m * Manager ) Update ( ctx context . Context , _a1 * model . Policy ) error {
ret := _m . Called ( ctx , _a1 )
2024-04-11 05:37:59 +02:00
if len ( ret ) == 0 {
panic ( "no return value specified for Update" )
}
2021-03-22 10:27:23 +01:00
var r0 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * model . Policy ) error ) ; ok {
r0 = rf ( ctx , _a1 )
} else {
r0 = ret . Error ( 0 )
}
return r0
}
2022-06-07 10:04:33 +02:00
2023-11-05 07:37:25 +01:00
// NewManager creates a new instance of Manager. 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 NewManager ( t interface {
2022-06-07 10:04:33 +02:00
mock . TestingT
Cleanup ( func ( ) )
2023-11-05 07:37:25 +01:00
} ) * Manager {
2022-06-07 10:04:33 +02:00
mock := & Manager { }
mock . Mock . Test ( t )
t . Cleanup ( func ( ) { mock . AssertExpectations ( t ) } )
return mock
}