2022-08-22 08:21:13 +02:00
// Code generated by mockery v2.14.0. DO NOT EDIT.
2019-12-20 02:24:57 +01:00
2020-01-22 11:28:08 +01:00
package repository
2019-12-20 02:24:57 +01:00
import (
2021-05-19 05:29:35 +02:00
context "context"
model "github.com/goharbor/harbor/src/pkg/repository/model"
mock "github.com/stretchr/testify/mock"
q "github.com/goharbor/harbor/src/lib/q"
2019-12-20 02:24:57 +01:00
)
2021-05-19 05:29:35 +02:00
// Controller is an autogenerated mock type for the Controller type
type Controller struct {
2019-12-20 02:24:57 +01:00
mock . Mock
}
2021-12-10 06:26:17 +01:00
// AddPullCount provides a mock function with given fields: ctx, id, count
func ( _m * Controller ) AddPullCount ( ctx context . Context , id int64 , count uint64 ) error {
ret := _m . Called ( ctx , id , count )
2021-05-19 05:29:35 +02:00
var r0 error
2021-12-10 06:26:17 +01:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 , uint64 ) error ) ; ok {
r0 = rf ( ctx , id , count )
2021-05-19 05:29:35 +02:00
} else {
r0 = ret . Error ( 0 )
}
return r0
2020-01-22 11:28:08 +01:00
}
2021-05-19 05:29:35 +02:00
// Count provides a mock function with given fields: ctx, query
func ( _m * Controller ) 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 )
}
2020-02-18 08:51:38 +01:00
2021-05-19 05:29:35 +02:00
return r0 , r1
2020-02-18 08:51:38 +01:00
}
2021-05-19 05:29:35 +02:00
// Delete provides a mock function with given fields: ctx, id
func ( _m * Controller ) 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 )
2020-01-16 01:04:22 +01:00
}
2020-01-22 11:28:08 +01:00
2021-05-19 05:29:35 +02:00
return r0
2019-12-20 02:24:57 +01:00
}
2021-05-19 05:29:35 +02:00
// Ensure provides a mock function with given fields: ctx, name
func ( _m * Controller ) Ensure ( ctx context . Context , name string ) ( bool , int64 , error ) {
ret := _m . Called ( ctx , name )
var r0 bool
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , string ) bool ) ; ok {
r0 = rf ( ctx , name )
} else {
r0 = ret . Get ( 0 ) . ( bool )
}
var r1 int64
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , string ) int64 ) ; ok {
r1 = rf ( ctx , name )
} else {
r1 = ret . Get ( 1 ) . ( int64 )
2020-01-16 01:04:22 +01:00
}
2021-05-19 05:29:35 +02:00
var r2 error
if rf , ok := ret . Get ( 2 ) . ( func ( context . Context , string ) error ) ; ok {
r2 = rf ( ctx , name )
} else {
r2 = ret . Error ( 2 )
}
return r0 , r1 , r2
2020-01-16 01:04:22 +01:00
}
2021-05-19 05:29:35 +02:00
// Get provides a mock function with given fields: ctx, id
func ( _m * Controller ) Get ( ctx context . Context , id int64 ) ( * model . RepoRecord , error ) {
ret := _m . Called ( ctx , id )
var r0 * model . RepoRecord
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 ) * model . RepoRecord ) ; ok {
r0 = rf ( ctx , id )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * model . RepoRecord )
}
2020-01-16 01:04:22 +01:00
}
2021-05-19 05:29:35 +02:00
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
2019-12-20 02:24:57 +01:00
}
2020-02-07 02:11:54 +01:00
2021-05-19 05:29:35 +02:00
// GetByName provides a mock function with given fields: ctx, name
func ( _m * Controller ) GetByName ( ctx context . Context , name string ) ( * model . RepoRecord , error ) {
ret := _m . Called ( ctx , name )
var r0 * model . RepoRecord
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , string ) * model . RepoRecord ) ; ok {
r0 = rf ( ctx , name )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * model . RepoRecord )
}
}
var r1 error
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , string ) error ) ; ok {
r1 = rf ( ctx , name )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
2020-02-07 02:11:54 +01:00
}
2020-02-18 08:51:38 +01:00
2021-05-19 05:29:35 +02:00
// List provides a mock function with given fields: ctx, query
func ( _m * Controller ) List ( ctx context . Context , query * q . Query ) ( [ ] * model . RepoRecord , error ) {
ret := _m . Called ( ctx , query )
var r0 [ ] * model . RepoRecord
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * q . Query ) [ ] * model . RepoRecord ) ; ok {
r0 = rf ( ctx , query )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( [ ] * model . RepoRecord )
}
}
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
2020-02-18 08:51:38 +01:00
}
2020-03-18 11:38:37 +01:00
2021-05-19 05:29:35 +02:00
// Update provides a mock function with given fields: ctx, _a1, properties
func ( _m * Controller ) Update ( ctx context . Context , _a1 * model . RepoRecord , properties ... string ) error {
_va := make ( [ ] interface { } , len ( properties ) )
for _i := range properties {
_va [ _i ] = properties [ _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 , * model . RepoRecord , ... string ) error ) ; ok {
r0 = rf ( ctx , _a1 , properties ... )
} else {
r0 = ret . Error ( 0 )
}
return r0
2020-03-18 11:38:37 +01:00
}
2022-06-07 10:04:33 +02:00
2022-08-22 08:21:13 +02:00
type mockConstructorTestingTNewController interface {
2022-06-07 10:04:33 +02:00
mock . TestingT
Cleanup ( func ( ) )
}
// 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.
2022-08-22 08:21:13 +02:00
func NewController ( t mockConstructorTestingTNewController ) * Controller {
2022-06-07 10:04:33 +02:00
mock := & Controller { }
mock . Mock . Test ( t )
t . Cleanup ( func ( ) { mock . AssertExpectations ( t ) } )
return mock
}