2024-04-11 05:37:59 +02:00
// Code generated by mockery v2.42.2. DO NOT EDIT.
2020-09-30 09:14:00 +02:00
package replication
import (
context "context"
2021-03-31 09:49:23 +02:00
model "github.com/goharbor/harbor/src/controller/replication/model"
2020-09-30 09:14:00 +02:00
mock "github.com/stretchr/testify/mock"
q "github.com/goharbor/harbor/src/lib/q"
2021-03-31 09:49:23 +02:00
regmodel "github.com/goharbor/harbor/src/pkg/reg/model"
replication "github.com/goharbor/harbor/src/controller/replication"
2020-09-30 09:14:00 +02:00
)
// Controller is an autogenerated mock type for the Controller type
type Controller struct {
mock . Mock
}
2021-03-03 04:45:33 +01:00
// CreatePolicy provides a mock function with given fields: ctx, policy
2021-03-31 09:49:23 +02:00
func ( _m * Controller ) CreatePolicy ( ctx context . Context , policy * model . Policy ) ( int64 , error ) {
2021-03-03 04:45:33 +01:00
ret := _m . Called ( ctx , policy )
2024-04-11 05:37:59 +02:00
if len ( ret ) == 0 {
panic ( "no return value specified for CreatePolicy" )
}
2021-03-03 04:45:33 +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 , policy )
}
2021-03-31 09:49:23 +02:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * model . Policy ) int64 ) ; ok {
2021-03-03 04:45:33 +01:00
r0 = rf ( ctx , policy )
} else {
r0 = ret . Get ( 0 ) . ( int64 )
}
2021-03-31 09:49:23 +02:00
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , * model . Policy ) error ) ; ok {
2021-03-03 04:45:33 +01:00
r1 = rf ( ctx , policy )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// DeletePolicy provides a mock function with given fields: ctx, id
func ( _m * Controller ) DeletePolicy ( ctx context . Context , id int64 ) error {
ret := _m . Called ( ctx , id )
2024-04-11 05:37:59 +02:00
if len ( ret ) == 0 {
panic ( "no return value specified for DeletePolicy" )
}
2021-03-03 04:45:33 +01:00
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
}
2020-09-30 09:14:00 +02:00
// ExecutionCount provides a mock function with given fields: ctx, query
func ( _m * Controller ) ExecutionCount ( 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 ExecutionCount" )
}
2020-09-30 09:14:00 +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 , * q . Query ) ( int64 , error ) ) ; ok {
return rf ( ctx , query )
}
2020-09-30 09:14:00 +02: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
}
// GetExecution provides a mock function with given fields: ctx, executionID
2021-03-31 09:49:23 +02:00
func ( _m * Controller ) GetExecution ( ctx context . Context , executionID int64 ) ( * replication . Execution , error ) {
2020-09-30 09:14:00 +02:00
ret := _m . Called ( ctx , executionID )
2024-04-11 05:37:59 +02:00
if len ( ret ) == 0 {
panic ( "no return value specified for GetExecution" )
}
2021-03-31 09:49:23 +02:00
var r0 * replication . Execution
2023-03-11 01:16:11 +01:00
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 ) ( * replication . Execution , error ) ) ; ok {
return rf ( ctx , executionID )
}
2021-03-31 09:49:23 +02:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 ) * replication . Execution ) ; ok {
2020-09-30 09:14:00 +02:00
r0 = rf ( ctx , executionID )
} else {
if ret . Get ( 0 ) != nil {
2021-03-31 09:49:23 +02:00
r0 = ret . Get ( 0 ) . ( * replication . Execution )
2020-09-30 09:14:00 +02:00
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , int64 ) error ) ; ok {
r1 = rf ( ctx , executionID )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
2021-03-03 04:45:33 +01:00
// GetPolicy provides a mock function with given fields: ctx, id
2021-03-31 09:49:23 +02:00
func ( _m * Controller ) GetPolicy ( ctx context . Context , id int64 ) ( * model . Policy , error ) {
2021-03-03 04:45:33 +01:00
ret := _m . Called ( ctx , id )
2024-04-11 05:37:59 +02:00
if len ( ret ) == 0 {
panic ( "no return value specified for GetPolicy" )
}
2021-03-31 09:49:23 +02: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-31 09:49:23 +02:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 ) * model . Policy ) ; ok {
2021-03-03 04:45:33 +01:00
r0 = rf ( ctx , id )
} else {
if ret . Get ( 0 ) != nil {
2021-03-31 09:49:23 +02:00
r0 = ret . Get ( 0 ) . ( * model . Policy )
2021-03-03 04:45:33 +01:00
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , int64 ) error ) ; ok {
r1 = rf ( ctx , id )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
2020-09-30 09:14:00 +02:00
// GetTask provides a mock function with given fields: ctx, taskID
2021-03-31 09:49:23 +02:00
func ( _m * Controller ) GetTask ( ctx context . Context , taskID int64 ) ( * replication . Task , error ) {
2020-09-30 09:14:00 +02:00
ret := _m . Called ( ctx , taskID )
2024-04-11 05:37:59 +02:00
if len ( ret ) == 0 {
panic ( "no return value specified for GetTask" )
}
2021-03-31 09:49:23 +02:00
var r0 * replication . Task
2023-03-11 01:16:11 +01:00
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 ) ( * replication . Task , error ) ) ; ok {
return rf ( ctx , taskID )
}
2021-03-31 09:49:23 +02:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 ) * replication . Task ) ; ok {
2020-09-30 09:14:00 +02:00
r0 = rf ( ctx , taskID )
} else {
if ret . Get ( 0 ) != nil {
2021-03-31 09:49:23 +02:00
r0 = ret . Get ( 0 ) . ( * replication . Task )
2020-09-30 09:14:00 +02:00
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , int64 ) error ) ; ok {
r1 = rf ( ctx , taskID )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// GetTaskLog provides a mock function with given fields: ctx, taskID
func ( _m * Controller ) GetTaskLog ( ctx context . Context , taskID int64 ) ( [ ] byte , error ) {
ret := _m . Called ( ctx , taskID )
2024-04-11 05:37:59 +02:00
if len ( ret ) == 0 {
panic ( "no return value specified for GetTaskLog" )
}
2020-09-30 09:14:00 +02:00
var r0 [ ] byte
2023-03-11 01:16:11 +01:00
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 ) ( [ ] byte , error ) ) ; ok {
return rf ( ctx , taskID )
}
2020-09-30 09:14:00 +02:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 ) [ ] byte ) ; ok {
r0 = rf ( ctx , taskID )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( [ ] byte )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , int64 ) error ) ; ok {
r1 = rf ( ctx , taskID )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// ListExecutions provides a mock function with given fields: ctx, query
2021-03-31 09:49:23 +02:00
func ( _m * Controller ) ListExecutions ( ctx context . Context , query * q . Query ) ( [ ] * replication . Execution , error ) {
2020-09-30 09:14:00 +02:00
ret := _m . Called ( ctx , query )
2024-04-11 05:37:59 +02:00
if len ( ret ) == 0 {
panic ( "no return value specified for ListExecutions" )
}
2021-03-31 09:49:23 +02:00
var r0 [ ] * replication . Execution
2023-03-11 01:16:11 +01:00
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * q . Query ) ( [ ] * replication . Execution , error ) ) ; ok {
return rf ( ctx , query )
}
2021-03-31 09:49:23 +02:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * q . Query ) [ ] * replication . Execution ) ; ok {
2020-09-30 09:14:00 +02:00
r0 = rf ( ctx , query )
} else {
if ret . Get ( 0 ) != nil {
2021-03-31 09:49:23 +02:00
r0 = ret . Get ( 0 ) . ( [ ] * replication . Execution )
2021-03-03 04:45:33 +01:00
}
}
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
}
// ListPolicies provides a mock function with given fields: ctx, query
2021-03-31 09:49:23 +02:00
func ( _m * Controller ) ListPolicies ( ctx context . Context , query * q . Query ) ( [ ] * model . Policy , error ) {
2021-03-03 04:45:33 +01:00
ret := _m . Called ( ctx , query )
2024-04-11 05:37:59 +02:00
if len ( ret ) == 0 {
panic ( "no return value specified for ListPolicies" )
}
2021-03-31 09:49:23 +02: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-31 09:49:23 +02:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * q . Query ) [ ] * model . Policy ) ; ok {
2021-03-03 04:45:33 +01:00
r0 = rf ( ctx , query )
} else {
if ret . Get ( 0 ) != nil {
2021-03-31 09:49:23 +02:00
r0 = ret . Get ( 0 ) . ( [ ] * model . Policy )
2020-09-30 09:14:00 +02:00
}
}
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
}
// ListTasks provides a mock function with given fields: ctx, query
2021-03-31 09:49:23 +02:00
func ( _m * Controller ) ListTasks ( ctx context . Context , query * q . Query ) ( [ ] * replication . Task , error ) {
2020-09-30 09:14:00 +02:00
ret := _m . Called ( ctx , query )
2024-04-11 05:37:59 +02:00
if len ( ret ) == 0 {
panic ( "no return value specified for ListTasks" )
}
2021-03-31 09:49:23 +02:00
var r0 [ ] * replication . Task
2023-03-11 01:16:11 +01:00
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * q . Query ) ( [ ] * replication . Task , error ) ) ; ok {
return rf ( ctx , query )
}
2021-03-31 09:49:23 +02:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * q . Query ) [ ] * replication . Task ) ; ok {
2020-09-30 09:14:00 +02:00
r0 = rf ( ctx , query )
} else {
if ret . Get ( 0 ) != nil {
2021-03-31 09:49:23 +02:00
r0 = ret . Get ( 0 ) . ( [ ] * replication . Task )
2020-09-30 09:14:00 +02:00
}
}
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
}
2021-03-03 04:45:33 +01:00
// PolicyCount provides a mock function with given fields: ctx, query
func ( _m * Controller ) PolicyCount ( 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 PolicyCount" )
}
2021-03-03 04:45:33 +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-03 04:45:33 +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-09-30 09:14:00 +02:00
// Start provides a mock function with given fields: ctx, policy, resource, trigger
2021-03-31 09:49:23 +02:00
func ( _m * Controller ) Start ( ctx context . Context , policy * model . Policy , resource * regmodel . Resource , trigger string ) ( int64 , error ) {
2020-09-30 09:14:00 +02:00
ret := _m . Called ( ctx , policy , resource , trigger )
2024-04-11 05:37:59 +02:00
if len ( ret ) == 0 {
panic ( "no return value specified for Start" )
}
2020-09-30 09:14:00 +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 , * model . Policy , * regmodel . Resource , string ) ( int64 , error ) ) ; ok {
return rf ( ctx , policy , resource , trigger )
}
2021-03-31 09:49:23 +02:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * model . Policy , * regmodel . Resource , string ) int64 ) ; ok {
2020-09-30 09:14:00 +02:00
r0 = rf ( ctx , policy , resource , trigger )
} else {
r0 = ret . Get ( 0 ) . ( int64 )
}
2021-03-31 09:49:23 +02:00
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , * model . Policy , * regmodel . Resource , string ) error ) ; ok {
2020-09-30 09:14:00 +02:00
r1 = rf ( ctx , policy , resource , trigger )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// Stop provides a mock function with given fields: ctx, executionID
func ( _m * Controller ) Stop ( ctx context . Context , executionID int64 ) error {
ret := _m . Called ( ctx , executionID )
2024-04-11 05:37:59 +02:00
if len ( ret ) == 0 {
panic ( "no return value specified for Stop" )
}
2020-09-30 09:14:00 +02:00
var r0 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 ) error ) ; ok {
r0 = rf ( ctx , executionID )
} else {
r0 = ret . Error ( 0 )
}
return r0
}
// TaskCount provides a mock function with given fields: ctx, query
func ( _m * Controller ) TaskCount ( 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 TaskCount" )
}
2020-09-30 09:14:00 +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 , * q . Query ) ( int64 , error ) ) ; ok {
return rf ( ctx , query )
}
2020-09-30 09:14:00 +02: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
}
2021-03-03 04:45:33 +01:00
// UpdatePolicy provides a mock function with given fields: ctx, policy, props
2021-03-31 09:49:23 +02:00
func ( _m * Controller ) UpdatePolicy ( ctx context . Context , policy * model . Policy , props ... string ) error {
2021-03-03 04:45:33 +01:00
_va := make ( [ ] interface { } , len ( props ) )
for _i := range props {
_va [ _i ] = props [ _i ]
}
var _ca [ ] interface { }
_ca = append ( _ca , ctx , policy )
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
2024-04-11 05:37:59 +02:00
if len ( ret ) == 0 {
panic ( "no return value specified for UpdatePolicy" )
}
2021-03-03 04:45:33 +01:00
var r0 error
2021-03-31 09:49:23 +02:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * model . Policy , ... string ) error ) ; ok {
2021-03-03 04:45:33 +01:00
r0 = rf ( ctx , policy , props ... )
} else {
r0 = ret . Error ( 0 )
}
return r0
}
2022-06-07 10:04:33 +02:00
2023-11-05 07:37:25 +01:00
// NewController creates a new instance of Controller. 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 NewController ( t interface {
2022-06-07 10:04:33 +02:00
mock . TestingT
Cleanup ( func ( ) )
2023-11-05 07:37:25 +01:00
} ) * Controller {
2022-06-07 10:04:33 +02:00
mock := & Controller { }
mock . Mock . Test ( t )
t . Cleanup ( func ( ) { mock . AssertExpectations ( t ) } )
return mock
}